python - 如果记录具有与其相关的外键对象，则过滤 SQLAlchemy 对象中的记录

Question

使用 the Flask-SQLAlchemy Quickstart 中的示例:

from datetime import datetime

class Post(db.Model):
    id = db.Column(db.Integer, primary_key=True)
    title = db.Column(db.String(80))
    body = db.Column(db.Text)
    pub_date = db.Column(db.DateTime)

    category_id = db.Column(db.Integer, db.ForeignKey('category.id'))
    category = db.relationship('Category',
        backref=db.backref('posts', lazy='dynamic'))

    def __init__(self, title, body, category, pub_date=None):
        self.title = title
        self.body = body
        if pub_date is None:
            pub_date = datetime.utcnow()
        self.pub_date = pub_date
        self.category = category

    def __repr__(self):
        return '<Post %r>' % self.title


class Category(db.Model):
    id = db.Column(db.Integer, primary_key=True)
    name = db.Column(db.String(50))

    def __init__(self, name):
        self.name = name

    def __repr__(self):
        return '<Category %r>' % self.name

我按照与示例相同的方式重新创建了类别和帖子:

>>> py = Category('Python')
>>> p = Post('Hello Python!', 'Python is pretty cool', py)
>>> db.session.add(py)
>>> db.session.add(p)
>>> py.posts.all()
[<Post 'Hello Python!'>]

然后我添加一个新类别:

>>> ru = Category('Ruby')
>>> db.session.add(ru)

如何构建查询来获取包含帖子的所有类别。

想做这样的事情，但使用 SQLAlchemy 过滤器:

>>> categories_with_post = []
>>> For c in Category.query.all():
...     if len(c.posts.all()) > 0:
...         categories_with_post.append(c) 

Answer 1

由于您想要的只是包含帖子的类别，因此您可以尝试如下操作:

from sqlalchemy import distinct
db.session.query(distinct(Category.id), Category).join(Post.category).all()

加入将阻止返回任何没有帖子的类别。

注意distinct函数的使用，当多个帖子指向同一类别时，它将防止类别重复。但是它不能用于 Category 类本身，因此我在 Category.id 属性中使用它，并添加 Category 作为第二个参数来获取整个对象。它将返回 (id, Category) 元组列表。