django - 在 Graphene Django 中查询多个模型

Question

我在 Django 中有多个模型，名为model1, model2, model3. 每个模型都有相同的字段，并且该字段id是主键。我想使用 GraphQL 查询这个数据库。查询结构应如下所示。

query{
      pod(model: "model1",id: "1")
        {
           id
           data1
           data2
        }
     }

这应该让我从模型 1 中获取 id 1 的对象。如何在 Django 中使用 Graphene 执行此操作？

Answer 1

为您的模型创建一个地图，以及您希望如何从您的 graphql 查询中访问它。这样做：

def resolve_pod(self, root, info, model, id):
    model_map = {'model1': Model1, 'model2': Model2, 'model3': Model3}
    if model in model_map.keys():
        return model_map[model].objects.get(id=1)
    return None