diff --git a/migrations/versions/0083_set_international.py b/migrations/versions/0083_set_international.py new file mode 100644 index 000000000..8251901eb --- /dev/null +++ b/migrations/versions/0083_set_international.py @@ -0,0 +1,45 @@ +"""empty message + +Revision ID: 0083_set_international +Revises: 0082_add_go_live_template +Create Date: 2017-05-05 15:26:34.621670 + +""" +from datetime import datetime +from alembic import op + +# revision identifiers, used by Alembic. +revision = '0083_set_international' +down_revision = '0082_add_go_live_template' + + +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