序列化程序扩展用户字段与Django Rest Framework中的主要用户按同一级别扩展 [英] Serializer Extend User field by same level with Main User in Django Rest Framework

问题描述

我在Django中这样编写一个扩展用户序列化程序：

I write a Extend User serializers in Django like this:

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

File Models.py Extend User App:

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',
        ]

此序列化程序是为了扩展主用户序列化程序：

This serializer in order to extend for Main User Serializers:

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

因此，我的API将呈现为：

As a result, my API will render into this:

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

但现在我想使用与主要用户相同的API，如下所示：

But now I want to API in same level with main User like this:

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

请帮助我做到这一点。

Please help me to do this. Thanks in advance!

推荐答案

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

You can write your own to_representation method.

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

这篇关于序列化程序扩展用户字段与Django Rest Framework中的主要用户按同一级别扩展的文章就介绍到这了，希望我们推荐的答案对大家有所帮助，也希望大家多多支持IT屋！