使psycopg2将参数化查询传递给PostgreSQL的最佳方法是什么?我不想编写自己的escpaing机制或适配器,并且很难在Web浏览器中阅读psycopg2源代码和示例。
如果我需要切换到类似PyGreSQL或其他python pg适配器的东西,那对我来说没关系。我只想简单的参数化。
答案 0 :(得分:73)
psycopg2
遵循DB-API 2.0的规则(在PEP-249中设置)。这意味着您可以从execute
对象调用cursor
方法并使用pyformat
绑定样式,它将为您执行转义。例如,以下 应该是安全的(和工作):
cursor.execute("SELECT * FROM student WHERE last_name = %(lname)s",
{"lname": "Robert'); DROP TABLE students;--"})
答案 1 :(得分:16)
来自psycopg文档
(http://initd.org/psycopg/docs/usage.html)
警告永远不要,永远不要使用Python字符串连接(+)或字符串参数插值(%)将变量传递给SQL查询字符串。甚至在枪口下也没有。
在SQL命令中传递变量的正确方法是使用execute()方法的第二个参数:
SQL = "INSERT INTO authors (name) VALUES (%s);" # Note: no quotes
data = ("O'Reilly", )
cur.execute(SQL, data) # Note: no % operator
答案 2 :(得分:12)
以下是一些您可能会感兴趣的示例
cursor.execute('SELECT * from table where id = %(some_id)d', {'some_id': 1234})
或者您可以根据字段名称,值:
的字典动态构建查询fields = ', '.join(my_dict.keys())
values = ', '.join(['%%(%s)s' % x for x in my_dict])
query = 'INSERT INTO some_table (%s) VALUES (%s)' % (fields, values)
cursor.execute(query, my_dict)
注意:必须在代码中定义字段,不用户输入,否则您将容易受到SQL注入。