python - Alembic + Sqlalchemy 多列唯一约束

Question

我正在尝试使用 sqlalchemy 创建一个多列唯一约束，Alembic 会在其自动升级脚本生成器中选取该约束。

我已经使用以下方法创建约束:

在我的模型中从 sqlalchemy import UniqueConstraint

UniqueConstraint('col1', 'col2', 'number', name='uix_table_col1_col2_col3')

但是，Alembic 不会在其自动脚本生成中使用它。

我可以通过添加在 Alembic 脚本中手动创建它。

op.create_unique_constraint('uq_table_col1_col2_col3', 'table', ['col1', 'col2', 'col3'])

有没有办法让 Alembic 自动生成它？

感谢您的帮助。

Answer 1

我遇到了同样的问题，发现如果你使用声明模型定义表，你可以将它的唯一约束添加到 __table_args__ 并且 Alembic 会选择它:

class MyClass(Base):
    __tablename__ = 'table'
    __table_args__ = (
        UniqueConstraint('col1', 'col2', 'number', name='uix_table_col1_col2_col3'),
    )

http://docs.sqlalchemy.org/en/latest/orm/extensions/declarative/table_config.html