| 123456789101112131415161718192021222324252627282930313233343536373839404142434445464748 |
- """Modifying case tasks to remove assignee id for instead, adding a table named task_assignee
- Revision ID: 875edc4adb40
- Revises: fcc375ed37d1
- Create Date: 2022-07-17 14:57:22.809977
- """
- from alembic import op
- from sqlalchemy import text
- from app.alembic.alembic_utils import _has_table
- from app.alembic.alembic_utils import _table_has_column
- # revision identifiers, used by Alembic.
- revision = '875edc4adb40'
- down_revision = 'fcc375ed37d1'
- branch_labels = None
- depends_on = None
- def upgrade():
- conn = op.get_bind()
- # Get all users with their roles
- if _has_table("case_tasks"):
- if _table_has_column("case_tasks", "task_assignee_id"):
- res = conn.execute(text("select id, task_assignee_id from case_tasks"))
- results_tasks = res.fetchall()
- for task in results_tasks:
- task_id = task[0]
- user_id = task[1]
- if not user_id:
- user_id = 1
- # Migrate assignees to task_assignee
- conn.execute(text(f"insert into task_assignee (user_id, task_id) values ({user_id}, {task_id}) "
- f"on conflict do nothing;"))
- op.drop_column(
- table_name='case_tasks',
- column_name='task_assignee_id'
- )
- def downgrade():
- pass
|
