2015-11-01 71 views
1

我延长auth.user模型时返回用户扩展字段:Django的 - 创建用户

Modles.py:

class UserProfile(models.Model): 
    user = models.OneToOneField(User, on_delete=models.CASCADE) 
    phone_number = models.CharField(max_length=15) 

这是我UserSerializer。

当我创建的用户,我可以更新其userprofile场,但我无法弄清楚如何将它添加到系列化的用户,并返回用户字段+的扩展字段

class UserSerializer(serializers.ModelSerializer): 

    class Meta: 
     model = User 
     fields = ('id', 'username', 'first_name', 'last_name', 'password',) 
     read_only_fields = ('phone_number',) 
     extra_kwargs = {'password': {'write_only': True}} 

    def create(self, validated_data): 
     user = User(
      first_name=validated_data['first_name'], 
      username=validated_data['username'], 
      last_name=validated_data['last_name'] 
     ) 

     user.set_password(validated_data['password']) 
     user.save() 
     user.userprofile.phone_number = self.initial_data["phone_number"] 
     user.userprofile.save() 
     the_dict = { 
      'id': user.id, 
      'username': user.username, 
      'first_name': user.first_name, 
      'last_name': user.last_name, 
      'phone_number': user.userprofile.phone_number 
     } 

     return the_dict 

正如你所看到的,我想以某种方式将其添加到字典中,但在我的客户,我只得到了用户的字段(“ID”,“用户名”,“FIRST_NAME”,“姓氏”) 。

如何添加phone_number?

回答

1

试试这个:

class UserSerializer(serializers.ModelSerializer): 

    # source is required to show where this field is coming from 
    phone_number = serializers.CharField(source='userprofile.phone_number') 

    class Meta: 
     model = User 
     # Add phone_number to the fields 
     fields = ('id', 'username', 'first_name', 'last_name', 'password', 'phone_number',) 
     read_only_fields = ('phone_number',) 
     extra_kwargs = {'password': {'write_only': True}}  

我评论上面的代码解释。

+0

是的,我发现问题后才发现。它效果很好 –