Update migration script for service permissions

This commit is contained in:
Ken Tsang
2017-05-24 13:22:36 +01:00
parent eda3e412f7
commit 640ab665d0

View File

@@ -1,13 +1,13 @@
"""empty message
Revision ID: 0086_migrate_default_svc_perms
Revision ID: 0086_migrate_existing_svc_perms
Revises: 0085_update_incoming_to_inbound
Create Date: 2017-05-23 18:13:03.532095
"""
# revision identifiers, used by Alembic.
revision = '0086_migrate_default_svc_perms'
revision = '0086_migrate_existing_svc_perms'
down_revision = '0085_update_incoming_to_inbound'
from alembic import op
@@ -20,8 +20,17 @@ def upgrade():
"id NOT IN (SELECT service_id FROM service_permissions "\
"WHERE service_id=id AND permission='{0}')".format(permission)
def get_values_from_flag(permission, flag):
return "SELECT id, '{0}' FROM services WHERE "\
"{1} AND id NOT IN (SELECT service_id FROM service_permissions "\
"WHERE service_id=id AND permission='{0}')".format(permission, flag)
op.execute("INSERT INTO service_permissions (service_id, permission) {}".format(get_values('sms')))
op.execute("INSERT INTO service_permissions (service_id, permission) {}".format(get_values('email')))
op.execute("INSERT INTO service_permissions (service_id, permission) {}".format(
get_values_from_flag('letter', 'can_send_letters')))
op.execute("INSERT INTO service_permissions (service_id, permission) {}".format(
get_values_from_flag('international_sms', 'can_send_international_sms')))
def downgrade():