"""empty message

Revision ID: a726fff4ed59
Revises: 0022_add_pending_status
Create Date: 2016-05-25 15:47:32.568097

"""

# revision identifiers, used by Alembic.
revision = "0022_add_pending_status"
down_revision = "0021_add_delivered_failed_counts"

import sqlalchemy as sa
from alembic import op


def upgrade():
    ### commands auto generated by Alembic - please adjust! ###
    status_type = sa.Enum(
        "sending",
        "delivered",
        "pending",
        "failed",
        "technical-failure",
        "temporary-failure",
        "permanent-failure",
        name="notify_status_types",
    )
    status_type.create(op.get_bind())
    op.add_column("notifications", sa.Column("new_status", status_type, nullable=True))
    op.execute(
        "update notifications set new_status = CAST(CAST(status as text) as notify_status_types)"
    )
    op.alter_column("notifications", "status", new_column_name="old_status")
    op.alter_column("notifications", "new_status", new_column_name="status")
    op.drop_column("notifications", "old_status")
    op.get_bind()
    op.execute("DROP TYPE notification_status_type")
    op.alter_column("notifications", "status", nullable=False)

    ### end Alembic commands ###


def downgrade():
    ### commands auto generated by Alembic - please adjust! ###
    status_type = sa.Enum(
        "sending",
        "delivered",
        "failed",
        "technical-failure",
        "temporary-failure",
        "permanent-failure",
        name="notification_status_type",
    )
    status_type.create(op.get_bind())
    op.add_column("notifications", sa.Column("old_status", status_type, nullable=True))

    op.execute("update notifications set status = 'sending' where status = 'pending'")
    op.execute(
        "update notifications set old_status = CAST(CAST(status as text) as notification_status_type)"
    )
    op.alter_column("notifications", "status", new_column_name="new_status")
    op.alter_column("notifications", "old_status", new_column_name="status")
    op.drop_column("notifications", "new_status")
    op.get_bind()
    op.execute("DROP TYPE notify_status_types")
    op.alter_column("notifications", "status", nullable=False)
    ### end Alembic commands ###