mirror of
https://github.com/GSA/notifications-api.git
synced 2026-09-10 01:56:03 -04:00
reformat
This commit is contained in:
@@ -9,20 +9,20 @@ import os
|
||||
|
||||
from alembic import op
|
||||
|
||||
revision = '0320_optimise_notifications'
|
||||
down_revision = '0319_contact_list_archived'
|
||||
environment = os.environ['NOTIFY_ENVIRONMENT']
|
||||
revision = "0320_optimise_notifications"
|
||||
down_revision = "0319_contact_list_archived"
|
||||
environment = os.environ["NOTIFY_ENVIRONMENT"]
|
||||
|
||||
|
||||
def upgrade():
|
||||
# We like to run this operation on live via the command prompt, to watch the progress and stop if necessary
|
||||
if environment not in ["live", "production"]:
|
||||
# Drop indexes notifications - no need to recreate these indexes
|
||||
op.execute('DROP INDEX IF EXISTS ix_notifications_key_type')
|
||||
op.execute('DROP INDEX IF EXISTS ix_notifications_api_key_id')
|
||||
op.execute('DROP INDEX IF EXISTS ix_notifications_notification_status')
|
||||
op.execute('DROP INDEX IF EXISTS ix_notifications_notification_type')
|
||||
op.execute('DROP INDEX IF EXISTS ix_notifications_service_id')
|
||||
op.execute("DROP INDEX IF EXISTS ix_notifications_key_type")
|
||||
op.execute("DROP INDEX IF EXISTS ix_notifications_api_key_id")
|
||||
op.execute("DROP INDEX IF EXISTS ix_notifications_notification_status")
|
||||
op.execute("DROP INDEX IF EXISTS ix_notifications_notification_type")
|
||||
op.execute("DROP INDEX IF EXISTS ix_notifications_service_id")
|
||||
|
||||
# Create new composite indexes
|
||||
# PLEASE NOTE: that if you create index on production you need to add concurrently to the create statement,
|
||||
@@ -39,32 +39,41 @@ def upgrade():
|
||||
"""
|
||||
op.execute(ix_notifications_notification_type_composite)
|
||||
# DROP and CREATE all other indexes
|
||||
op.execute('DROP INDEX IF EXISTS ix_notifications_client_reference')
|
||||
op.execute('CREATE INDEX ix_notifications_client_reference ON notifications (client_reference)')
|
||||
|
||||
op.execute('DROP INDEX IF EXISTS ix_notifications_created_at')
|
||||
op.execute('CREATE INDEX ix_notifications_created_at ON notifications (created_at)')
|
||||
|
||||
op.execute('DROP INDEX IF EXISTS ix_notifications_job_id')
|
||||
op.execute('CREATE INDEX ix_notifications_job_id ON notifications (job_id)')
|
||||
|
||||
op.execute('DROP INDEX IF EXISTS ix_notifications_reference')
|
||||
op.execute('CREATE INDEX ix_notifications_reference ON notifications (reference)')
|
||||
|
||||
op.execute('DROP INDEX IF EXISTS ix_notifications_service_created_at')
|
||||
op.execute("DROP INDEX IF EXISTS ix_notifications_client_reference")
|
||||
op.execute(
|
||||
'CREATE INDEX ix_notifications_service_created_at ON notifications (service_id, created_at)')
|
||||
"CREATE INDEX ix_notifications_client_reference ON notifications (client_reference)"
|
||||
)
|
||||
|
||||
op.execute('DROP INDEX IF EXISTS ix_notifications_template_id')
|
||||
op.execute('CREATE INDEX ix_notifications_template_id ON notifications (template_id)')
|
||||
op.execute("DROP INDEX IF EXISTS ix_notifications_created_at")
|
||||
op.execute(
|
||||
"CREATE INDEX ix_notifications_created_at ON notifications (created_at)"
|
||||
)
|
||||
|
||||
op.execute("DROP INDEX IF EXISTS ix_notifications_job_id")
|
||||
op.execute("CREATE INDEX ix_notifications_job_id ON notifications (job_id)")
|
||||
|
||||
op.execute("DROP INDEX IF EXISTS ix_notifications_reference")
|
||||
op.execute(
|
||||
"CREATE INDEX ix_notifications_reference ON notifications (reference)"
|
||||
)
|
||||
|
||||
op.execute("DROP INDEX IF EXISTS ix_notifications_service_created_at")
|
||||
op.execute(
|
||||
"CREATE INDEX ix_notifications_service_created_at ON notifications (service_id, created_at)"
|
||||
)
|
||||
|
||||
op.execute("DROP INDEX IF EXISTS ix_notifications_template_id")
|
||||
op.execute(
|
||||
"CREATE INDEX ix_notifications_template_id ON notifications (template_id)"
|
||||
)
|
||||
|
||||
# Drop indexes notification_history
|
||||
op.execute('DROP INDEX IF EXISTS ix_notification_history_key_type')
|
||||
op.execute('DROP INDEX IF EXISTS ix_notification_history_api_key_id')
|
||||
op.execute('DROP INDEX IF EXISTS ix_notification_history_notification_status')
|
||||
op.execute('DROP INDEX IF EXISTS ix_notification_history_notification_type')
|
||||
op.execute('DROP INDEX IF EXISTS ix_notification_history_service_id')
|
||||
op.execute('DROP INDEX IF EXISTS ix_notification_history_week_created')
|
||||
op.execute("DROP INDEX IF EXISTS ix_notification_history_key_type")
|
||||
op.execute("DROP INDEX IF EXISTS ix_notification_history_api_key_id")
|
||||
op.execute("DROP INDEX IF EXISTS ix_notification_history_notification_status")
|
||||
op.execute("DROP INDEX IF EXISTS ix_notification_history_notification_type")
|
||||
op.execute("DROP INDEX IF EXISTS ix_notification_history_service_id")
|
||||
op.execute("DROP INDEX IF EXISTS ix_notification_history_week_created")
|
||||
|
||||
|
||||
def downgrade():
|
||||
|
||||
Reference in New Issue
Block a user