按联合 SQLAlchemy 查询中的标记列进行过滤

Question

实现标题几乎等同于如何在 sqlalchemy 过滤器中使用带标签的列？我认为这是一个单独的问题。

为了大大简化我的问题：假设我有两个查询，每个查询都返回我想要的结果的子集，我将它们联合在一起：

initial_task = "initial_task"
scheduled_task = "scheduled_task"

initial = session.query(Task.task_id,
                        User.signup_date.label('due_date'),
                        literal(initial_task).label('type'))\
                 .join(Task.user)

schedule = session.query(Task.task_id,
                        Schedule.due_date.label('due_date'),
                        literal(scheduled_task).label('type'))\
                  .join(Task.schedule)

tasks = initial.union_all(schedule)

需要明确的是：我意识到这个例子可以在没有联合的情况下重写；我的实际用例有五个单独的查询，除了结果强制转换为这种正常格式之外，几乎没有任何共同点。

如何过滤tasks以仅包含 2017 年 4 月 1 日之后到期的任务？从概念上讲，类似于：

tasks.filter(tasks.c.due_date >= datetime(2017, 4, 1))

主要问题是我不知道如何以due_date一般方式引用该专栏。我从文档中尝试的所有内容似乎都在谈论较低级别的 API，并且在 ORM 层上导致：

'Query' object has no attribute 'c'

Answer 1

正如您所注意到的，该Query.union_all()方法生成一个新实例，该实例与构造生成的Query实例有点不同。您可以使用过滤查询：CompoundSelectsql.expression.union_all()literal_column()

In [18]: tasks.filter(literal_column('due_date') >= datetime(2017, 4, 1))
Out[18]: <sqlalchemy.orm.query.Query at 0x7f1d2e191b38>

In [19]: print(_)
SELECT anon_1.task_id AS anon_1_task_id, anon_1.due_date AS anon_1_due_date, anon_1.type AS anon_1_type 
FROM (SELECT task.id AS task_id, user.signup_date AS due_date, ? AS type 
FROM task JOIN user ON task.id = user.task_id UNION ALL SELECT task.id AS task_id, schedule.due_date AS due_date, ? AS type 
FROM task JOIN schedule ON task.id = schedule.task_id) AS anon_1 
WHERE due_date >= ?

另一方面，您可以在各自的日期列上单独过滤联合的各个部分。

最后，如果这更接近您的实际目标（被简化的示例隐藏），您可以将联合包装在子查询中：

In [26]: tasks_sq = tasks.subquery()

In [27]: session.query(tasks_sq).\
    ...:     filter(tasks_sq.c.due_date >= datetime(2017, 4, 1))
Out[27]: <sqlalchemy.orm.query.Query at 0x7f1d2e1d4828>

In [28]: print(_)
SELECT anon_1.task_id AS anon_1_task_id, anon_1.due_date AS anon_1_due_date, anon_1.type AS anon_1_type 
FROM (SELECT anon_2.task_id AS task_id, anon_2.due_date AS due_date, anon_2.type AS type 
FROM (SELECT task.id AS task_id, user.signup_date AS due_date, ? AS type 
FROM task JOIN user ON task.id = user.task_id UNION ALL SELECT task.id AS task_id, schedule.due_date AS due_date, ? AS type 
FROM task JOIN schedule ON task.id = schedule.task_id) AS anon_2) AS anon_1 
WHERE anon_1.due_date >= ?