Clean up migration

This commit is contained in:
richard-dds 2019-10-04 16:25:52 -04:00
parent 2ad30b5fa4
commit 31d2380afc

View File

@ -17,7 +17,6 @@ depends_on = None
def upgrade(): def upgrade():
# op.alter_column("environment_roles", "status", new_column_name="status_old")
op.alter_column( op.alter_column(
"environment_roles", "environment_roles",
"status", "status",
@ -30,18 +29,10 @@ def upgrade():
native_enum=False, native_enum=False,
), ),
existing_type=sa.Enum( existing_type=sa.Enum(
"PENDING", "PENDING", "COMPLETED", "PENDING_DELETE", name="status", native_enum=False
"COMPLETED",
"PENDING_DELETE",
name="status",
native_enum=False,
), ),
) )
# conn = op.get_bind()
# conn.execute("UPDATE environment_roles SET status = status_old")
# op.drop_column("environment_roles", "status_old")
def downgrade(): def downgrade():
conn = op.get_bind() conn = op.get_bind()
@ -51,16 +42,11 @@ def downgrade():
SET status = (CASE WHEN status = 'DELETED' THEN 'PENDING_DELETE' ELSE status END) SET status = (CASE WHEN status = 'DELETED' THEN 'PENDING_DELETE' ELSE status END)
""" """
) )
op.alter_column( op.alter_column(
"environment_roles", "environment_roles",
"status", "status",
type_=sa.Enum( type_=sa.Enum(
"PENDING", "PENDING", "COMPLETED", "PENDING_DELETE", name="status", native_enum=False
"COMPLETED",
"PENDING_DELETE",
name="status",
native_enum=False,
), ),
existing_type=sa.Enum( existing_type=sa.Enum(
"PENDING", "PENDING",