如何使用Flask-Sqlalchemy筛选一对多关系的模型?



我是Flask-SQL-Alchemy的新手,所以这可能是一个新手问题。假设我有一个Tweet模型:

class Tweet(db.Model):
__tablename__ = 'tweet'
id = db.Column(db.Integer, primary_key=True)
text = db.Column(db.String(500), nullable=False)
user_id = db.Column(db.Integer, db.ForeignKey('user.id'), nullable=False)

和一个User模型,它可以有很多tweets:

class User(db.Model):
__tablename__ = 'user'
id = db.Column(db.Integer, primary_key=True)
first_name = db.Column(db.String(50), nullable=False)
last_name = db.Column(db.String(50), nullable=False)
birth_year = db.Column(db.Integer)
tweets = db.relationship('Tweet', backref='user', lazy=True)

现在,我想在User表上应用不同的过滤器。例如,要获取所有出生在1970且名字为John的用户

filters = (
User.first_name == 'John',
User.birth_year == 1970
)
users = User.query.filter(*filters).all()

到目前为止,一切顺利!但现在我想为tweets属性添加一个过滤器,注意不是一个真实的数据库列,而是由ORM提供的东西。如何获取发布了超过20条推文的John?我试着:

filters = (
User.first_name == 'John',
len(User.tweets) > 20
)

但是这会引发一个异常:object of type 'InstrumentedAttribute' has no len()。我还尝试在用户模型中添加一个新的混合属性:

class User(db.Model):
...
@hybrid_property
def tweet_count(self):
return len(self.tweets)
filters = (
User.first_name == 'John',
User.tweet_count > 20
)

但是我仍然得到相同的错误。这似乎是一个常见的任务,但我无法找到任何文档或相关的例子。我处理问题的方式是否错误?

我想你错过了这里看到的第二部分SQLAlchemy -为子数编写混合方法


from sqlalchemy.sql import select, func
class User(db.Model):
#...
@tweet_count.expression
def tweet_count(cls):
return (select([func.count(Tweet.id)]).
where(Tweet.user_id == cls.id).
label("tweet_count")
)

最新更新