Alembic 1.9.0的check命令无法检测约束名称差异求助

我正在一个包含SQLite、SQLAlchemy和Alembic的项目中工作，当前任务是编写测试确保迁移脚本与SQLAlchemy模型保持同步。我发现Alembic 1.9.0新增了内置的check命令正好用于该场景。测试代码如下：

from alembic.command import check, upgrade
from alembic.config import Config

# pytest
def test_migrations_and_sqlalchelmy_models_are_in_sync():
    config = Config(CONFIG_PATH)
    upgrade(config, "head")
    check(config)

该命令整体运行正常，但在检测约束名称（测试了sqlalchemy.UniqueConstraint和sqlalchemy.ForeignKeyConstraint）时，无法识别两者间的差异。请问该命令是否应该检测约束名称的差异？是否有解决方法？

环境版本：python3.10.6、sqlalchemy1.4.23、alembic==1.9.0

1. check命令是否应该检测约束名称差异？ #

Alembic 1.9.0的check命令默认不会检测约束名称的差异。这是因为该命令的核心逻辑是对比数据库结构与模型生成的元数据结构，但在SQLAlchemy和Alembic的设计中，约束名称的匹配并非默认校验项——尤其是SQLite本身对约束名称的处理有特殊性（比如部分约束名称在SQLite中不会被持久化或查询返回），这进一步导致check命令无法识别名称差异。

2. 解决方法 #

方法一：自定义校验逻辑 #

在现有测试基础上，添加手动校验约束名称的逻辑，直接对比模型定义与数据库实际的约束名称：

from sqlalchemy import create_engine, inspect
from alembic.command import upgrade, check
from alembic.config import Config
from your_models import Base  # 替换为你的模型基类导入路径

def test_migrations_and_models_sync():
    config = Config(CONFIG_PATH)
    upgrade(config, "head")
    
    # 先执行Alembic自带的check命令
    check(config)
    
    # 自定义约束名称校验逻辑
    engine = create_engine(config.get_main_option("sqlalchemy.url"))
    inspector = inspect(engine)
    
    # 遍历所有模型表，逐一对比约束名称
    for table_name in Base.metadata.tables:
        model_table = Base.metadata.tables[table_name]
        # 收集模型中定义的约束名称（统一小写避免大小写差异）
        model_constraints = set()
        # 收集唯一约束名称
        for constraint in model_table.constraints:
            if hasattr(constraint, "name") and constraint.name:
                model_constraints.add(constraint.name.lower())
        # 收集外键约束名称
        for fk in model_table.foreign_keys:
            if fk.name:
                model_constraints.add(fk.name.lower())
        
        # 收集数据库中实际存在的约束名称
        db_constraints = set()
        # 获取数据库中的唯一约束
        for uc in inspector.get_unique_constraints(table_name):
            if uc["name"]:
                db_constraints.add(uc["name"].lower())
        # 获取数据库中的外键约束
        for fk in inspector.get_foreign_keys(table_name):
            if fk["name"]:
                db_constraints.add(fk["name"].lower())
        
        # 断言两者一致
        assert model_constraints == db_constraints, \
            f"表 {table_name} 约束名称不匹配：\n模型中有但数据库缺失：{model_constraints - db_constraints}\n数据库中有但模型缺失：{db_constraints - model_constraints}"

方法二：升级Alembic版本 #

后续版本的Alembic（如1.10及以上）对check命令的校验逻辑做了优化，已经支持约束名称的检测。可以尝试升级到最新稳定版：

pip install --upgrade alembic

升级前请务必在测试环境验证项目兼容性，避免依赖冲突。

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