diff --git a/module_build_service/migrations/versions/40b2c7d988d7_add_reused_module_id_column.py b/module_build_service/migrations/versions/40b2c7d988d7_add_reused_module_id_column.py index 4035178f..83c112b0 100644 --- a/module_build_service/migrations/versions/40b2c7d988d7_add_reused_module_id_column.py +++ b/module_build_service/migrations/versions/40b2c7d988d7_add_reused_module_id_column.py @@ -16,7 +16,6 @@ import sqlalchemy as sa def upgrade(): op.add_column('module_builds', sa.Column('reused_module_id', sa.Integer(), nullable=True)) - sa.ForeignKeyConstraint(['reused_module_id'], ['module_builds.id'], ), def downgrade(): diff --git a/module_build_service/migrations/versions/d5188b4a7bf1_add_fk_reused_module_id.py b/module_build_service/migrations/versions/d5188b4a7bf1_add_fk_reused_module_id.py new file mode 100644 index 00000000..136d78d9 --- /dev/null +++ b/module_build_service/migrations/versions/d5188b4a7bf1_add_fk_reused_module_id.py @@ -0,0 +1,23 @@ +"""Add the missing foreign key constraint for reused_module_id + +Revision ID: d5188b4a7bf1 +Revises: 0b00036c540f +Create Date: 2019-08-02 13:22:14.257869 +""" + +# revision identifiers, used by Alembic. +revision = "d5188b4a7bf1" +down_revision = "0b00036c540f" + +from alembic import op +import sqlalchemy as sa + + +def upgrade(): + with op.batch_alter_table("module_builds", schema=None) as batch_op: + batch_op.create_foreign_key("reused_module", "module_builds", ["reused_module_id"], ["id"]) + + +def downgrade(): + with op.batch_alter_table("module_builds", schema=None) as batch_op: + batch_op.drop_constraint("reused_module", type_="foreignkey")