2013-12-15 16 views
0

我有类似这样的一个问题:Additional (non model) fields in ModelSerializer在ModelSerializer其他信息不在使用型号

我想创建一个对象,具有这样的modelserializer:

class CreateUserSerializer(serializers.ModelSerializer): 
    user = serializers.CharField(source="username") 
    password = serializers.CharField() 
    password2 = serializers.WritableField() 
    ... 

    class Meta: 
     model = User 
    fields = ('user', "password", "password2", ...) 

很显然,我不是有兴趣保存密码2,这只是为了比较。

理想这应该做的伎俩,但对于原因,其没有按预期工作:

def restore_object(self, attrs, instance=None): 
    tags = attrs.pop('password2', None) 
    obj = super(CreateUserSerializer, self).restore_object(attrs, instance) 
    obj.tags = tags 
    logging.info(u"Object gets restored, field 'password2' gets removed, list with arguments for object: {0}".format(tags)) 
    return obj 

我得到的错误是:

'User' object has no attribute 'password2' 
... 
/home/jan/projekte/alarmapp/eclipse_workspace/AlarmServer/AlarmApp/src/external/rest_framework/fields.py in get_component 

55. val = getattr(obj, attr_name) 

任何想法,为什么这是不工作?

回答

0

好像this small Gist应该是有帮助的:

from django.contrib.auth.models import User 
from rest_framework import serializers 

class CreateUserSerializer(serializers.ModelSerializer): 
    password2 = serializers.CharField() 

    def validate_password2(self, attrs, source): 
     password2 = attrs.pop(source) 
     if attrs['password'] != password2: 
      raise serializers.ValidationError('password mismatch') 
     return attrs 

    def to_native(self, obj): 
     self.fields.pop('password2') 
     return super(CreateUserSerializer, self).to_native(obj) 

    class Meta: 
     model = User