4

我有两个模型与模型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')

我怎样才能用反向关系解决这个问题?

4

2 回答 2

7

目前因为您在 person 属性上使用 ForeignKey 字段,这意味着当您访问反向关系时它会返回一个列表。

一种解决方案是使用与 slug 相关的字段,尽管它必须具有manyread_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')
于 2018-03-22T00:14:56.743 回答
0

您可以使用自定义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')
于 2018-03-21T19:36:48.053 回答