在Python中的SELECT语句中传递列名

Question

    if count == 1:
        cursor.execute("SELECT * FROM PacketManager WHERE ? = ?", filters[0], parameters[0])
        all_rows = cursor.fetchall()

    elif count == 2:
        cursor.execute("SELECT * FROM PacketManager WHERE ? = ? AND ? = ?", filters[0], parameters[0], filters[1], parameters[1])
        all_rows = cursor.fetchall()

    elif count == 3 :
        cursor.execute("SELECT * FROM PacketManager WHERE ? = ? AND ? = ? AND ? = ?", filters[0], parameters[0], filters[1], parameters[1], filters[2], parameters[2])
        all_rows = cursor.fetchall()

这是我程序中的代码片段。我打算做的是在查询中传递列名和参数。

过滤器数组包含列名，参数数组包含参数。计数是用户设置的过滤器数。过滤器和参数数组已经准备就绪，没有问题。我只需要将其传递给查询即可执行。这给我一个错误“
TypeError：函数最多接受2个参数”

Answer 1

您只能使用设置参数?，不能使用表或列名称。

您可以使用预定义的查询来构建字典。

queries = {
    "foo": "SELECT * FROM PacketManager WHERE foo = ?",
    "bar": "SELECT * FROM PacketManager WHERE bar = ?",
    "foo_bar": "SELECT * FROM PacketManager WHERE foo = ? AND bar = ?",
}

# count == 1
cursor.execute(queries[filters[0], parameters[0])

# count == 2
cursor.execute(queries[filters[0] + "_" + queries[filters[1], parameters[0])

这种方法将使您避免从中进行SQL注入filters[0]。