darkhelm/notifications-api
1
0
Fork 0
mirror of https://github.com/GSA/notifications-api.git synced 2025-12-24 09:21:39 -05:00
Code Issues Packages Projects Releases Wiki Activity

Add new notification status types for technical_failure, temporary_failure, permanent_failure

Browse Source
This commit is contained in:
Rebecca Law
2016-05-17 11:46:28 +01:00
parent 74df271739
commit 49db4e81d5
2 changed files with 50 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
app/models.py
View File

@@ -304,7 +304,8 @@ class VerifyCode(db.Model):
return check_hash(cde, self._code)
NOTIFICATION_STATUS_TYPES = ['sending', 'delivered', 'failed']
NOTIFICATION_STATUS_TYPES = ['sending', 'delivered', 'failed',
'technical_failure', 'temporary_failure', 'permanent_failure']
class Notification(db.Model):
@@ -339,7 +340,7 @@ class Notification(db.Model):
nullable=True,
onupdate=datetime.datetime.utcnow)
status = db.Column(
db.Enum(*NOTIFICATION_STATUS_TYPES, name='notification_status_types'), nullable=False, default='sending')
db.Enum(*NOTIFICATION_STATUS_TYPES, name='notification_status_type'), nullable=False, default='sending')
reference = db.Column(db.String, nullable=True, index=True)

47
migrations/versions/0017_add_failure_types.py Normal file
View File

@@ -0,0 +1,47 @@
"""empty message
Revision ID: 0017_add_failure_types
Revises: 0015_fix_template_data
Create Date: 2016-05-17 11:23:36.881219
"""
# revision identifiers, used by Alembic.
revision = '0017_add_failure_types'
down_revision = '0015_fix_template_data'
from alembic import op
import sqlalchemy as sa
from alembic import op
import sqlalchemy as sa
from sqlalchemy.dialects import postgresql
def upgrade():
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('new_status', status_type, nullable=True))
op.execute('update notifications set new_status = CAST(CAST(status as text) as notification_status_type)')
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_types')
op.alter_column('notifications', 'status', nullable=False)
def downgrade():
status_type = sa.Enum('sending', 'delivered', 'failed',
name='notification_status_types')
status_type.create(op.get_bind())
op.add_column('notifications', sa.Column('old_status', status_type, nullable=True))
op.execute('update notifications set old_status = CAST(CAST(status as text) as notification_status_types)')
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 notification_status_type')
op.alter_column('notifications', 'status', nullable=False)