Files
notifications-api/migrations/versions/0022_add_pending_status.py
Rebecca Law 25a1b7f31c Firetext does not have a status code for temporary-failure.
In order to set a message as temporary-failure, we check if it is in pending status first.
Otherwise a delivery receipt for failure is set to permanent failure.
2016-05-26 16:46:00 +01:00

52 lines
2.1 KiB
Python

"""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'
from alembic import op
import sqlalchemy as sa
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 ###