如果记录具有与之关联的外键对象,则在SQLAlchemy对象中过滤记录

时间:2015-09-21 19:03:35

标签: python flask sqlalchemy flask-sqlalchemy

使用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) 

1 个答案:

答案 0 :(得分:2)

由于你想要的只是拥有帖子的类别,你可以尝试这样的事情:

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

联接将阻止任何没有帖子的类别被返回。

注意使用distinct函数,只要多个帖子指向同一类别,就会阻止重复类别。但是它不能在Category类本身上使用,所以我在Category.id属性中使用它并添加Category作为第二个参数来获取整个对象。它将返回(id,Category)元组列表。