如何序列化DRF中的泛型关系?

2024-04-26 22:53:43 发布

您现在位置:Python中文网/ 问答频道 /正文

我有一个通过泛型关系与其他三个模型关联的管理员模型。你知道吗

A = 'Active'
    D = 'Disabled'
    status = (
        (A, 'Active'),
        (D, 'Disabled')
    )

    limit_choices = (models.Q(app_label='core', model='track') | models.Q(app_label='core',
                    model='company') | models.Q(app_label='core', model='member') )
    content_type = models.ForeignKey(ContentType, limit_choices_to=limit_choices, on_delete=models.CASCADE)
    object_id = models.PositiveIntegerField()
    content_object = GenericForeignKey('content_type', 'object_id')
    status = models.CharField(max_length=7, choices=status, default=D)

class Track(models.Model):
    name = models.CharField(max_length=70)



class Company(models.Model):
    partner_id = models.ForeignKey(Track, related_name='companies', on_delete=models.CASCADE)
    name = models.CharField(max_length=70)


class Member(models.Model):
    carrier_id = models.ForeignKey(Track,  on_delete=models.CASCADE)
    name = models.CharField(max_length=70)

对于序列化,我使用官方文档中的示例code。你知道吗

序列化程序.py

class AdministratorSerializer(serializers.RelatedField):

    class Meta:
        fields = ('id', 'status', )
        model = Administrator

    def to_representation(self, value):

        if isinstance(value, Track):
            return value.id
        elif isinstance(value, Member):
            return value.id
        elif isinstance(value, Company):
            return value.id
        raise Exception('Unexpected type of tagged object')

但我错了

TypeError at /administrator/
__init__() takes exactly 1 argument (3 given)

请帮我解决这个任务/


Tags: nameidappmodelobjectvaluemodelsstatus
1条回答
网友
1楼 · 发布于 2024-04-26 22:53:43

DRF文档说,创建一个通用的序列化程序字段(这里,GenericField),并连接到序列化程序类(这里,AdministratorSerializer)。你知道吗

class GenericField(serializers.RelatedField):

    def to_representation(self, value):

        if isinstance(value, Track):
            return value.id
        elif isinstance(value, Member):
            return value.id
        elif isinstance(value, Company):
            return value.id
        raise Exception('Unexpected type of tagged object')


class AdministratorSerializer(serializers.ModelSerializer):
    generic_data = GenericField(source='content_object', read_only=True)

    class Meta:
        fields = ('id', 'status', 'generic_data')
        model = Administrator

相关问题 更多 >