[英]SQLAlchemy - How can I make eager loading count property
我想为包含计数的模型创建一个属性。
因为我总是需要属性,所以我想使用JOIN
进行查询,例如sqlalchemy.orm.relationship
with lazy='joined'
例如,我定义了如下的模型
import sqlalchemy as s, func
from sqlalchemy.orm import relatioship
# ...
class Foo(Base):
__tablename__ = 'foo'
id = s.Column(s.Integer, primary_key=True)
bar_id = s.Column(s.Integer, s.ForeignKey('bar.id'))
bar = relationship('Bar')
class Bar(Base):
__tablename__ = 'bar'
id = s.Column(s.Integer, primary_key=True)
@property
def foo_count(self):
return Foo.query.filter_by(bar=self).count()
当我访问属性foo_count
,它会向DBMS发送查询。
因为我总是访问这个属性,我想急切地加载它这样的计数属性
# Not session.query(Bar, func.count(Foo.id)).join(Foo) ...
bar = Bar.query.first()
SQL将是这样的
SELECT id, COUNT(Foo.id)
FROM bar
INNER JOIN foo
ON bar.id = foo.id
然后bar.foo_count
将不会发生SQL查询。
我怎样才能创建像foo_count
这样的属性?
我使用sqlalchemy.orm.column_property
解决了这个问题
我通过以下方式替换了foo_count
import sqlalchemy as s, func, select
from sqlalchemy.orm import relationship, column_property
# ...
class Foo(Base):
__tablename__ = 'foo'
id = s.Column(s.Integer, primary_key=True)
bar_id = s.Column(s.Integer, s.ForeignKey('bar.id'))
bar = relationship('Bar')
class Bar(Base):
__tablename__ = 'bar'
id = s.Column(s.Integer, primary_key=True)
foo_count = column_property(
select([func.count(Foo.id)])
.where(Foo.bar_id == id)
)
请查看Hybrid Attribute扩展。
您的对象模型将类似于以下内容:
class Foo(Base):
__tablename__ = 'foo'
id = Column(Integer, primary_key=True)
bar_id = Column(Integer, ForeignKey('bar.id'))
bar = relationship('Bar')
class Bar(Base):
__tablename__ = 'bar'
id = Column(Integer, primary_key=True)
@hybrid_property
def foo_count(self):
return object_session(self).query(Foo).filter(Foo.bar==self).count()
@foo_count.expression
def foo_count(cls):
return select([func.count(Foo.id)]).where(Foo.bar_id == cls.id).label('foo_count')
foo_count
不会被急切加载,但你可以在下面的查询中使用它(在SELECT
和WHERE
子句中:
qry = session.query(Bar, Bar.foo_count).filter(Bar.foo_count > 0)
for (bar, bar_foo_count) in qry:
print bar, bar_foo_count
如您所见,查询将仅在一个查询中返回(Bar, foo_count)
元组,现在您可以使用它执行所需的操作。
声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.