Django REST: Объедините две модели

Question

Я только начал использовать фреймворк django rest, и это может быть старый вопрос, но до сих пор я не нашел для него правильного ответа на SO.

Я хочу добавить дополнительные поля профиля в существующую модель пользователя, предоставленную DRF. Прочитав документ, я решил создать новую модель CustomUser и связать ее с моделью пользователя по умолчанию с помощью внешнего ключа.

Вопрос возник, когда я пытался сериализовать модель CustomUser. Я хочу быть JSON как это:

{ 
    "username" : "from_default_user_model", 
    "extra_field" : "from_custom_user_model" 
} 

, но с использованием вложенных сериализаторов следующим образом, я получил вложенное представление JSon.

class UserSerializer(serializers.ModelSerializer): 
    class Meta: 
     model = User 
     fields = ('username',) 

class CustomUserSerializer(serializers.ModelSerializer): 
    user = UserSerializer() 
    class Meta: 
     model = CustomUser 
     fields = ('extra_field',) 

вложенные один выглядит следующим образом:

{ 
    "username" : "from_default_user_model", 
    "user" : 
      { 
       "extra_field" : "from_custom_user_model" 
      } 
} 

То, что я хочу, это свой рода как база данных соединения между двумя таблицами, которые просто дают простую таблицу, а не вложенная один. Как я мог это достичь?

Answer 1

Я предполагаю, что вы собираетесь использовать по умолчанию модели Us Джанго как OneToOneField с пользовательской моделью пользователя , Вам не нужен другой сериализатор для модели пользователя. Поскольку дополнительные поля будут в CustomUser, достаточно написать сериализатор только для пользовательского пользователя.

from django.contrib.auth.models import User 
class CustomUser(models.Model): 
    user = models.OneToOneField(User) 
    # Add your extra fields here. 

Это, как вы это делаете:

CustomUserSerializer(serializers.ModelSerializer): 
    username = Serializers.CharField(source='user.username') 
    email = Serializers.EmailField(source='user.email') 

    class Meta: 
     model = CustomUser 
     fields = ('username', 'email', 'extra_fields_from_CustomUser',) 

К счастью, source позволяет достичь вашего OneToOneField легко, так что вы ничего не нужно.

Answer 2

На модели CustomUser, создать ярлык свойства, а затем добавить его к определению полей в мета

@property 
def username(self): 
    return self.user.username 

class meta: 
    model = ... 
    fields = ("username", "extra-field",...)