如何在Python的psycopg2中用connection_factory连接指定Schema

基础用法（复用默认连接类） #

你的代码中指定connection_factory=psycopg2.extensions.connection本身是正确的，连接失败大概率与connection_factory无关，更可能是Schema不存在、用户无权限或options参数有误。以下是可运行的正确示例：

import psycopg2
from psycopg2.extensions import connection

# 定义连接参数，包含指定Schema的options
connection_info = {
    'database': 'postgres',
    'host': '127.0.0.1',
    'port': 5432,
    'user': 'postgres',
    'password': 'postgres',
    'sslmode': 'prefer',
    'options': '-c search_path=myschema,public'
}

# 使用默认连接类作为工厂创建连接
conn = psycopg2.connect(connection_factory=connection, **connection_info)

# 验证Schema是否生效
with conn.cursor() as cur:
    cur.execute("SHOW search_path;")
    print("当前生效的search_path:", cur.fetchone()[0])

conn.close()

自定义连接工厂（自动配置Schema） #

如果希望连接建立后自动设置Schema，可以自定义连接类作为connection_factory，无需依赖options参数：

import psycopg2
from psycopg2.extensions import connection

class SchemaAwareConnection(connection):
    def __init__(self, *args, schema="public", **kwargs):
        super().__init__(*args, **kwargs)
        # 连接成功后立即设置目标Schema
        with self.cursor() as cur:
            cur.execute(f"SET search_path TO {schema}, public;")
            self.commit()

# 使用自定义工厂连接，指定目标Schema
conn = psycopg2.connect(
    database='postgres',
    host='127.0.0.1',
    port=5432,
    user='postgres',
    password='postgres',
    sslmode='prefer',
    connection_factory=lambda *args, **kwargs: SchemaAwareConnection(*args, schema="myschema", **kwargs)
)

# 验证
with conn.cursor() as cur:
    cur.execute("SHOW search_path;")
    print("当前生效的search_path:", cur.fetchone()[0])

conn.close()

连接失败排查要点 #

如果仍连接失败，请优先检查：

• 数据库中是否存在名为myschema的Schema
• 连接用户postgres是否拥有myschema的访问权限（可执行GRANT USAGE ON SCHEMA myschema TO postgres;授权）
• options参数的拼写是否正确（注意search_path后的逗号和Schema名称无拼写错误）

内容的提问来源于stack exchange，提问作者Grant.S