如何在SQLAlchemy结合psycopg2环境中使用SQL存储过程查询动态表？

嘿，我一眼就揪出问题所在了！你写的存储过程是SQL Server那套语法，但你用的是PostgreSQL（psycopg2是PostgreSQL专属的Python驱动），PostgreSQL根本不认识@参数和GO命令，这就是语法报错的直接原因。

咱们一步步来解决这个动态查表的需求：

方案1：直接构造安全的动态SQL（推荐） #

其实你完全没必要创建存储过程——psycopg2自带了psycopg2.sql模块，专门用来处理动态标识符（比如表名、列名），它会自动帮你转义特殊字符，比直接用f-string拼接安全多了，还能避免SQL注入风险。修改后的代码如下：

import psycopg2
import sqlalchemy.pool as pool
from psycopg2 import sql

def get_conn_pool():
    conn = psycopg2.connect(
        user=settings.POSTGRES_USER,
        password=settings.POSTGRES_PASSWORD,
        database=settings.POSTGRES_DB,
        host=settings.POSTGRES_HOST,
        port=settings.POSTGRES_PORT
    )
    return conn

db_pool = pool.QueuePool(get_conn_pool, max_overflow=10, pool_size=5)

conn = db_pool.connect()
cursor = conn.cursor()

tables = ['current_block', 'tasks']

for table in tables:
    # 用sql.Identifier安全包装表名，自动处理特殊字符转义
    query = sql.SQL("SELECT * FROM {}").format(sql.Identifier(table))
    cursor.execute(query)
    result = cursor.fetchall()
    print(result)

# 用完记得手动关闭资源
cursor.close()
conn.close()

方案2：用PostgreSQL风格的函数封装（如果一定要用存储逻辑） #

如果你确实需要用数据库端的封装逻辑，得改成PostgreSQL支持的plpgsql语法——注意PostgreSQL里用函数返回结果更方便，存储过程默认不返回查询结果：

第一步：先在PostgreSQL中创建函数

执行这段SQL语句：

CREATE OR REPLACE FUNCTION GetTableData(p_tablename text)
RETURNS SETOF record
LANGUAGE plpgsql
AS $$
BEGIN
    -- 用format和%I来安全转义表名，防止注入
    RETURN QUERY EXECUTE format('SELECT * FROM %I', p_tablename);
END;
$$;

第二步：在Python中调用这个函数

还是用psycopg2.sql模块来安全传递参数：

for table in tables:
    cursor.execute(sql.SQL("SELECT * FROM GetTableData({})").format(sql.Literal(table)))
    result = cursor.fetchall()
    print(result)

再补一句原代码的问题点 #

你原代码里的CREATE PROCEDURE GetTableData @TableName nvarchar(30) AS ... GO是纯SQL Server语法：

• PostgreSQL用p_param或$1这种方式定义参数，不是@参数
• GO是SQL Server的批处理分隔符，PostgreSQL完全不认识
• 而且PostgreSQL里不能直接用参数代替表名，必须通过动态SQL转义标识符

备注：内容来源于stack exchange，提问作者Vadim