SQLAlchemy 并排放置子查询而不是嵌套

Question

我有一个包含两个子查询的查询。在 SQL 中大致如下：

SELECT sq2.foo,
       sq2.bar
FROM (
      SELECT foo FROM bar WHERE foo = '123'
) AS sq1,
LATERAL (
      SELECT foo, bar FROM bar WHERE s1.foo = 'zomething'
) AS sq2
WHERE sq2.foo = 'something';

这就是我需要的 SQLAlchemy。当我把它翻译成炼金术时，我大致得到了这个：

sq1 = session.query(Bar.foo).filter(Bar.foo == '123').subquery()
sq2 = session.query(Bar.foo).filter(sq1.foo == 'zomething').subquery().lateral()
session.query(sq2.foo, sq2.bar).filter(sq.foo = 'something').all()

但是，这不会转化为与顶级 SQL 相同的 SQL，而是像这样嵌套查询：

SELECT sq2.foo,
       sq2.bar
FROM LATERAL (
    SELECT foo, bar FROM (
        SELECT foo FROM bar WHERE foo = '123'
    ) AS sq1,
    bar
    WHERE sq1.foo = 'zomething'
) AS sq2
WHERE sq2.foo = 'something';

这不会产生相同的结果，原因可能过于复杂，无法根据这个简化的示例进行解释。但是，如果我将最终查询更改为包含 sq1 中我并不真正需要的内容，SQLAlchemy 确实会做正确的事情，如下所示：

session.query(sq2.foo, sq2.bar, sq1.foo).filter(sq.foo = 'something').all()

现在它返回具有正确结构的查询。有没有办法在不包含无价值数据的情况下强制并排子查询？

Answer 1

你需要使用select_from

sq1 = session.query(Bar.foo).filter(Bar.foo == '123').subquery().alias('sq1')
sq2 = session.query(Bar.foo, Bar.bar).filter(sq1.c.foo == 'zomething').subquery().lateral('sq2')
q = session.query(sq2.c.foo, sq2.c.bar).select_from(sq1, sq2).filter(sq2.c.foo == 'something')

# >> print(q)
# SELECT sq2.foo AS sq2_foo, sq2.bar AS sq2_bar 
# FROM LATERAL (SELECT bar.foo AS foo, bar.bar AS bar 
# FROM bar 
# WHERE sq1.foo = %(foo_1)s) AS sq2, (SELECT bar.foo AS foo 
# FROM bar 
# WHERE bar.foo = %(foo_2)s) AS sq1
# WHERE sq2.foo = %(foo_3)s