如何使用自定义保存/还原逻辑创建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