Django REST框架字段,具有自定义保存/还原逻辑



如何使用自定义保存/还原逻辑创建DJANGO REST框架读写字段?我正在使用DRF 3.5.4和Django 1.8。

类似的东西:

class UserSerializer(serializer.Serializer):
    # username is a field in the backing model, no problemo
    username = serializers.CharField()
    # type is not an actual model field, but reflects whether the user has 
    # Django permission "foo" set. I.e. "normal" = no permission, "silly" = 
    # permission is set.
    type = serializers.ChoiceField(["normal", "silly"])

换句话说:创建用户时,如果type == "silly",请为他设置权限" foo",而如果类型为"正常",则不设置。在检索用户时,type == "silly"如果用户有权限" FOO"设置,则"正常"(如果不是)。

不允许/需要更新用户,但是我有兴趣知道如何将其应用于更新。

没有测试它,但是这种方法应该起作用。仅供阅读,您可以用户序列化。但是这种类型的字段是只读的。要在validated_data中使用type,您需要使用自定义字段。

from rest_framework import serializers   
class UserTypeField(serializers.CharField):
    def get_attribute(self, instance):
        return 'silly' if instance.has_perm('foo') else 'normal'

def set_permission_by_type(instance, user_type):
    if user_type == 'silly':
        instance.add_perm('foo')
    elif user_type == 'normal':
        instance.remove_perm('foo')

class UserSerializer(serializers.ModelSerializer):
    type = UserTypeField()
    def update(self, instance, validated_data):
        user_type = validated_data.pop('type', None)
        instance = super().update(instance, validated_data)
        if user_type:
            set_permission_by_type(instance, user_type)
        return instance
    def create(self, validated_data):
        user_type = validated_data.pop('type', None)
        instance = super().create(validated_data)
        if user_type:
            set_permission_by_type(instance, user_type)
        return instance

最新更新