notifications-api/migrations/versions/0078_add_sent_notification_status.py

"""empty message

Revision ID: 0078_sent_notification_status
Revises: 0077_add_intl_notification
Create Date: 2017-04-24 16:55:20.731069

"""

# revision identifiers, used by Alembic.
revision = "0078_sent_notification_status"
down_revision = "0077_add_intl_notification"

import sqlalchemy as sa
from alembic import op

enum_name = "notify_status_type"
tmp_name = "tmp_" + enum_name

old_options = (
    "created",
    "sending",
    "delivered",
    "pending",
    "failed",
    "technical-failure",
    "temporary-failure",
    "permanent-failure",
)
new_options = old_options + ("sent",)

old_type = sa.Enum(*old_options, name=enum_name)
new_type = sa.Enum(*new_options, name=enum_name)

alter_str = "ALTER TABLE {table} ALTER COLUMN status TYPE {enum} USING status::text::notify_status_type "


def upgrade():
    op.execute("ALTER TYPE notify_status_type RENAME TO tmp_notify_status_type")

    new_type.create(op.get_bind())
    op.execute(
        "ALTER TABLE notifications ALTER COLUMN status TYPE notify_status_type USING status::text::notify_status_type"
    )
    op.execute(
        "ALTER TABLE notification_history ALTER COLUMN status TYPE notify_status_type USING status::text::notify_status_type"
    )

    op.execute("DROP TYPE tmp_notify_status_type")


def downgrade():
    op.execute("ALTER TYPE notify_status_type RENAME TO tmp_notify_status_type")

    op.execute("UPDATE notifications SET status='sending' where status='sent'")
    op.execute("UPDATE notification_history SET status='sending' where status='sent'")

    old_type.create(op.get_bind())

    op.execute(
        "ALTER TABLE notifications ALTER COLUMN status TYPE notify_status_type USING status::text::notify_status_type"
    )
    op.execute(
        "ALTER TABLE notification_history ALTER COLUMN status TYPE notify_status_type USING status::text::notify_status_type"
    )

    op.execute("DROP TYPE tmp_notify_status_type")