使用视图更改DRF响应

时间:2017-04-29 12:17:55

标签: django django-rest-framework

我想改变API的响应。

但是,它不会正确改变结果。我得到KeyError: 'game'

我不知道为什么,因为我的API响应(通过网址)似乎有值game。我可能会对JSON响应和python对象感到困惑。

我在下面有一个API响应示例

来自API的

结果

{
    "pk": 995,
    "game": [
        {
            "name": "Finance",
            "gamelevel": 3
        },
        {
            "name": "Data",
            "gamelevel": 1
        }
    ]
},

views.py

class TagList(viewsets.ModelViewSet):
    queryset = Task.objects.filter(game__isnull=False).all()
    serializer_class = TagSortSerializer

    def get_queryset(self):
    test = self.queryset.values('title', 'game__name')
    result = defaultdict(set)
    for item in queryset:
        parent   = {'name': 'NoLevel_1'}
        children = []
        for game in item['game']:
            if game['gamelevel'] == 1:
                parent = game
            else:
                children.append((game['gamelevel'], game['name']))
        result[parent['name']].update(children)

    result = [ 
        {
            'name': parent, 
            'game_child': [
                {'name': name, 'gamelevel': gamelevel} for gamelevel, name in games
            ],
            'gamelevel': 1, 
        } for parent, games in result.items()
    ]

    return result

1 个答案:

答案 0 :(得分:0)

您使用values查询集方法仅从模型中获取一系列字段,并且您指定的唯一字段为titletag__name 。因此,您无法获得game或您使用的任何其他密钥。

你当然不想在这里使用values;只需通过点查找而不是字典来进行常规查询和访问字段。