6be831edfda7_add_investigation_roles_permissions.py 1.2 KB

1234567891011121314151617181920212223242526272829303132333435
  1. """add investigation roles permissions
  2. Revision ID: 6be831edfda7
  3. Revises: c82bf6af92e5
  4. Create Date: 2025-09-17 22:02:46.159090
  5. """
  6. from typing import Sequence, Union
  7. from alembic import op
  8. import sqlalchemy as sa
  9. from sqlalchemy.dialects import postgresql
  10. # revision identifiers, used by Alembic.
  11. revision: str = '6be831edfda7'
  12. down_revision: Union[str, None] = 'c82bf6af92e5'
  13. branch_labels: Union[str, Sequence[str], None] = None
  14. depends_on: Union[str, Sequence[str], None] = None
  15. def upgrade() -> None:
  16. """Upgrade schema."""
  17. # ### commands auto generated by Alembic - please adjust! ###
  18. op.add_column('investigation_user_roles', sa.Column('roles', postgresql.ARRAY(sa.Enum('OWNER', 'EDITOR', 'VIEWER', name='role_enum', create_constraint=True)), server_default='{}', nullable=False))
  19. op.drop_column('investigation_user_roles', 'role')
  20. # ### end Alembic commands ###
  21. def downgrade() -> None:
  22. """Downgrade schema."""
  23. # ### commands auto generated by Alembic - please adjust! ###
  24. op.add_column('investigation_user_roles', sa.Column('role', postgresql.ENUM('OWNER', 'EDITOR', 'VIEWER', name='role_enum'), autoincrement=False, nullable=False))
  25. op.drop_column('investigation_user_roles', 'roles')
  26. # ### end Alembic commands ###