diff --git a/migrations/versions/0082_set_international.py b/migrations/versions/0082_set_international.py new file mode 100644 index 000000000..fa89bd2f7 --- /dev/null +++ b/migrations/versions/0082_set_international.py @@ -0,0 +1,45 @@ +"""empty message + +Revision ID: 0082_set_international +Revises: 0081_noti_status_as_enum +Create Date: 2017-05-05 15:26:34.621670 + +""" +from datetime import datetime +from alembic import op + +# revision identifiers, used by Alembic. +revision = '0082_set_international' +down_revision = '0081_noti_status_as_enum' + + +def upgrade(): + conn = op.get_bind() + start = datetime.utcnow() + notification_history = "select id from notification_history where international is null limit 10000" + + results = conn.execute(notification_history) + res = results.fetchall() + + while len(res) > 0: + conn.execute("update notification_history set international = False where id in ({})".format( + notification_history)) + results = conn.execute(notification_history) + res = results.fetchall() + + notifications = "select id from notifications where international is null limit 10000" + results2 = conn.execute(notifications) + res2 = results2.fetchall() + while len(res2) > 0: + conn.execute("update notifications set international = False where id in ({})".format(notifications)) + + results2 = conn.execute(notifications) + res2 = results2.fetchall() + + end = datetime.utcnow() + print("Started at: {} ended at: {}".format(start, end)) + + +def downgrade(): + # There is no way to downgrade this update. + pass diff --git a/migrations/versions/0083_set_international_not_null.py b/migrations/versions/0083_set_international_not_null.py new file mode 100644 index 000000000..da3a7491b --- /dev/null +++ b/migrations/versions/0083_set_international_not_null.py @@ -0,0 +1,31 @@ +"""empty message + +Revision ID: 0083_set_international_not_null +Revises: 0082_set_international +Create Date: 2017-05-10 14:08:51.067762 + +""" +from alembic import op +import sqlalchemy as sa + +# revision identifiers, used by Alembic. +revision = '0083_set_international_not_null' +down_revision = '0082_set_international' + + +def upgrade(): + op.alter_column('notification_history', 'international', + existing_type=sa.BOOLEAN(), + nullable=False) + op.alter_column('notifications', 'international', + existing_type=sa.BOOLEAN(), + nullable=False) + + +def downgrade(): + op.alter_column('notifications', 'international', + existing_type=sa.BOOLEAN(), + nullable=True) + op.alter_column('notification_history', 'international', + existing_type=sa.BOOLEAN(), + nullable=True)