Alembic创建初始迁移时无法识别PostgreSQL默认public schema

问题概述 #

执行alembic revision --autogenerate -m "add company and user models"时触发如下错误：

sqlalchemy.exc.ProgrammingError: (psycopg2.errors.InvalidSchemaName) no schema has been selected to create in
LINE 2: CREATE TABLE alembic_version (
                     ^

未显式指定Schema，预期默认使用PostgreSQL的public Schema，但Alembic无法正常创建表。

原因分析 #

1. 数据库用户的search_path配置未包含public Schema，导致PostgreSQL无法确定表的创建位置
2. 连接字符串未显式指定目标Schema，Alembic无法自动定位到public
3. 极端情况下public Schema不存在或当前用户无操作权限
4. 模型文件存在语法错误，可能导致Schema元数据加载异常（次要但需修复）

解决方案 #

1. 修复Models.py中的语法错误 #

你的Company和User类的__repr__方法存在语法错误，缺少self参数和return语句，会导致模型无法正确加载。修改如下：

# Company类的__repr__
def __repr__(self):
    return f"<Company(id={self.id}, name={self.name}, domain={self.domain}, is_active={self.is_active})>"

# User类的__repr__
def __repr__(self):
    return f"<User(id={self.id}, email={self.email}, username={self.username})>"

2. 检查并设置数据库用户的Search Path #

进入PostgreSQL容器：

docker exec -it postgres psql -U postgres

查看当前search_path：

SHOW search_path;

如果结果中没有public，执行以下命令设置默认搜索路径：

ALTER ROLE postgres SET search_path TO public;

退出容器后重启数据库容器：

make db-stop && make db-start

3. 在连接字符串中显式指定Schema #

修改migrations/env.py中的连接字符串，添加options参数强制指定public Schema：

connection_string = (
    f"postgresql+psycopg2://{DB_USER}:{DB_PASSWORD}@{DB_HOST}:{DB_PORT}/{DB_NAME}?options=-csearch_path%3Dpublic"
)

4. 确保public Schema存在并授权 #

如果public Schema被意外删除，进入数据库执行：

CREATE SCHEMA IF NOT EXISTS public;
GRANT ALL ON SCHEMA public TO postgres;

5. 重新执行迁移命令 #

清理可能的错误状态后，重新生成并应用迁移：

alembic revision --autogenerate -m "add company and user models"
alembic upgrade head

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