加入 SQLAlchemy 中的 CTE

Question

我正在尝试制定一个 SQLAlchemy 查询，该查询使用 CTE 来构建元组输入列表的类似表的结构，并将它与我的一个表（后端数据库是 Postgres）连接起来。 从概念上讲，它看起来像：

WITH to_compare AS (
  SELECT * FROM (
    VALUES 
     (1, 'flimflam'),
     (2, 'fimblefamble'),
     (3, 'pigglywiggly'),
     (4, 'beepboop')
     -- repeat for a couple dozen or hundred rows
  ) AS t (field1, field2)
)
SELECT b.field1, b.field2, b.field3
FROM my_model b
JOIN to_compare c ON (c.field1 = b.field1) AND (c.field2 = b.field2)

我们的目标是看到什么field3的对(field1, field2)的表，如果是这样，对于一个中等大小的列表(field1, field2)对。

在 SQLAlchemy 中，我试图这样做：

stmts = [
    sa.select(
        [
            sa.cast(sa.literal(field1), sa.Integer).label("field1"),
            sa.cast(sa.literal(field2), sa.Text).label("field2"),
        ]
    )
    if idx == 0
    else sa.select([sa.literal(field1), sa.literal(field2)])
    for idx, (field1, field2) in enumerate(list_of_tuples)
]
cte = sa.union_all(*stmts).cte(name="temporary_table")

already_in_db_query = db.session.query(MyModel)\
    .join(cte,
          cte.c.field1 == MyModel.field1,
          cte.c.field2 == MyModel.field2,
    ).all()

但似乎 CTE 和 JOIN 不能很好地协同工作：错误出在join ，说：

sqlalchemy.exc.InvalidRequestError: Don't know how to join to ; please use an ON clause to more clearly establish the left side of this join

如果我尝试打印 cte，它看起来确实像一个非 SQL 实体：

$ from pprint import pformat
$ print(pformat(str(cte)), flush=True)
> ''

有没有办法做到这一点？ 或者有更好的方法来实现我的目标？

Answer 1

在这种情况下， Query.join()的第二个参数应该是完整的ON子句，而是将 3 个参数传递给join() 。 使用and_()组合谓词，就像在原始 SQL 中所做的那样：

already_in_db_query = db.session.query(MyModel)\
    .join(cte,
          and_(cte.c.field1 == MyModel.field1,
               cte.c.field2 == MyModel.field2),
    ).all()