python - Django 自定义字典响应

Question

我正在使用 Django Rest 框架，并且我有一个以字典格式返回信息的 View 。这是我的模型:

class Child(models.Model):
    name = models.CharField(max_length=50, unique=True)
    age = models.IntegerField(null=False, blank=False)
    height = models.IntegerField(null=False, blank=False)

我想将以下响应作为字典返回，其中 name 字段是键:

{
    "Alice Allen": {
        "age": 3,
        "height": 98
    },
    "Bob Brady": {
        "age": 4,
        "height": 105
    }
}

我知道我可以使用 GenericAPIView 并实现“get”方法中的逻辑(获取数据并解析到字典中)，但我想知道是否有另一种使用 Django 序列化器的解决方案。

谢谢。

Answer 1

如果您使用的是viewset，您可以尝试重写list方法并自定义响应:

class ChildViewSet(viewsets.ModelViewSet):

    def list(self, request, *args, **kwargs):
        response = super(ChildViewSet, self).list(request, *args, **kwargs)
        response.data = {result.pop('name'): result for result in response.data}
        return response