|
| 1 | +"""ws_sum_reports permissions |
| 2 | +
|
| 3 | +Revision ID: 2ee8b7eb119f |
| 4 | +Revises: 2063ac75ffb1 |
| 5 | +Create Date: 2025-05-30 16:22:47.068338+00:00 |
| 6 | +
|
| 7 | +""" |
| 8 | +from alembic import op |
| 9 | + |
| 10 | +from faraday.server.models import PermissionsUnitAction, Role |
| 11 | +from faraday.server.utils.permissions import GROUP_WS_SUM_REPORTS, UNIT_WS_SUM_REPORTS |
| 12 | + |
| 13 | +CREATE = PermissionsUnitAction.CREATE_ACTION |
| 14 | +READ = PermissionsUnitAction.READ_ACTION |
| 15 | +UPDATE = PermissionsUnitAction.UPDATE_ACTION |
| 16 | +DELETE = PermissionsUnitAction.DELETE_ACTION |
| 17 | +ACTIONS = [CREATE, READ, UPDATE, DELETE] |
| 18 | + |
| 19 | + |
| 20 | +# revision identifiers, used by Alembic. |
| 21 | +revision = '2ee8b7eb119f' |
| 22 | +down_revision = '2063ac75ffb1' |
| 23 | +branch_labels = None |
| 24 | +depends_on = None |
| 25 | + |
| 26 | + |
| 27 | +def upgrade(): |
| 28 | + op.execute( |
| 29 | + "SELECT setval('permissions_group_id_seq', (SELECT MAX(id) FROM permissions_group));" |
| 30 | + ) |
| 31 | + |
| 32 | + op.execute( |
| 33 | + "SELECT setval('permissions_unit_id_seq', (SELECT MAX(id) FROM permissions_unit));" |
| 34 | + ) |
| 35 | + |
| 36 | + op.execute( |
| 37 | + "SELECT setval('permissions_unit_action_id_seq', (SELECT MAX(id) FROM permissions_unit_action));" |
| 38 | + ) |
| 39 | + |
| 40 | + op.execute(f"INSERT INTO permissions_group (name) VALUES ('{GROUP_WS_SUM_REPORTS}');") # nosec B608 |
| 41 | + |
| 42 | + result = op.get_bind().execute( |
| 43 | + f"SELECT id FROM permissions_group WHERE name = '{GROUP_WS_SUM_REPORTS}';" # nosec B608 |
| 44 | + ) |
| 45 | + group_id = result.scalar() |
| 46 | + |
| 47 | + op.execute( |
| 48 | + f"INSERT INTO permissions_unit (name, permissions_group_id) VALUES ('{UNIT_WS_SUM_REPORTS}', {group_id});" # nosec B608 |
| 49 | + ) |
| 50 | + |
| 51 | + result = op.get_bind().execute( |
| 52 | + f"SELECT id FROM permissions_unit WHERE name = '{UNIT_WS_SUM_REPORTS}';" # nosec B608 |
| 53 | + ) |
| 54 | + unit_id = result.scalar() |
| 55 | + |
| 56 | + op.execute( |
| 57 | + f"INSERT INTO permissions_unit_action (action_type, permissions_unit_id) VALUES " |
| 58 | + f"('{CREATE}', {unit_id}), ('{READ}', {unit_id}), ('{UPDATE}', {unit_id}), ('{DELETE}', {unit_id});" # nosec B608 |
| 59 | + ) |
| 60 | + |
| 61 | + permisison_unit_action_ids = [] |
| 62 | + for action in ACTIONS: |
| 63 | + result = op.get_bind().execute( |
| 64 | + f"SELECT id FROM permissions_unit_action WHERE action_type = '{action}' AND permissions_unit_id = {unit_id};" # nosec B608 |
| 65 | + ) |
| 66 | + permisison_unit_action_ids.append(result.scalar()) |
| 67 | + |
| 68 | + roles = Role.query.all() |
| 69 | + for action_id in permisison_unit_action_ids: |
| 70 | + for role in roles: |
| 71 | + op.execute( |
| 72 | + f"INSERT INTO role_permission (unit_action_id, role_id, allowed) VALUES ({action_id}, {role.id}, true);" # nosec B608 |
| 73 | + ) |
| 74 | + |
| 75 | + |
| 76 | +def downgrade(): |
| 77 | + result = op.get_bind().execute( |
| 78 | + f"SELECT id FROM permissions_unit WHERE name = '{UNIT_WS_SUM_REPORTS}';" # nosec B608 |
| 79 | + ) |
| 80 | + unit_id = result.scalar() |
| 81 | + |
| 82 | + permisison_unit_action_ids = [] |
| 83 | + for action in ACTIONS: |
| 84 | + result = op.get_bind().execute( |
| 85 | + f"SELECT id FROM permissions_unit_action WHERE action_type = '{action}' AND permissions_unit_id = {unit_id};" # nosec B608 |
| 86 | + ) |
| 87 | + permisison_unit_action_ids.append(result.scalar()) |
| 88 | + |
| 89 | + for action_id in permisison_unit_action_ids: |
| 90 | + op.execute( |
| 91 | + f"DELETE FROM role_permission WHERE unit_action_id = {action_id};" # nosec B608 |
| 92 | + ) |
| 93 | + |
| 94 | + op.execute( |
| 95 | + f"DELETE FROM permissions_unit_action WHERE permissions_unit_id = (SELECT id FROM permissions_unit WHERE name = '{UNIT_WS_SUM_REPORTS}');" # nosec B608 |
| 96 | + ) |
| 97 | + |
| 98 | + op.execute( |
| 99 | + f"DELETE FROM permissions_unit WHERE id = {unit_id};" # nosec B608 |
| 100 | + ) |
| 101 | + |
| 102 | + op.execute( |
| 103 | + f"DELETE FROM permissions_group WHERE name = '{GROUP_WS_SUM_REPORTS}';" # nosec B608 |
| 104 | + ) |
0 commit comments