37 lines
1023 B
Python
37 lines
1023 B
Python
from django.db import models
|
|
from django.forms.models import model_to_dict
|
|
from rest_framework import serializers
|
|
|
|
|
|
class SecondDbManager(models.Manager):
|
|
def get_queryset(self):
|
|
qs = super().get_queryset()
|
|
|
|
# if `use_db` is set on model use that for choosing the DB
|
|
if hasattr(self.model, 'use_db'):
|
|
qs = qs.using(self.model.use_db)
|
|
|
|
return qs
|
|
|
|
|
|
class MigrateMixin(models.Model):
|
|
"""Mixin to data transfer from legacy database"""
|
|
use_db = 'legacy'
|
|
objects = SecondDbManager()
|
|
|
|
def _parse_instance_fields(self, fields):
|
|
model_dict = model_to_dict(self, fields)
|
|
print(getattr(self, "using", "Using not found"))
|
|
|
|
class Meta:
|
|
abstract = True
|
|
|
|
|
|
class TransferSerializerMixin(serializers.ModelSerializer):
|
|
"""Mixin for transferring legacy db models."""
|
|
|
|
def create(self, validated_data):
|
|
qs = self.Meta.model.objects.filter(**validated_data)
|
|
if not qs.exists():
|
|
return super().create(validated_data)
|