如何在Python中参数化PostgreSQL的CREATE TABLE函数？

问题分析 #

1. 表名语法错误：首次报错syntax error at or near "'test_node_3'"，原因是直接用普通字符串格式化传递表名时，会给表名添加单引号，但PostgreSQL的表名属于标识符，不需要单引号。使用psycopg2.sql.Identifier处理表名是正确的解决方案，但需确保代码逻辑无其他问题。
2. 递归深度超出错误：RecursionError: maximum recursion depth exceeded通常由代码缩进错误导致逻辑混乱引发，你已确认setparams()正常，因此重点排查create_table函数的缩进和变量作用域即可。

修正后的完整代码 #

import psycopg2
from psycopg2 import sql

def create_table(table_name):
    """在PostgreSQL数据库中创建表"""
    # 使用sql.SQL和sql.Identifier安全构造动态SQL
    composed_str = sql.SQL(
        """CREATE TABLE {}
        (
            id SERIAL PRIMARY KEY,
            col1 decimal,
            col2 decimal
        )
        """).format(sql.Identifier(table_name))

    conn = None
    cur = None  # 提前初始化游标，避免finally块引用未定义变量
    try:
        # 读取数据库配置
        params = setparams()
        # 连接数据库
        conn = psycopg2.connect(**params)
        # 创建游标
        cur = conn.cursor()
        # 执行建表语句
        cur.execute(composed_str)
        # 提交事务
        conn.commit()
    except (Exception, psycopg2.DatabaseError) as error:
        print(error)
        # 出错时回滚事务
        if conn is not None:
            conn.rollback()
    finally:
        # 确保游标和连接被正确关闭
        if cur is not None:
            cur.close()
        if conn is not None:
            conn.close()

if __name__ == '__main__':
    create_table("test_table_1")

关键修正点 #

• 缩进修复：统一调整try/except/finally块及内部代码的缩进，保证逻辑执行顺序正确。
• 标识符处理：保留sql.Identifier(table_name)来安全传递表名，避免单引号语法错误。
• 语法细节修正：移除建表语句中最后一列col2 decimal后的逗号，PostgreSQL不允许列定义末尾加逗号。
• 资源安全释放：提前初始化游标变量，在finally块中统一关闭游标和连接，避免资源泄漏；出错时添加事务回滚，保证数据一致性。
• 注释修正：将错误的INSERT注释改为CREATE TABLE，保持代码注释与实际逻辑一致。

内容的提问来源于stack exchange，提问作者Kyle Sponable