SQLAlchemy:以3个表的union_all子查询为前缀的列名

时间:2014-10-16 13:03:33

标签: python sql sqlalchemy flask-sqlalchemy

这是我的mssql代码段

cnt = func.count(pvr_svc.ic_idn).label('cnt')

x = session.query(pvr_svc.inc_type_md, cnt, cast(pvr_svc.crt_dt,DATE)
         .label('crt_dt'))
         .filter(pvr_svc.inc_type_md.in_(['PM','OM','OP-HU']))
         .group_by(cast(pvr_svc.crt_dt, DATE), pvr_svc.inc_type_md)

y = session.query(pvr_svc.inc_type_md, cnt, cast(pvr_svc.crt_dt,DATE)
         .label('crt_dt'))
         .filter(pvr_svc.gold_idn==2)
         .group_by(cast(pvr_svc.crt_dt, DATE), pvr_svc.inc_type_md)

这就是我要做的是

from sqlalchemy import union_all

u1 = x.union_all(y)     # ----- 1

“u1”中的列名提取如下

 >>>[i['name'] for i in u1.column_descriptions]  

 >>>['inc_type_md', 'cnt', 'crt_dt']   # column names

现在如果我想在将来使用'u1'这就是我的工作

>>>v1 = u1.subquery()    #------ 2

从“v1”访问列名,我这样做

>>>v1.c.keys()

>>>[u'pvr_svc_inc_type_md', u'cnt', u'crt_dt']

现在,如果你看到'u1'中的第一个键和'v1'(它是u1的子查询)是不同的。我期待它们是一样的。

为避免这种情况,我会在“x”和“y”查询中标记列名称

x = session.query(pvr_svc.inc_type_md.label('inc_type_md'), cnt, cast(pvr_svc.crt_dt,DATE).label('crt_dt')
         .label('crt_dt'))
         .filter(pvr_svc.inc_type_md.in_(['PM','OM','OP-HU']))
         .group_by(cast(pvr_svc.crt_dt, DATE), pvr_svc.inc_type_md)

y = session.query(pvr_svc.inc_type_md.label('inc_type_md'), cnt, cast(pvr_svc.crt_dt,DATE).label('crt_dt')
         .label('crt_dt'))
         .filter(pvr_svc.gold_idn==2)
         .group_by(cast(pvr_svc.crt_dt, DATE), pvr_svc.inc_type_md)

并重复步骤1和2,它工作正常。

现在我的问题是

我想用第三个查询“z”做一个“u1”的union_all,我在“z”中标记了列名

z = session.query(pvr_svc.inc_type_md.label('inc_type_md'), cnt, cast(pvr_svc.crt_dt,DATE).label('crt_dt')
         .label('crt_dt'))
         .filter(pvr_svc.gold_idn==4)
         .group_by(cast(pvr_svc.crt_dt, DATE), pvr_svc.inc_type_md)

我会这样做来做3个查询的union_all

>>>union_xyz = u1.union_all(z)

现在我想在将来的查询中使用“union_xyz”,所以我创建了一个子查询

>>>sub_xyz = union_xyz.subquery()

现在的问题是,sub_xyz中的列名前缀为一些整数

所以,这就是我得到的,

>>>sub_xyz.c.keys()
>>>[u'%(2911061292 anon)s_inc_type_md', u'%(2911061292 anon)s_cnt', u'%(2911061292 anon)s_crt_dt']

如何避免列名前缀?

我浏览了sqlalchemy doc

我发现了一个类似的问题 How can I prevent sqlalchemy from prefixing the column names of a CTE?

1 个答案:

答案 0 :(得分:2)

我不知道有什么方法可以阻止前缀,但您可以将子查询的c属性转换为列表或元组,然后按位置访问各列。这里有一个使用元组解包的例子:

>>> inc_type_md_col, cnt_col, ctr_dt_col = tuple(subq_xyz.c)

inc_type_md_colcnt_colctr_dt_col是有效的列对象,您可以随意使用它们,例如

>>> session.query(inc_type_md_col).filter(cnt_col > 0)