From ad80adb641e98f2124a15ca92997373057b379b5 Mon Sep 17 00:00:00 2001 From: Adam Shimali Date: Wed, 15 May 2024 16:33:04 +0100 Subject: [PATCH] missing migration file --- ...75e0c068_add_local_land_charges_boolean.py | 54 +++++++++++++++++++ 1 file changed, 54 insertions(+) create mode 100644 migrations/versions/376575e0c068_add_local_land_charges_boolean.py diff --git a/migrations/versions/376575e0c068_add_local_land_charges_boolean.py b/migrations/versions/376575e0c068_add_local_land_charges_boolean.py new file mode 100644 index 0000000..e1a7416 --- /dev/null +++ b/migrations/versions/376575e0c068_add_local_land_charges_boolean.py @@ -0,0 +1,54 @@ +"""add local land charges boolean + +Revision ID: 376575e0c068 +Revises: 7d785f99a5fb +Create Date: 2024-05-15 15:59:40.160245 + +""" + +import sqlalchemy as sa +from alembic import op + +# revision identifiers, used by Alembic. +revision = "376575e0c068" +down_revision = "7d785f99a5fb" +branch_labels = None +depends_on = None + + +def upgrade(): + # ### commands auto generated by Alembic - please adjust! ### + with op.batch_alter_table("consideration", schema=None) as batch_op: + batch_op.add_column( + sa.Column("is_local_land_charge", sa.Boolean(), nullable=True) + ) + + known_local_land_charges = [ + "conservation-areas", + "enforcement-notices", + "tree-preservation-orders", + "assets-of-community-value", + "compulsory-purchase-orders", + "opencast-coal-prospecting-areas", + "listed-buildings", + ] + + op.execute("UPDATE consideration SET is_local_land_charge = false") + + for local_land_charge in known_local_land_charges: + op.execute( + f"UPDATE consideration SET is_local_land_charge = true WHERE slug = '{local_land_charge}'" + ) + + with op.batch_alter_table("consideration", schema=None) as batch_op: + batch_op.alter_column("is_local_land_charge", nullable=False) + + # ### end Alembic commands ### + + +def downgrade(): + # ### commands auto generated by Alembic - please adjust! ### + with op.batch_alter_table("consideration", schema=None) as batch_op: + batch_op.drop_column("is_local_land_charge") + + # ### end Alembic commands ###