mirror of
https://github.com/GSA/notifications-api.git
synced 2026-08-18 13:38:53 -04:00
49 lines
2.1 KiB
Python
49 lines
2.1 KiB
Python
"""empty message
|
|
|
|
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 = '0032_update_permission_to_enum'
|
|
down_revision = '0031_add_manage_team_permission'
|
|
|
|
from alembic import op
|
|
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()
|
|
op.drop_constraint('uix_service_user_permission', 'permissions', type_='unique')
|
|
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(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_constraint('uix_service_user_permission', 'permissions', type_='unique')
|
|
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(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 ###
|