Alembic,如何改变ForeigenKey列

时间:2017-06-19 12:54:33

标签: python mysql sqlalchemy alembic turbogears2

我使用的是Alembic 0.8.9,SQLAlchemy 1.1.4,我的数据库是MySQL数据库。
我正在改变一个表和一个外来列:

在我的数据库中,我重命名了器官'要命名的表' 目的'。 使用

from alembic import op

def upgrade():
    op.rename_table('organs', 'purposes')

之后,我想在不同的表中更新我的ForeignKey:

class Order(DeclarativeBase):
    __tablename__ = 'orders'
    id = Column(Integer, autoincrement=True, primary_key=True)

    organ_id = Column(Integer, ForeignKey('organs.id'))

之后

class Order(DeclarativeBase):
    __tablename__ = 'orders'
    id = Column(Integer, autoincrement=True, primary_key=True)

    purpose_id = Column(Integer, ForeignKey('purposes.id'))

我需要帮助编写一个Alembic迁移脚本,以便将此更改反映在数据库中。如何更改ForeignKey列?

感谢您的帮助

1 个答案:

答案 0 :(得分:5)

感谢您提供有用的评论,让我更多地搜索SQL外键。我想我现在明白了。
这个答案告诉我的方式:
How to change the foreign key referential action? (behavior)

基本上我需要做的是重命名列(保存数据),删除旧的外键(约束?)并改为创建一个新的。

这是我的迁移脚本:

from alembic import op
import sqlalchemy as sa


def upgrade():
    op.rename_table('organs', 'purposes')
    op.alter_column('orders', 'organ_id', new_column_name='purpose_id', existing_type=sa.Integer)
    op.drop_constraint(constraint_name="orders_ibfk_2", table_name="orders", type_="foreignkey")
    op.create_foreign_key(
        constraint_name="orders_ibfk_2",
        source_table="orders",
        referent_table="purposes",
        local_cols=["purpose_id"],
        remote_cols=["id"])


def downgrade():
    op.rename_table('purposes', 'organs')
    op.alter_column('orders', 'purpose_id', new_column_name='organ_id', existing_type=sa.Integer)
    op.drop_constraint(constraint_name="orders_ibfk_2", table_name="orders", type_="foreignkey")
    op.create_foreign_key(
        constraint_name="orders_ibfk_2",
        source_table="orders",
        referent_table="organs",
        local_cols=["organ_id"],
        remote_cols=["id"])