python - 多个模型上的 Alembic 迁移

Question

我正在尝试使用 Alembic 为两个模型创建一个 --autogenerate 的修订版，但收到重复的表键错误。是否需要指定模式？如果可以，如何设置？我读过的文档说使用 __table_args__ = {'schema': 'somename'} ，但这没有帮助。非常感谢任何提示或建议。

我目前的设置是:

base.py

from sqlalchemy.ext.declarative import declarative_base
Base = declarative_base()

工作空间.py

from sqlalchemy import Column, Integer, String
from base import Base

class WorkspaceModel(Base):

    __tablename__ = 'workspaces'

    id = Column(Integer, primary_key=True)
    name = Column(String)

主机.py

from sqlalchemy import Column, Integer, String
from base import Base

class HostModel(Base):

    __tablename__ = 'hosts'

    id = Column(Integer, primary_key=true)
    ip = Column(String)

alembic/env.py

from host import HostModel
from workspace import WorkspaceModel
target_metadata = [HostModel.metadata, WorkspaceModel.metadata]

错误

ValueError: Duplicate table keys across multiple MetaData objects: "hosts", "workspaces"

Answer 1

为了从 @esdotzed 和 @univerio 说清楚，你必须使用单个 Base.metadata - 但 仍然是 import 个别模型 。
在最初的问题中，alembic/env.py 应该是这样的:

from base import Base

# This two won't be referenced, but *have* to be imported to populate `Base.metadata`
from host import HostModel
from workspace import WorkspaceModel

target_metadata = Base.metadata

如果您没有 import 两个模型，自动生成的迁移最终会删除您的整个数据库 - 因为 Base.metadata 本身不知道任何模型。