使循环依赖的模型属性在graphene-django中可查询

Question

假设我有如下的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)

(我故意不为属性编写查询集，以提高代码的可读性。)

我希望能够在石墨烯-django查询中看到这些属性。我的ModelType类如下所示：

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

    projects = List(ProjectType)


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

    users = List(UserType)

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

我尝试完全不包含属性，在创建了所有ModelType类之后，我尝试通过setattr()添加属性。这样的黑客肯定会设置属性，使类具有完全相同的结构，但似乎为时已晚:属性根本无法从GraphQL访问。

现在我的想法用完了。有什么方法可以实现预期的行为吗？

Answer 1

只有在初始化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)

并且这两个属性都能按预期工作。