Question

我正在使用Python脚本将数据导入MySQL数据库。我使用MySQLdb来执行我的查询。

我还使用一个函数来设置查询及其带有命名参数的数据，以便在请求中自动替换，例如：

req = "SELECT * FROM sample WHERE `id` = %(id)s and `param1` = %(param1)s"
data = {'id': 15, 'param1': None}
cursor.execute(req, data)

我非常喜欢这种方式，因为我使用相同的数据dict进行3次或4次查询。

问题是，当我尝试使用WHERE值生成None条件时，它确实将None替换为NULL，但我希望它能够将`param1` = NULL替换为`param1` IS NULL，以便条件的计算结果为真。

有没有办法直接解决这个问题？或者我可以只使用参数替换（不执行查询），然后自己替换（= NULL到IS NULL），然后执行查询。

Answer 1

您可以执行以下操作。

def query(params):
    # Make sure it's hardcoded and malicious user can't overwrite it
    param_whitelist = 'id', 'param1'

    sql = '''
        SELECT * 
        FROM sample 
        WHERE {placeholders}
    '''
    placeholders = ' AND '.join(
        '`{field}` {op} %s'.format(field=k, op='IS' if params[k] is None else '=') 
        for k in param_whitelist)
    sql = sql.format(placeholders=placeholders)

    cursor.execute(sql, params)
    return cursor


data = {'id': 15, 'param1': None}
print([r for r in query(data)])

Python MySQLdb查询参数＆＃34; IS NULL＆＃34;而不是＆＃34; = NULL＆＃34;

1 个答案: