如何在 DRF 中序列化 GenericForeignKey?

Mad*_*bat 5 django django-rest-framework

我有一个模型,它使用通用外键使用“content_type”字段来存储内容类型和“object_id”来存储对象 ID。该模型需要使用 CRUD API 进行操作,而我正在使用 DRF。我有一个模型的序列化程序,但我遇到了一些问题。如果我只是将 content_type 添加到这样的字段列表中

类 MySerializer(ModelSerializer):
    元类:
        字段 = ('name', 'state', 'content_type', 'object_id')

序列化程序将 JSON 表示设置为 ContentType 模型实例的 ID。API 的用户不知道这些 ID,我不想用另一个 API 公开 ContentType 模型。API 用户确实知道他们想要链接的对象类型,因此他们可以发送内容类型名称。所以,我已经定义了这样的序列化器

类 MySerializer(ModelSerializer):
    content_type = serializers.CharField(source="content_type.name")
    元类:
        字段 = ('name', 'state', 'content_type', 'object_id')

序列化模型工作得很好。如果我使用通用关系来链接 User 实例,我会得到类似{'name': 'test', 'state': 'NY', 'content_type': 'user', 'object_id': 123}. 但是当我使用 JSON 结构提交 PUT 或 POST 请求时,DRF 会将其转换为类似{'name': 'test', 'state': 'NY', 'content_type': {'name': {'name': 'user'}}, 'object_id': 123}. 我可以写一些类似的东西

    def 创建(自我,validated_data):
        ct_model =validated_data['content_type']['name']['name']
        validate_data['content_type'] = ContentType.objects.get(model=ct_model)
        返回 MyModel.objects.create(**validated_data)

但这似乎是随意和脆弱的。处理这种情况的正确方法是什么?

更新 #1:目前,我已经通过使用此代码覆盖 to_internal_value() 解决了这个问题

def to_internal_value(self, data):
    content_type = data.get('content_type', None)
    验证数据 = 超级()。to_internal_value(数据)
    尝试:
        validate_data['content_type'] = ContentType.objects.get(model=content_type)
    除了 ContentType.DoesNotExist:
        raise serializers.ValidationError("无效的内容类型 %s" % content_type)
    返回validated_data

显示相关对象似乎是一种丑陋的技巧。

Iva*_*huk 0

您可能必须进行自定义,提供您自己的字段类to_representationto_internal_value方法来序列化此类关系。没那么复杂,你试试吧。

考虑使用 uri 表示法/api/resource/object_id作为序列化格式。

将内容类型转换为资源 uri 并返回可能有点具有挑战性(如果您有多种可能的类型),但我确信一定有一些简单的方法可以做到这一点。