clean up after model update

2026-08-23 15:56:45 -04:00 · 2023-02-28 15:29:36 -05:00
parent d6633aea5c
commit 8adf1a19bb
5 changed files with 93 additions and 14 deletions
--- a/migrations/versions/0389_no_more_letters.py
+++ b/migrations/versions/0389_no_more_letters.py
@@ -0,0 +1,56 @@
+"""
+
+Revision ID: 0389_no_more_letters.py
+Revises: 0388_no_serv_letter_contact.py
+Create Date: 2023-02-28 08:58:38.310095
+
+"""
+from alembic import op
+import sqlalchemy as sa
+from sqlalchemy.dialects import postgresql
+
+revision = '0389_no_more_letters.py'
+down_revision = '0388_no_serv_letter_contact.py'
+
+
+def upgrade():
+    # ### commands auto generated by Alembic - please adjust! ###
+    op.drop_index('ix_returned_letters_service_id', table_name='returned_letters')
+    op.drop_table('returned_letters')
+    op.drop_index('ix_daily_sorted_letter_billing_day', table_name='daily_sorted_letter')
+    op.drop_index('ix_daily_sorted_letter_file_name', table_name='daily_sorted_letter')
+    op.drop_table('daily_sorted_letter')
+    op.drop_column('services', 'volume_letter')
+    op.drop_column('services_history', 'volume_letter')
+    # ### end Alembic commands ###
+
+
+def downgrade():
+    # ### commands auto generated by Alembic - please adjust! ###
+    op.add_column('services_history', sa.Column('volume_letter', sa.INTEGER(), autoincrement=False, nullable=True))
+    op.add_column('services', sa.Column('volume_letter', sa.INTEGER(), autoincrement=False, nullable=True))
+    op.create_table('daily_sorted_letter',
+        sa.Column('id', postgresql.UUID(), autoincrement=False, nullable=False),
+        sa.Column('billing_day', sa.DATE(), autoincrement=False, nullable=False),
+        sa.Column('unsorted_count', sa.INTEGER(), autoincrement=False, nullable=False),
+        sa.Column('sorted_count', sa.INTEGER(), autoincrement=False, nullable=False),
+        sa.Column('updated_at', postgresql.TIMESTAMP(), autoincrement=False, nullable=True),
+        sa.Column('file_name', sa.VARCHAR(), autoincrement=False, nullable=True),
+        sa.PrimaryKeyConstraint('id', name='daily_sorted_letter_pkey'),
+        sa.UniqueConstraint('file_name', 'billing_day', name='uix_file_name_billing_day')
+    )
+    op.create_index('ix_daily_sorted_letter_file_name', 'daily_sorted_letter', ['file_name'], unique=False)
+    op.create_index('ix_daily_sorted_letter_billing_day', 'daily_sorted_letter', ['billing_day'], unique=False)
+    op.create_table('returned_letters',
+        sa.Column('id', postgresql.UUID(), autoincrement=False, nullable=False),
+        sa.Column('reported_at', sa.DATE(), autoincrement=False, nullable=False),
+        sa.Column('service_id', postgresql.UUID(), autoincrement=False, nullable=False),
+        sa.Column('notification_id', postgresql.UUID(), autoincrement=False, nullable=False),
+        sa.Column('created_at', postgresql.TIMESTAMP(), autoincrement=False, nullable=False),
+        sa.Column('updated_at', postgresql.TIMESTAMP(), autoincrement=False, nullable=True),
+        sa.ForeignKeyConstraint(['service_id'], ['services.id'], name='returned_letters_service_id_fkey'),
+        sa.PrimaryKeyConstraint('id', name='returned_letters_pkey'),
+        sa.UniqueConstraint('notification_id', name='returned_letters_notification_id_key')
+    )
+    op.create_index('ix_returned_letters_service_id', 'returned_letters', ['service_id'], unique=False)
+    # ### end Alembic commands ###
--- a/migrations/versions/0390_drop_dvla_provider.py
+++ b/migrations/versions/0390_drop_dvla_provider.py
@@ -0,0 +1,33 @@
+"""
+
+Revision ID: 0390_drop_dvla_provider.py
+Revises: 0389_no_more_letters.py
+Create Date: 2023-02-28 14:25:50.751952
+
+"""
+from alembic import op
+import sqlalchemy as sa
+from sqlalchemy.dialects import postgresql
+
+revision = '0390_drop_dvla_provider.py'
+down_revision = '0389_no_more_letters.py'
+
+
+def upgrade():
+    # op.execute("DELETE FROM provider_rates where provider_id = (SELECT id from provider_details where display_name='DVLA')")
+    op.execute("DELETE FROM provider_details_history where display_name = 'DVLA'")
+    op.execute("DELETE FROM provider_details where display_name = 'DVLA'")
+    # ### end Alembic commands ###
+
+
+def downgrade():
+    # migration 0066 in reverse
+    provider_id = str(uuid.uuid4())
+    op.execute(
+        "INSERT INTO provider_details (id, display_name, identifier, priority, notification_type, active, version) values ('{}', 'DVLA', 'dvla', 50, 'letter', true, 1)".format(provider_id)
+    )
+    op.execute(
+        "INSERT INTO provider_details_history (id, display_name, identifier, priority, notification_type, active, version) values ('{}', 'DVLA', 'dvla', 50, 'letter', true, 1)".format(provider_id)
+    )
+    op.execute("INSERT INTO provider_rates (id, valid_from, rate, provider_id) VALUES ('{}', '{}', 1.0, '{}')".format(uuid.uuid4(), datetime.utcnow(), provider_id))
+    # ### end Alembic commands ###