如何在 django 的不同应用程序中扩展 AbstractUser 模型

Question

我有一个类User，它继承自另一个名为accounts 的不同django 应用程序中的AbstractUser。然后我在user 字段的Employee 类中引用了它。问题是我无法在 django_rest 的 API 视图中查看用户的全名。

class User(AbstractUser):

   def get_full_name(self):
        full_name = '%s %s' % (self.last_name.upper(), self.first_name)
        return full_name.strip()



class Employee(models.Model):

    user = models.OneToOneField(User,on_delete=models.CASCADE)
    tax_id_number = models.CharField(
        max_length=20, null=True, verbose_name='Tax ID')
    account_number = models.CharField(
        max_length=20, null=True, verbose_name='Account Number')
    joining_date = models.DateField(null=True, verbose_name="Joining Date")
    designation = models.ForeignKey(Group,related_name='designation',blank=True)
    department = models.ForeignKey(Group,related_name='department',blank=True)

员工序列化器如下

class EmployeeSerializer(serializers.ModelSerializer):
    # TODO: Define serializer fields here
    user = UserSerializer
    # designation = GroupListSerializer()
    # department = GroupListSerializer()

    class Meta:
        model = Employee
        # fields = '__all__'
        fields = ['user','tax_id_number','account_number','joining_date','designation','department']



    def create(self, validated_data):
        user_data = validated_data.pop('user')
        employee = Employee.objects.create(**validated_data)

        for user in user_data:
            user, created = User.objects.get()
            employee.user.add(user)
        return employee

Answer 1

如果您使用 get_full_name 作为模型序列化程序中的字段之一，那么问题在于它不是属性。

您需要为模型序列化程序使用 @property 装饰器才能访问它。

class User(AbstractUser):
    @property 
    def get_full_name(self):
        full_name = '%s %s' % (self.last_name.upper(), self.first_name)
        return full_name.strip()

不同之处在于，当您使用 @property 装饰器时，python 对象允许像字段一样访问定义的方法。 DRF 的序列化程序假定 fields 元组中列出的名称是属性

例子：

>>> class A:
...  @property
...  def x(self):
...   return 2
...  def y(self):
...   return 3
... 
>>> a = A()
>>> a.x
2
>>> a.y
<bound method A.y of <__main__.A instance at 0x10556d368>>
>>> 

最后一条建议是不要使用get_full_name 作为属性的名称。即使属性是方法也不应该有动词作为名称。您可以简单地使用full_name 作为属性名称。