我正在尝试运行以下形式的查询:
SELECT {} from TABLE where foo = VALUE
但我希望能够提供一个列表来替换 {}
根据 psycopg 文档,为了安全地执行此操作,您需要使用sql.Identifier
函数,以便正确转义参数,然后执行以下操作:
SQL = sql.SQL(
"SELECT {} FROM foo WHERE bar = %s"
).format(identifier)
cursor.execute(SQL, [VALUE])
当identifier
是单个元素时,这有效,但我需要它是一个任意数字。例如,如果:
identifier = ["abc", "def"]
和
VALUE = 4
SQL = SELECT abc def FROM foo WHERE bar = 4
我尝试为identifier
的每个成员运行sql.Identifier(x)
,但这给了"abc""def"
,这显然是不正确的。
你需要使用 sql.join((,让它像列表推导一样工作:
from psycopg2 import sql
cols = ["abc", "def"]
query = sql.SQL(
"select {0} from {1} where abc = %s").format(
sql.SQL(', ').join([sql.Identifier(c) for c in cols]),
sql.Identifier('foo')
)
cur = conn.cursor()
print(query)
print(cur.mogrify(query, ('1', )))
cur.execute(query, ('1', ))
print (cur.rowcount, cur.fetchall())
输出:
Composed([SQL('select '), Composed([Identifier('abc'), SQL(', '), Identifier('def')]), SQL(' from '), Identifier('foo'), SQL(' where abc = %s')])
select "abc", "def" from "foo" where abc = '1'
(1, [('1', '2')])