在 Python 中区分变量名和数据库字段名

Question

代码

import psycopg2
import psycopg2.extras
def store_values_to_pg94(file_size, connection): 
    cursor = connection.cursor()
    cursor.execute(
        INSERT INTO measurements
        (file_size) 
        VALUES (file_size)

其中 PostgreSQL 9.4 中列的字段名称是 file_size 但在 Python 中变量名称也是 file_size。

如何避免这种冲突？

Answer 1

您示例中的代码无法编译，即它不是有效的 python。 cursor.execute() 接受一个字符串和一组可选参数，这些参数根据数据库模块的paramstyle 扩展为字符串。

事实上，当使用DBAPI时，你的问题甚至不会浮出水面，因为SQL语法和python语法是完全分开的。

如果你使用psycopg2访问PostgreSQL，paramstyle就是pyformat，因此

cursor.execute(
    'INSERT INTO measurements (file_size) VALUES (%(file_size)s);',
    file_size=file_size # pass variable file_size as named parameter file_size
);

如您所见，您指定 SQL 查询并为要从 python 代码传入的值保留占位符。

如果paramstyle == 'pyformat'，也可以使用位置参数：

cursor.execute(
    'INSERT INTO measurements (file_size) VALUES (%s);',
    file_size # pass variable file_size as positional parameter
);

请注意，不同的数据库访问模块实现不同的paramstyle，需要不同的参数占位符（或合适的包装器）。

所有参数替换实现都应根据数据库后端的规则正确转义其参数，这是一个很好且重要的功能，因为您不必手动转义数据。