使用Python调用PostgreSQL存储过程时遇错误无法排查

问题分析与修复点 #

• 错误1：callproc参数格式错误
  cursor.callproc()仅需传入存储过程名称，不需要添加call 前缀。原代码中procname = "call proc_test"的写法不符合该方法的要求，应改为直接传入存储过程名。

• 错误2：PostgreSQL存储过程调用兼容性问题
  psycopg2的callproc方法对PostgreSQL存储过程的支持有限，尤其是涉及结果集返回时，更推荐直接使用cursor.execute("CALL proc_test();")的方式调用。

• 错误3：资源管理不规范
  手动关闭游标和连接容易因异常导致资源泄漏，建议使用with语句自动管理连接与游标，无需手动执行关闭操作。

修复后的完整代码 #

import psycopg2
from psycopg2 import OperationalError

def create_connection():
    # 替换为你的实际数据库连接参数
    try:
        conn = psycopg2.connect(
            dbname="你的数据库名",
            user="你的用户名",
            password="你的密码",
            host="localhost",
            port="5432"
        )
        return conn
    except OperationalError as e:
        print(f"数据库连接失败: {e}")
        return None

def call_procedure_without_arguments():
    try:
        with create_connection() as connection:
            if not connection:
                print("连接失败，无法调用存储过程。")
                return
            
            with connection.cursor() as cursor:
                # 调用无参存储过程
                cursor.execute("CALL proc_test();")
                # 若存储过程返回结果集，执行以下语句获取结果
                results = cursor.fetchall()
                print("Results:", results)

                # 存储过程涉及数据修改时需提交事务
                connection.commit()

            print("存储过程执行成功。")
    except Exception as e:
        print("错误：无法调用存储过程。", e)
        # 异常时回滚事务，避免数据不一致
        if 'connection' in locals() and connection:
            connection.rollback()

if __name__ == "__main__":
    call_procedure_without_arguments()

补充说明 #

• 若存储过程无结果集返回，可删除results = cursor.fetchall()语句。
• 确保create_connection函数中的数据库参数与实际环境匹配。
• 存储过程包含写操作时，必须在执行成功后提交事务；异常时务必回滚，防止数据异常。

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