具有反向关系的Django rest框架序列化器

Question

我有两个employee与person模型有关系但与模型person没有关系的employee模型。

喜欢：

class Person(models.Model):
    name = models.CharField(max_length=100)
    address = models.CharField(max_length=100)

class Employee(models.Model):
    person = models.ForeignKey(Person, related_name='person_info')
    code = models.CharField()

在这种情况下，我需要code人员序列化程序中的字段数据。

---

我可以使用个人模型中的编写方法或在个人序列化器中使用SerializerMethodField来解决此问题

像这样：

def get_employee_code(self):
    return Employee.objects.get(person=self).id

并将其添加为亲自序列化器中的源

employee_code = serializers.CharField(source='get_employee_code')

或将员工序列化程序添加到人员序列化程序中

class PersonSerializer(serializers.ModelSerializer):
    employee = EmployeeSerializer()
    class Meta:
        model = Person
        fields = ('name', 'address', 'employee')

---

但是我试图用反向关系做到这一点，但我做不到。我已经尝试过这样，它给出了一个错误

序列化器：

class PersonSerializer(serializers.ModelSerializer):
    employee_code = serializers.CharField(source='person_info.code')
    class Meta:
        model = Person
        fields = ('name', 'address', 'employee_code')

我该如何解决反向关系呢？

Answer 1

目前，由于您正在使用person属性上的ForeignKey字段，这意味着当您访问反向关系时，它将返回一个列表。

一种解决方案是使用与子段相关的字段，尽管该字段必须具有many并且read_only设置为True，并且由于ForeignKey字段而将返回列表。

class PersonSerializer(serializers.ModelSerializer):
    employee_code = serializers.SlugRelatedField(
        source='person_info',
        slug_field='code',
        many=True,
        read_only=True,
    )

    class Meta:
        model = Person
        fields = ('name', 'address', 'employee_code')

另一个选择是将ForeignKey更改为OneToOneField，仍然需要read_only将其设置为True，但不会返回列表。

class Person(models.Model):
    name = models.CharField(max_length=100)
    address = models.CharField(max_length=100)

class Employee(models.Model):
    person = models.OneToOneField(Person, related_name='person_info')
    code = models.CharField()

class PersonSerializer(serializers.ModelSerializer):
    employee_code = serializers.SlugRelatedField(
        source='person_info',
        slug_field='code',
        read_only=True,
    )

    class Meta:
        model = Person
        fields = ('name', 'address', 'employee_code')

或者，如果您不想更改ForeignKey，则可以employee_code向模型添加属性方法，而不是返回person_info关系中的第一个员工代码。

class Person(models.Model):
    name = models.CharField(max_length=100)
    address = models.CharField(max_length=100)

    @property
    def employee_code(self):
        employees = self.person_info.filter()
        if employees.exists():
            return employees.first().code
        return ''

class Employee(models.Model):
    person = models.OneToOneField(Person, related_name='person_info')
    code = models.CharField()

class PersonSerializer(serializers.ModelSerializer):
    employee_code = serializers.CharField(
        read_only=True,
    )

    class Meta:
        model = Person
        fields = ('name', 'address', 'employee_code')

Answer 2

您可以使用自定义SerializerMethodField()访问反向关系

class PersonSerializer(serializers.ModelSerializer):
    employee_code = serializers.SerializerMethodField()

    def get_employee_code(self, obj):
        return obj.person_info.code

    class Meta:
        model = Person
        fields = ('name', 'address', 'employee_code')