在Alembic迁移中使用PostgreSQL命名游标遇InvalidCursorName错误求助

你的猜想不正确，psycopg2的命名游标不需要预先通过SQL语句声明，报错的根源是命名游标（服务器端游标）与当前SQLAlchemy事务管理的兼容问题。以下是具体的解决方案：

方案一：改用普通游标（推荐） #

COPY操作本身已经是PostgreSQL高效导出数据的方式，不需要依赖服务器端游标。只需移除游标名称参数，使用普通客户端游标即可解决问题：

def export_data_to_csv(connection: sa.engine.base.Connection, csv_filename: str, table_name: str):
    assert connection.in_transaction()

    # 移除name参数，创建普通客户端游标
    with connection.connection.cursor() as cursor:
        with open(csv_filename, 'w', newline='') as csv_file:
            cursor.copy_expert(f"COPY (SELECT * FROM ONLY {table_name}) TO STDOUT WITH CSV HEADER", csv_file)

普通游标不需要在服务器端创建持久化的游标对象，完全适配SQLAlchemy的事务管理逻辑，同时能高效完成CSV导出任务。

方案二：手动管理服务器端游标生命周期（若必须使用） #

如果因极端大数据量需求必须使用服务器端游标，可绕过上下文管理器，手动创建和关闭游标，避免SQLAlchemy事务边界导致的游标提前销毁：

def export_data_to_csv(connection: sa.engine.base.Connection, csv_filename: str, table_name: str):
    assert connection.in_transaction()

    pg_conn = connection.connection
    # 手动创建服务器端游标
    cursor = pg_conn.cursor(name="migration_cursor")
    try:
        with open(csv_filename, 'w', newline='') as csv_file:
            cursor.copy_expert(f"COPY (SELECT * FROM ONLY {table_name}) TO STDOUT WITH CSV HEADER", csv_file)
    finally:
        # 确保游标被手动关闭
        cursor.close()

额外注意事项 #

• 若table_name为动态传入，避免SQL注入风险：建议通过SQLAlchemy的表对象获取名称（如my_table.name），而非直接拼接字符串。
• 确认导出路径具备写入权限，避免后续出现IO错误。

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