Django Rest Framework使OnetoOne关系船感觉它是一个模型

时间:2013-08-29 17:31:42

标签: django rest django-models django-rest-framework

我将User保存在两个不同的模型UserProfileUser中。现在从API的角度来看,没有人真正关心这两者是不同的。

所以我在这里:

class UserSerializer(serializers.HyperlinkedModelSerializer):
    class Meta:
        model = User
        fields = ('url', 'username', 'first_name', 'last_name', 'email')

class UserPSerializer(serializers.HyperlinkedModelSerializer):
    full_name = Field(source='full_name')
    class Meta:
        model = UserProfile
        fields = ('url', 'mobile', 'user','favourite_locations')

因此,在UserPSerializer中,字段user只是该资源的链接。但从用户的角度来看,他根本没有理由了解User

是否有一些技巧可以将它们混合在一起并作为一个模型呈现给用户,或者我必须以某种方式手动执行此操作。

3 个答案:

答案 0 :(得分:33)

如果您还覆盖序列化程序上的创建和更新方法,则可以使用@kahlo's approach进行POST和PUT。

给出这样的个人资料模型:

class Profile(models.Model):
    user = models.OneToOneField(User)
    avatar_url = models.URLField(default='', blank=True)  # e.g.

这是一个用户序列化程序,它读取和写入其他配置文件字段:

class UserSerializer(serializers.HyperlinkedModelSerializer):
    # A field from the user's profile:
    avatar_url = serializers.URLField(source='profile.avatar_url', allow_blank=True)

    class Meta:
        model = User
        fields = ('url', 'username', 'avatar_url')

    def create(self, validated_data):
        profile_data = validated_data.pop('profile', None)
        user = super(UserSerializer, self).create(validated_data)
        self.update_or_create_profile(user, profile_data)
        return user

    def update(self, instance, validated_data):
        profile_data = validated_data.pop('profile', None)
        self.update_or_create_profile(instance, profile_data)
        return super(UserSerializer, self).update(instance, validated_data)

    def update_or_create_profile(self, user, profile_data):
        # This always creates a Profile if the User is missing one;
        # change the logic here if that's not right for your app
        Profile.objects.update_or_create(user=user, defaults=profile_data)

生成的API会根据需要显示扁平的用户资源:

GET /users/5/
{
    "url": "http://localhost:9090/users/5/", 
    "username": "test", 
    "avatar_url": "http://example.com/avatar.jpg"
}

您可以在POST和PUT请求中包含配置文件的avatar_url字段。 (并且用户资源上的DELETE也将删除其Profile模型,尽管这只是Django的正常删除级联。)

如果缺少(在任何更新时),此处的逻辑将始终为用户创建配置文件模型。对于用户和个人资料,这可能就是您想要的。对于其他关系,它可能不是,您需要更改更新或创建逻辑。 (这就是DRF doesn't automatically write through a nested relationship的原因。)

答案 1 :(得分:12)

我刚刚遇到过这个;我还没有找到一个好的解决方案,尤其是回写我的UserUserProfile模型。我目前正在使用SerializerMethodField手动展平我的序列化程序,这非常令人恼火,但它确实有效:

class UserSerializer(serializers.HyperlinkedModelSerializer):

    mobile = serializers.SerializerMethodField('get_mobile')
    favourite_locations = serializers.SerializerMethodField('get_favourite_locations')

    class Meta:
        model = User
        fields = ('url', 'username', 'first_name', 'last_name', 'email', 'mobile', 'favourite_locations')

    def get_mobile(self, obj):
        return obj.get_profile().mobile

    def get_favourite_locations(self, obj):
        return obj.get_profile().favourite_locations

这是可怕的手册,但你最终得到:

{
    "url": "http://example.com/api/users/1",
    "username": "jondoe",
    "first_name": "Jon",
    "last_name": "Doe",
    "email": "jdoe@example.com",
    "mobile": "701-680-3095",
    "favourite_locations": [
        "Paris",
        "London",
        "Tokyo"
    ]
}

其中,我想你正在寻找。

答案 2 :(得分:6)

我会在UserPSerializer上实施修改,因为字段不会增长:

class UserSerializer(serializers.HyperlinkedModelSerializer):
    class Meta:
        model = User
        fields = ('url', 'username', 'first_name', 'last_name', 'email')

class UserPSerializer(serializers.HyperlinkedModelSerializer):
    url = serializers.CharField(source='user.url')
    username = serializers.CharField(source='user.username')
    first_name = serializers.CharField(source='user.first_name')
    last_name = serializers.CharField(source='user.last_name')
    email = serializers.CharField(source='user.email')

    class Meta:
        model = UserProfile
        fields = ('mobile', 'favourite_locations',
                  'url', 'username', 'first_name', 'last_name', 'email')