python - 在 psycopg2 中传递参数

Question

我正在尝试使用 psycopg2 访问 PostgreSQL:

sql = """
SELECT
    %s
FROM
    table;
"""

cur = con.cursor()
input = (['id', 'name'], )
cur.execute(sql, input)

data = pd.DataFrame.from_records(cur.fetchall())

但是，返回的结果是:

             0
0   [id, name]
1   [id, name]
2   [id, name]
3   [id, name]
4   [id, name]

如果我尝试访问单个列，它看起来像:

     0
0   id
1   id
2   id
3   id
4   id

列名周围的引号似乎有问题(不应存在的单引号):

In [49]: print cur.mogrify(sql, input)

SELECT
    'id'
FROM
    table;

但我正在关注文档:http://initd.org/psycopg/docs/usage.html#

谁能告诉我这是怎么回事？非常感谢!!!

Answer 1

使用 AsIs 扩展

import psycopg2
from psycopg2.extensions import AsIs

column_list = ['id','name']
columns = ', '.join(column_list)

cursor.execute("SELECT %s FROM table", (AsIs(columns),))

并且 mogrify 将显示它没有引用列名并将它们按原样传递。