如何使用SQL Alchemy与pg8000指定PostgreSQL搜索路径？

pg8000驱动不支持psycopg2那样的options参数方式，你可以用下面两种方法解决：

方法一：利用SQLAlchemy连接事件（通用方案） #

通过监听连接建立事件，每次新连接创建后自动执行SET search_path语句，这种方式对所有PostgreSQL驱动都适用：

from sqlalchemy import create_engine, event

# 创建引擎
engine = create_engine('postgresql+pg8000://dbuser@dbhost:5432/dbname')

# 监听连接建立事件，设置搜索路径
@event.listens_for(engine, 'connect')
def set_search_path(dbapi_connection, connection_record):
    cursor = dbapi_connection.cursor()
    # 注意：如果dbschema包含特殊字符，建议用参数化避免SQL注入
    cursor.execute("SET search_path TO %s", (dbschema,))
    cursor.close()

提示：如果你的schema名称包含特殊字符（比如空格、大写字母），一定要用参数化的方式执行SQL，避免注入风险和语法错误。

方法二：pg8000原生参数（简洁方案） #

pg8000驱动本身支持直接在连接参数中指定search_path，可以直接传入connect_args：

from sqlalchemy import create_engine

engine = create_engine(
    'postgresql+pg8000://dbuser@dbhost:5432/dbname',
    connect_args={'search_path': dbschema}
)

这种方式不需要额外的事件监听，直接通过驱动原生参数完成设置，更简洁。

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