如何解决Supabase自定义Postgres Schema连接错误并写入Pandas DataFrame？

我尝试通过SQLAlchemy连接Supabase的Postgres自定义Schema，将Pandas DataFrame写入其中，但一直报错。相关代码如下：

from sqlalchemy import create_engine

def server_access():
    # Create SQLAlchemy connection string
    conn_str = (
        f"postgresql+psycopg2://{"[USER]"}:{"[PASSWORD]"}"
        f"@{"[HOST]"}:{[PORT]}/{"[SCHEMA]"}?client_encoding=utf8"
    )
    engine = create_engine(url = conn_str)
    return engine
    
engine = server_access()
df.to_sql('tbl', engine, if_exists='append', index=False)

我已完成自定义Schema的暴露配置，但未生效。使用public Schema时，无需在连接字符串中定义client_encoding即可正常运行。

补充信息：
示例连接字符串：postgresql+psycopg2://user:password@host:6543/custom_schema?client_encoding=utf8
将其中的custom_schema替换为postgres即可正常工作。

报错信息：

conn = _connect(dsn, connection_factory=connection_factory, **kwasync)
           ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
sqlalchemy.exc.OperationalError: (psycopg2.OperationalError) server didn't return client encoding
(Background on this error at: https://sqlalche.me/e/20/e3q8)

核心问题定位 #

你的连接字符串错误地将自定义Schema名放在了数据库名的位置。PostgreSQL连接字符串中，/后面的是数据库名称（Supabase默认数据库名为postgres），Schema需要通过其他方式指定。

方法一：创建Engine时指定Schema搜索路径 #

修改连接字符串指向默认数据库，通过connect_args将自定义Schema设为默认搜索路径：

from sqlalchemy import create_engine

def server_access():
    # 替换USER、PASSWORD、HOST为你的实际信息，数据库名固定为postgres
    conn_str = "postgresql+psycopg2://[USER]:[PASSWORD]@[HOST]:6543/postgres?client_encoding=utf8"
    # 将自定义schema设置为搜索路径首位
    engine = create_engine(conn_str, connect_args={"options": "-c search_path=custom_schema"})
    return engine

engine = server_access()
df.to_sql('tbl', engine, if_exists='append', index=False)

方法二：在to_sql中直接指定Schema #

保持连接字符串指向默认数据库，写入DataFrame时显式指定目标Schema：

from sqlalchemy import create_engine

def server_access():
    conn_str = "postgresql+psycopg2://[USER]:[PASSWORD]@[HOST]:6543/postgres?client_encoding=utf8"
    engine = create_engine(conn_str)
    return engine

engine = server_access()
# 通过schema参数指定目标自定义Schema
df.to_sql('tbl', engine, schema='custom_schema', if_exists='append', index=False)

额外检查项 #

• 确认自定义Schema已在Supabase控制台完成权限配置，当前数据库用户拥有该Schema的USAGE权限，以及目标表的INSERT权限
• 若仍有编码问题，可尝试移除连接字符串中的client_encoding=utf8，Supabase通常会自动适配编码

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