序列化器将用户字段扩展为与 Django Rest Framework 中的主用户相同的级别

Question

我在 Django 中编写了一个扩展用户序列化器，如下所示：

文件Models.py扩展用户应用程序：

class Profile(models.Model):
    user = models.OneToOneField(User, unique=True)
    nickname = models.CharField(max_length=50, null=True, blank=True)
    gender = models.CharField(max_length=100, choices=GENDER_CHOICES, blank=True, null=True, default='MALE')

文件serializers.py扩展用户应用程序：

class ProfilePatchSerializer(ModelSerializer):
    languages = serializers.ChoiceField(choices=LANGUAGE_CHOICES, default='EN')
    gender = serializers.ChoiceField(choices=GENDER_CHOICES, default='1')
    class Meta:
        model = Profile
        fields = [
            'gender',
            'language',
        ]

该序列化器是为了扩展主用户序列化器：

class UserCreateUpdateSerializer(ModelSerializer):
    profile = ProfilePatchSerializer()
    class Meta:
        model = User
        fields = [
            'username',
            'email',
            'first_name',
            'last_name',
            'profile',
        ]

结果，我的 API 将呈现为：

{
    "id": 1,
    "username": "feedgit",
    "first_name": "Feed Git",
    "profile": {
        "gender": Male,
        "languages": English
    }
}

但现在我想要与主用户处于同一级别的 API，如下所示：

{
   "id": 1,
   "username": "feedgit",
   "first_name": "Feed Git",
   "gender": Male,
   "languages": English
}

请帮助我做到这一点。提前致谢！

Answer 1

您可以编写自己的to_representation方法。

class UserCreateUpdateSerializer(ModelSerializer):
    profile = ProfilePatchSerializer()
    class Meta:
        model = User
        fields = [
            'username',
            'email',
            'first_name',
            'last_name',
            'profile',
        ]

    def to_representation(self, instance):
        data = super(UserCreateUpdateSerializer, self).to_representation(instance)
        profile = data.pop('profile')
        for key, val in profile.items():
            data.update({key: val})
        return data