darkhelm/notifications-api
1
0
Fork 0
mirror of https://github.com/GSA/notifications-api.git synced 2026-01-30 06:21:50 -05:00
Code Issues Packages Projects Releases Wiki Activity

Upgrade and downgrade now works.

Browse Source 
Fix bug with repeated upgrade and downgrade.

Fix script revision number.
This commit is contained in:
Nicholas Staples
2016-03-02 10:26:20 +00:00
parent 47c4368007
commit bd898a4526
1 changed files with 19 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

31
migrations/versions/0032_update_permission_to_enum.py
View File

@@ -1,13 +1,13 @@
"""empty message
Revision ID: 49380ad07c88
Revision ID: 0032_update_permission_to_enum
Revises: 0031_add_manage_team_permission
Create Date: 2016-03-01 17:08:12.184393
"""
# revision identifiers, used by Alembic.
revision = '49380ad07c88'
revision = '0032_update_permission_to_enum'
down_revision = '0031_add_manage_team_permission'
from alembic import op
@@ -17,25 +17,32 @@ import sqlalchemy as sa
def upgrade():
### commands auto generated by Alembic - please adjust! ###
conn = op.get_bind()
permissions = conn.execute("SELECT (id, permission) FROM permissions").fetchall()
print(permissions)
op.drop_column('permissions', 'permission')
permissions = conn.execute("SELECT id, permission FROM permissions").fetchall()
op.drop_constraint('uix_service_user_permission', 'permissions', type_='unique')
op.add_column('permissions', sa.Column('permission', sa.Enum('manage_service', 'send_messages', 'manage_api_keys', 'manage_templates', 'manage_team', 'view_activity', name='permission_types'), nullable=False))
op.drop_column('permissions', 'permission')
permission_types = sa.Enum('manage_service', 'send_messages', 'manage_api_keys', 'manage_templates', 'manage_team', 'view_activity', name='permission_types')
permission_types.create(op.get_bind())
op.add_column('permissions', sa.Column('permission', permission_types, nullable=True))
for p in permissions:
conn.execute("UPDATE permissions SET permission='{}' WHERE ID='{}'".format(p[1], p[0]))
op.create_unique_constraint('uix_service_user_permission', 'permissions', ['service_id', 'user_id', '_permission'])
conn.execute("UPDATE permissions SET permission='{}' WHERE id='{}'".format(str(p[1]), str(p[0])))
op.create_unique_constraint('uix_service_user_permission', 'permissions', ['service_id', 'user_id', 'permission'])
op.alter_column('permissions', 'permission', nullable=False)
### end Alembic commands ###
def downgrade():
### commands auto generated by Alembic - please adjust! ###
conn = op.get_bind()
permissions = conn.execute("SELECT (id, permission) FROM permissions").fetchall()
op.drop_column('permissions', 'permission')
permissions = conn.execute("SELECT id, permission FROM permissions").fetchall()
op.drop_constraint('uix_service_user_permission', 'permissions', type_='unique')
op.add_column('permissions', sa.Column('permission', sa.VARCHAR(length=255), autoincrement=False, nullable=False))
op.drop_column('permissions', 'permission')
try:
sa.Enum(name='permission_types').drop(conn, checkfirst=False)
except:
pass
op.add_column('permissions', sa.Column('permission', sa.VARCHAR(length=255), autoincrement=False, nullable=True))
for p in permissions:
conn.execute("UPDATE permissions SET permission='{}' WHERE ID='{}'".format(p[1], p[0]))
conn.execute("UPDATE permissions SET permission='{}' WHERE ID='{}'".format(str(p[1]), str(p[0])))
op.create_unique_constraint('uix_service_user_permission', 'permissions', ['service_id', 'user_id', 'permission'])
op.alter_column('permissions', 'permission', nullable=False)
### end Alembic commands ###