python - instalar - pip3 install rest_framework

Django Rest Framework: establece la base de datos dinámicamente desde el parámetro URL

Question

Estoy tratando de encontrar la manera correcta de hacer esto:

Servicio de usuarios:

/api/<country>/users
/api/us/users

Ese servicio debe usar la base de datos correspondiente al país en la URL.

settings.py:

DATABASES = {
''default'': {},
''us'': {
    ''ENGINE'': ''django.db.backends.postgresql_psycopg2'',
    ''NAME'': ''XXX_US'', 
    ''USER'': ''US'',
    ''PASSWORD'': ''XXX'',
    ''HOST'': ''localhost'',
    ''PORT'': ''5432'',
},
''es'': {
    ''ENGINE'': ''django.db.backends.postgresql_psycopg2'',
    ''NAME'': ''XXX_ES'', 
    ''USER'': ''ES'',
    ''PASSWORD'': ''XXX'',
    ''HOST'': ''localhost'',
    ''PORT'': ''5432'',
}  }

Para configurar la base de datos en el ModelViewSet I a esto:

class UserViewSet(viewsets.ModelViewSet):

    model = User
    serializer_class = UserSerializer

    def get_queryset(self):
        country = self.kwargs[''country'']
        return User.objects.using(country).all()

El problema aparece cuando trato de hacer un POST o PUT. ¿Debo sobrescribir el método create () o save () del serializador? Hay alguna otra manera de hacer esto?

¡Muchas gracias!

Answer 1

Creo que el mejor lugar para tal funcionalidad es un QuerySet o un ModelManager . Por ejemplo, el serializador predeterminado del DRF usa el administrador del modelo predeterminado para crear objetos. Desafortunadamente, QuerySet no tiene una manera de cambiar fácilmente la base de datos actual ( self.db ) dependiendo de los campos de los modelos, por lo que deberá anular todos los métodos relevantes.

class UserQuerySet(models.QuerySet):
    def create(self, **kwargs):
        obj = self.model(**kwargs)
        self._for_write = True
        obj.save(force_insert=True, using=kwargs.get(''country''))
        return obj

class User(models.Model):
    objects = UserQuerySet.as_manager()

Answer 2

Gracias Alex! Tu respuesta, ayúdame mucho. Para resolver completamente el problema, tengo que hacer esto:

views.py

class UserViewSet(viewsets.ModelViewSet):

    model = User
    serializer_class = UserSerializer

    def perform_create(self, serializer):        
        serializer.validated_data[''country''] = self.kwargs.get(''country'')
        serializer.create(serializer.validated_data)

Y luego en el models.py:

class UserQuerySet(models.QuerySet):

    def create(self, **kwargs):
        country = kwargs.get(''country'')  #Get the country
        kwargs.pop(''country'')            #Pop the country from kwargs
        obj = self.model(**kwargs)
        self._for_write = True                
        obj.save(force_insert=True, using=country)
        return obj

class User(models.Model):

    objects = UserQuerySet.as_manager()
    id = models.AutoField(primary_key=True,db_column=''user_id'')
    name = models.CharField(max_length=40)

    class Meta:
        managed = True
        db_table = Constants().USER

¡Y el problema está resuelto! Muchas gracias.