如何在Graphene Flask中实现用户关联文章的草稿过滤查询？

首先得先修正你SQLAlchemy模型里的关系定义，这是基础：你的Post模型里的外键字段名和relationship写反了，正确的写法应该是这样（把post_id改成更清晰的user_id，并修正relationship指向）：

class Post(Base):
    __tablename__ = 'post'
    id = Column(Integer, primary_key=True)
    title = Column(String)
    text = Column(String)
    in_draft = Column(Boolean)
    user_id = Column(Integer, ForeignKey('user.id'))  # 外键指向user表的id
    user = relationship("User", backref='posts')  # 关联User模型，backref让User拥有posts属性

接下来，针对你要过滤用户草稿文章的需求，有两种实用的实现思路：

思路一：扩展User对象的posts字段，添加过滤参数 #

这种方法不需要新增字段，直接让原有的posts支持in_draft过滤，性能也更好（直接在SQL层面过滤）：

class User(SQLAlchemyObjectType):
    """User Object."""
    # 覆盖模型自动生成的posts字段，添加in_draft参数
    posts = graphene.List(lambda: Post, in_draft=graphene.Boolean())

    class Meta:
        model = UserModel
        interfaces = (relay.Node, )
        exclude_fields = ("posts",)  # 排除模型默认生成的posts字段，避免冲突

    def resolve_posts(self, info, in_draft=None):
        # 获取Post的查询对象
        query = Post.get_query(info)
        # 先过滤当前用户的所有文章
        query = query.filter(PostModel.user_id == self.id)
        # 如果传入了in_draft参数，再添加草稿状态过滤
        if in_draft is not None:
            query = query.filter(PostModel.in_draft == in_draft)
        return query.all()

之后你就可以直接在GraphQL查询里使用带参数的posts字段：

{
  user(username: "Jon") {
    name
    last_lame
    username
    posts(in_draft : true) {
      title
      text
      in_draft
      update_at
    }
  }
}

思路二：新增专门的draft_posts字段 #

如果你想把草稿文章和普通文章的查询分开，可以给User对象新增一个draft_posts字段：

class User(SQLAlchemyObjectType):
    """User Object."""
    # 新增草稿文章字段，默认过滤草稿状态为true
    draft_posts = graphene.List(lambda: Post, in_draft=graphene.Boolean(default_value=True))

    class Meta:
        model = UserModel
        interfaces = (relay.Node, )

    def resolve_draft_posts(self, info, in_draft=True):
        # 高性能写法（推荐）：直接在SQL层面过滤
        query = Post.get_query(info)
        return query.filter(PostModel.user_id == self.id, PostModel.in_draft == in_draft).all()
        
        # 小数据量场景可用简单写法：
        # return [post for post in self.posts if post.in_draft == in_draft]

对应的查询会变成：

{
  user(username: "Jon") {
    name
    last_lame
    username
    draft_posts {  # 默认取in_draft=true的文章，也可以手动指定参数
      title
      text
      in_draft
      update_at
    }
  }
}

注意事项 #

• 优先选择思路一，因为它直接在SQL层面做过滤，避免加载所有文章到内存后再筛选，性能更优，尤其是数据量大的场景。
• 确保你的SQLAlchemy模型关系正确，否则关联查询会失效。

内容的提问来源于stack exchange，提问作者Alex