能否通过SQLAlchemy和Alembic将迁移文件关联至对应数仓模型？

我计划使用SQLAlchemy和Alembic搭建数据仓库，初始设想的项目结构如下：

root
- database_schema
-- table1.py (SQLAlchemy model)
-- table2.py
...
- database_schema2
-- table1.py
...
- alembic

但Alembic默认会把所有迁移文件生成在同一个versions文件夹中，请问是否可以将迁移文件与对应的SQLAlchemy模型关联存放，实现如下项目结构：

root
- database_schema
-- table1
--- migrations
---- migration1.py (Alembic migration)
--- table1.py (SQLAlchemy model)
-- table2
--- migrations
--- table2.py
...
- database_schema2
-- table1.py
...
- alembic

可以实现这种结构，核心是利用Alembic的多版本路径配置和指定生成目录的参数，具体步骤如下：

1. 配置Alembic扫描多个迁移目录 #

修改alembic.ini中的version_locations参数，列出所有模型对应的migrations目录路径：

version_locations = database_schema/table1/migrations database_schema/table2/migrations database_schema2/table1/migrations

这样Alembic在执行迁移、查看历史时，会自动扫描所有指定目录下的迁移文件。

2. 生成迁移文件到对应模型目录 #

当需要为某个模型生成迁移时，使用--version-path参数指定目标migrations目录：

# 为table1生成初始结构迁移
alembic revision --autogenerate -m "init table1 schema" --version-path database_schema/table1/migrations

# 为table2添加字段的迁移
alembic revision --autogenerate -m "add status column to table2" --version-path database_schema/table2/migrations

执行后，迁移文件会直接生成到对应模型的migrations目录下。

3. 确保模型元数据被正确加载 #

在alembic/env.py中，需要导入所有SQLAlchemy模型的元数据，让Alembic能对比生成正确的迁移：

from database_schema.table1.table1 import Table1Model
from database_schema.table2.table2 import Table2Model
from database_schema2.table1 import Table1Schema2Model

# 将所有模型的metadata传入target_metadata
target_metadata = [
    Table1Model.metadata,
    Table2Model.metadata,
    Table1Schema2Model.metadata
]

如果模型数量较多，也可以通过遍历目录动态导入所有模型的metadata，减少手动维护的工作量。

注意事项 #

• 每个migrations目录下需要创建__init__.py文件，确保Python将其识别为有效模块。
• 迁移文件的revision ID由Alembic自动生成，全局唯一，无需手动干预。
• 如果需要单独执行某个模型目录下的迁移，同样使用--version-path指定路径：

  alembic upgrade head --version-path database_schema/table1/migrations
  

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