A R R R Associates
48 lines
2.9 KiB
Python
48 lines
2.9 KiB
Python
"""add structured checklist fields to service tasks
|
|
|
|
Revision ID: 20260719_task_checklist_fields
|
|
Revises: 20260713_bank_stmt_queue
|
|
"""
|
|
from alembic import op
|
|
import sqlalchemy as sa
|
|
|
|
revision = "20260719_task_checklist_fields"
|
|
down_revision = "20260713_bank_stmt_queue"
|
|
branch_labels = None
|
|
depends_on = None
|
|
|
|
_TEMPLATE_TABLES = ("service_default_task_templates", "firm_service_task_templates")
|
|
|
|
def upgrade():
|
|
for table in _TEMPLATE_TABLES:
|
|
op.add_column(table, sa.Column("task_category", sa.String(length=100), nullable=True))
|
|
op.add_column(table, sa.Column("response_required", sa.Boolean(), nullable=False, server_default=sa.false()))
|
|
op.add_column(table, sa.Column("response_type", sa.String(length=20), nullable=False, server_default="NONE"))
|
|
op.add_column(table, sa.Column("evidence_required", sa.Boolean(), nullable=False, server_default=sa.false()))
|
|
op.add_column(table, sa.Column("remarks_required_if_no", sa.Boolean(), nullable=False, server_default=sa.false()))
|
|
op.create_index(f"ix_{table}_task_category", table, ["task_category"], unique=False)
|
|
|
|
table = "client_service_task_instances"
|
|
op.add_column(table, sa.Column("task_category", sa.String(length=100), nullable=True))
|
|
op.add_column(table, sa.Column("response_required", sa.Boolean(), nullable=False, server_default=sa.false()))
|
|
op.add_column(table, sa.Column("response_type", sa.String(length=20), nullable=False, server_default="NONE"))
|
|
op.add_column(table, sa.Column("evidence_required", sa.Boolean(), nullable=False, server_default=sa.false()))
|
|
op.add_column(table, sa.Column("remarks_required_if_no", sa.Boolean(), nullable=False, server_default=sa.false()))
|
|
op.add_column(table, sa.Column("checklist_response", sa.String(length=10), nullable=True))
|
|
op.add_column(table, sa.Column("checklist_text_response", sa.Text(), nullable=True))
|
|
op.add_column(table, sa.Column("checklist_number_response", sa.Numeric(18, 2), nullable=True))
|
|
op.add_column(table, sa.Column("checklist_date_response", sa.Date(), nullable=True))
|
|
op.add_column(table, sa.Column("checklist_remarks", sa.Text(), nullable=True))
|
|
op.create_index("ix_client_service_task_instances_task_category", table, ["task_category"], unique=False)
|
|
|
|
def downgrade():
|
|
table = "client_service_task_instances"
|
|
op.drop_index("ix_client_service_task_instances_task_category", table_name=table)
|
|
for column in ("checklist_remarks", "checklist_date_response", "checklist_number_response", "checklist_text_response", "checklist_response", "remarks_required_if_no", "evidence_required", "response_type", "response_required", "task_category"):
|
|
op.drop_column(table, column)
|
|
|
|
for table in reversed(_TEMPLATE_TABLES):
|
|
op.drop_index(f"ix_{table}_task_category", table_name=table)
|
|
for column in ("remarks_required_if_no", "evidence_required", "response_type", "response_required", "task_category"):
|
|
op.drop_column(table, column)
|