如何返回sqlalchemy查询中相关实体的数量

问题描述:

我是sqlalchemy的新手,尽管文档看起来相当透彻,但我找不到完全符合我需要的方式。如何返回sqlalchemy查询中相关实体的数量

说我有两个表:论坛和帖子。每个论坛都有一个父母论坛和任意数量的帖子。我要的是:

  • 的顶级论坛列表
  • 热切加载的子论坛,访问过的顶级论坛
  • 每个孩子的帖子的计数论坛

所以我开始:

query(Forum).filter(Forum.parent==None).all() 

这给了我所有的顶级论坛。当然访问子论坛产生n个选择查询。

query(Forum).options(eagerload('children')).filter(Forum.parent==None).all() 

这解决了n选择问题。现在

我最好的猜测是这样的:

query(Forum, func.count(Forum.children.posts)).options(eagerload('children')).filter(Forum.parent==None).group_by(Forum.children.id).all() 

但我得到的是:

AttributeError: Neither 'InstrumentedAttribute' object nor 'Comparator' object has an attribute 'posts' 

我已经尝试了一些变化,但没有得到任何进一步。只是为了清楚起见,我在寻找这个SQL相当于:

select Forum.*, Child.*, count(Post.id) 
from Forum 
left join Forum Child on Child.parent = Forum.id 
left join Message on Message.forum = Child.id 
where Forum.parent is null 
group by Child.id 

因为你想要的岗位数要对孩子论坛访问的对象,您需要将其声明为列属性设置的时映射器。列财产申报应该是这个样子(假设你使用的声明):

Forum.post_count = column_property(select([func.count()], 
     Message.__table__.c.forum == Forum.__table__.c.id 
    ).correlate(Forum.__table__).as_scalar().label('post_count'), 
    deferred=True) 

然后你可以短语查询是这样的:

query(Forum).filter_by(parent=None).options(
    eagerload('children'), 
    undefer('children.post_count')) 

另一种选择是单独选择孩子和计数。在这种情况下,您需要自行完成结果分组:

ChildForum = aliased(Forum) 
q = (query(Forum, ChildForum, func.count(Message.id)) 
     .filter(Forum.parent == None) 
     .outerjoin((ChildForum, Forum.children)) 
     .outerjoin(ChildForum.posts) 
     .group_by(Forum, ChildForum) 
    ) 

from itertools import groupby 
from operator import attrgetter 

for forum, childforums in groupby(q, key=attrgetter('Node')): 
    for _, child, post_count in childforums: 
     if child is None: 
      # No children 
      break 
     # do something with child