时间戳

首页 > 解决方案 > 在graphene-django中使循环相关的模型属性可查询

python - 在graphene-django中使循环相关的模型属性可查询

问题描述

假设我的 Django 模型如下所示：

class User(Model):
    username = CharField(...)

    @property
    def projects(self):
        return # projects the user is assigned to

class Project(Model):
    name = CharField(...)

    @property
    def users(self):
        return # users assigned to the project

class Assignment(Model):
    user = ForeignKey(User)
    project = ForeignKey(Project)

（我故意没有为属性编写查询集以使代码更具可读性。）

我希望能够在 graphene-django 查询中看到这些属性。我的ModelType课程如下所示：

class UserType(DjangoObjectType):
    class Meta:
        model = User

    projects = List(ProjectType)


class ProjectType(DjangoObjectType):
    class Meta:
        model = Project

    users = List(UserType)

但是，此代码不起作用，因为ProjectType在UserType类中未定义，并且交换这些类不会有太大帮助。

我试图根本不包含属性，并且在创建了所有ModelType类之后，我尝试通过setattr(). 这种 hack 肯定会设置属性，因此类具有完全相同的结构，但似乎为时已晚：根本无法从 GraphQL 访问属性。

现在我没有主意了。有什么方法可以实现所需的行为？

标签： pythondjangographqlgraphene-python

解决方案

类的属性ModelType仅在类被初始化时才重要。稍后添加字段的一种方法是直接将它们添加到它们所在的位置。解决方案：

class UserType(DjangoObjectType):
    class Meta:
        model = User


class ProjectType(DjangoObjectType):
    class Meta:
        model = Project

UserType._meta.fields["projects"] = get_field_as(List(ProjectType), _as=Field)
ProjectType._meta.fields["users"] = get_field_as(List(UserType), _as=Field)

并且这两个属性都可以按需要工作。

推荐阅读