"""

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')")