Python sqlite参数扩展问题

Question

我有一个包含三列的表格，cell、trx 和 type。 这是我尝试运行的查询：

db.execute("SELECT cell,trx FROM tchdrop").fetchall()

它给出了正确的输出。

但是，当我尝试 a = ("cell", "trx") 然后

db.execute("SELECT ?,? FROM tchdrop", t).fetchall()

输出为 [(u'cell', u'trx'), (u'cell', u 'trx')] （这是错误的）

我这样做是为了弄清楚如何动态提取列，这是一个更大问题的一部分。

Answer 1

python DB-API（如sqlite3）的占位符（？）不支持传递列名称，因此您必须使用像这样的python字符串格式：

a = ("cell", "trx")

query = "SELECT {0},{1} FROM tchdrop".format(*a)
db.execute(query)

编辑：

如果您不知道列的长度，你想通过，你可以这样做：

a = ("cell", "trx", "foo", "bar")
a = ", ".join(a)

query = "SELECT {0} FROM tchdrop".format(a)
# OUTPUT : 'SELECT cell, trx, foo, bar FROM tchdrop'
db.execute(query)

Answer 2

该库将指定的值 ("cell", "trx") 替换为其引用的 SQL 等效值，因此您得到的是 SELECT "cell", "trx" FROM tchdrop 。结果是正确的。

使用 ? 语法无法实现您想要实现的目标。相反，您自己更换字符串。您可以使用正则表达式（例如 ^[a-zA-Z_]$）检查列名称以提高安全性。

例如：

columns = ",".join(("cell", "trx"))
db.execute("SELECT %s FROM tchdrop" % columns).fetchall()