mirror of
https://github.com/GSA/notifications-api.git
synced 2025-12-14 09:12:06 -05:00
41 lines
1.2 KiB
Python
41 lines
1.2 KiB
Python
"""
|
|
|
|
Revision ID: 0177_add_virus_scan_statuses
|
|
Revises: 0176_alter_billing_columns
|
|
Create Date: 2018-02-21 14:05:04.448977
|
|
|
|
"""
|
|
from alembic import op
|
|
|
|
revision = "0177_add_virus_scan_statuses"
|
|
down_revision = "0176_alter_billing_columns"
|
|
|
|
|
|
def upgrade():
|
|
op.execute(
|
|
"INSERT INTO notification_status_types (name) VALUES ('pending-virus-check')"
|
|
)
|
|
op.execute(
|
|
"INSERT INTO notification_status_types (name) VALUES ('virus-scan-failed')"
|
|
)
|
|
|
|
|
|
def downgrade():
|
|
op.execute(
|
|
"UPDATE notifications SET notification_status = 'created' WHERE notification_status = 'pending-virus-check'"
|
|
)
|
|
op.execute(
|
|
"UPDATE notification_history SET notification_status = 'created' WHERE notification_status = 'pending-virus-check'"
|
|
)
|
|
|
|
op.execute(
|
|
"UPDATE notifications SET notification_status = 'permanent-failure' WHERE notification_status = 'virus-scan-failed'"
|
|
)
|
|
op.execute(
|
|
"UPDATE notification_history SET notification_status = 'permanent-failure' WHERE notification_status = 'virus-scan-failed'"
|
|
)
|
|
|
|
op.execute(
|
|
"DELETE FROM notification_status_types WHERE name in ('pending-virus-check', 'virus-scan-failed')"
|
|
)
|