内爆列表以在 python MySQLDB IN 子句中使用

Question

我知道如何将列表映射到字符串：

foostring = ",".join( map(str, list_of_ids) )

而且我知道我可以使用以下方法将该字符串放入 IN 子句中：

cursor.execute("DELETE FROM foo.bar WHERE baz IN ('%s')" % (foostring))

我需要的是使用 MySQLDB 安全地完成同样的事情（避免 SQL 注入）。在上面的示例中，因为 foostring 没有作为参数传递给执行，所以它很容易受到攻击。我还必须在 mysql 库之外引用和转义。

（有一个related SO question，但那里列出的答案要么不适用于 MySQLDB，要么容易受到 SQL 注入的攻击。）

Answer 1

直接使用list_of_ids：

format_strings = ','.join(['%s'] * len(list_of_ids))
cursor.execute("DELETE FROM foo.bar WHERE baz IN (%s)" % format_strings,
                tuple(list_of_ids))

这样就避免了自己引用，也避免了各种sql注入。

请注意，数据（list_of_ids）作为参数（不在查询文本中）直接进入 mysql 的驱动程序，因此没有注入。您可以在字符串中保留任何您想要的字符，无需删除或引用字符。

Answer 2

list_of_ids = [ 1, 2, 3]
query = "select * from table where x in %s" % str(tuple(list_of_ids))
print query

如果您不希望关心必须传递参数以完成查询字符串的方法并且只想调用cursror.execute(query)，这可能适用于某些用例。

另一种可能是：

"select * from table where x in (%s)" % ', '.join(str(id) for id in list_of_ids)

Answer 3

很简单：只需使用下面的格式

rules_id = ["9","10"]

sql1 = "SELECT * FROM admission_rules_staff WHERE id in("+", ".join(map(str, rules_id))+")"

", ".join(map(str, rules_id))

Answer 4

如果你使用Django 2.0 or 2.1和Python 3.6，这是正确的方式：

from django.db import connection
RESULT_COLS = ['col1', 'col2', 'col3']
RESULT_COLS_STR = ', '.join(['a.'+'`'+i+'`' for i in RESULT_COLS])
QUERY_INDEX = RESULT_COLS[0]

TABLE_NAME = 'test'
search_value = ['ab', 'cd', 'ef']  # <-- a list
query = (
    f'SELECT DISTINCT {RESULT_COLS_STR} FROM {TABLE_NAME} a '
    f'WHERE a.`{RESULT_COLS[0]}` IN %s '
    f'ORDER BY a.`{RESULT_COLS[0]}`;'
)  # <- 'SELECT DISTINCT a.`col1`, a.`col2`, a.`col3` FROM test a WHERE a.`col1` IN %s ORDER BY a.`col1`;'
with connection.cursor() as cursor:
    cursor.execute(query, params=[search_value])  # params is a list with a list as its element

参考：https://stackoverflow.com/a/23891759/2803344 https://docs.djangoproject.com/en/2.1/topics/db/sql/#passing-parameters-into-raw

Answer 5

虽然这个问题已经很老了，但我认为最好留下回复，以防其他人正在寻找我想要的东西

当我们有很多参数或者我们想使用命名参数时，接受的答案会变得混乱

经过一番尝试

ids = [5, 3, ...]  # list of ids
cursor.execute('''
SELECT 
...
WHERE
  id IN %(ids)s
  AND created_at > %(start_dt)s
''', {
  'ids': tuple(ids), 'start_dt': '2019-10-31 00:00:00'
})

用python2.7、pymysql==0.7.11测试

Answer 6

另一个使用列表理解的简单解决方案：

# creating a new list of strings and convert to tuple
sql_list = tuple([ key.encode("UTF-8") for key in list_of_ids ])

# replace "{}" with "('id1','id2',...'idlast')"
cursor.execute("DELETE FROM foo.bar WHERE baz IN {}".format(sql_list))

Answer 7

虽然这个问题已经很老了。如果它可以帮助某人，我将分享我的解决方案。

list_to_check = ['A', 'B'] cursor.execute("DELETE FROM foo.bar WHERE baz IN ({})".format(str(list_to_check)[1:-1])

用Python=3.6测试

Answer 8

这似乎仍然是 2021 年 Python3 的问题，正如 Rubms 对 markk 答案的评论中所指出的那样。

在mysql连接器包的“cursor.py”中的“_process_params_dict”方法中添加大约9行代码来处理元组解决了我的问题：

def _process_params_dict(self, params):
    """Process query parameters given as dictionary"""
    try:
        to_mysql = self._connection.converter.to_mysql
        escape = self._connection.converter.escape
        quote = self._connection.converter.quote
        res = {}
        for key, value in list(params.items()):
            if type(value) is tuple: ### BEGIN MY ADDITIONS
                res[key.encode()] = b''
                for subvalue in value:
                    conv = subvalue
                    conv = to_mysql(conv)
                    conv = escape(conv)
                    conv = quote(conv)
                    res[key.encode()] = res[key.encode()] + b',' + conv if len(res[key.encode()]) else conv
            else: ### END MY ADDITIONS
                conv = value
                conv = to_mysql(conv)
                conv = escape(conv)
                conv = quote(conv)
                res[key.encode()] = conv
    except Exception as err:
        raise errors.ProgrammingError(
            "Failed processing pyformat-parameters; %s" % err)
    else:
        return res