使用 Sqlalchemy 连接和 Postgres 从原始字符串编译查询（不使用 .text(...)）

Posted 2023-02-24

【中文标题】使用 Sqlalchemy 连接和 Postgres 从原始字符串编译查询（不使用 .text(...)）

【英文标题】：Compile query from raw string (without using .text(...)) using Sqlalchemy connection and Postgres

【发布时间】：2019-10-10 06:04:57

【问题描述】：

我正在使用 Sqlalchemy 1.3 连接到 PostgreSQL 9.6 数据库（通过 Psycopg）。

我有一个非常 非常 使用 Psycopg2 语法格式化的原始 Sql 字符串，由于一些遗留问题，我无法修改：

statement_str = SELECT * FROM users WHERE user_id=%(user_id)s

注意%(user_id)s

我可以很高兴地使用 sqlalchemy 连接执行它，只需这样做：

connection = sqlalch_engine.connect()
rows = conn.execute(statement_str, user_id=self.user_id)

而且效果很好。我得到了我的用户，一切都很好。

现在，出于调试目的，我希望将 %(user_id)s 参数扩展为实际值的实际查询。例如：如果user_id = "foo"，则得到SELECT * FROM users WHERE user_id = 'foo'

我见过大量使用sqlalchemy.text(...) 生成语句然后获得编译版本的示例。这要归功于其他答案，例如 this one 或 this one 在我有 SqlAlchemy 查询时能够产生像样的 str。

但是，在这种特殊情况下，因为我使用的是更特定于光标的语法%(user_id)，所以我不能这样做。如果我尝试：

text(statement_str).bindparams(user_id="foo")

我得到： This text() construct doesn't define a bound parameter named 'user_id'

所以我想我正在寻找的是类似的东西

conn.compile(statement_str, user_id=self.user_id)

但我没能得到那个。

【参考方案1】：

不确定这是不是你想要的，但这里是。

假设 statement_str 实际上是一个字符串：

import sqlalchemy as sa

statement_str = "SELECT * FROM users WHERE user_id=%(user_id)s"
params = 'user_id': 'foo'

query_text = sa.text(statement_str % params)

# str(query_text) should print "select * from users where user_id=foo"

【讨论】：

感谢您的回答。问题在于text(...) 试图将%(user_id)s 转义为%%(user_id)

哦，好的。很高兴您找到了解决方案。

【参考方案2】：

好的，我认为我明白了。

SqlAlchemy 的raw_connection + Psycopg 的mogrify 的组合似乎就是答案。

conn = sqlalch_engine.raw_connection()
try:
    cursor = conn.cursor()
    s_str = cursor.mogrify(statement_str, 'user_id': self.user_id)
    s_str = s_str.decode("utf-8")  # mogrify returns bytes

    # Some cleanup for niceness:
    s_str = s_str.replace('\n', ' ')
    s_str = re.sub(r'\s2,', ' ', s_str)
finally:
    conn.close()

我希望其他人觉得这有帮助