SQL 标识符替换 - 使用列名列表

Question

我正在尝试运行表单的查询：

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"，这显然是不正确的。

Answer 1

你需要使用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')])