Files
arrr-erp/alembic/versions/20260831_task_tool_result_bridge.py
2026-08-31 15:24:09 +05:30

62 lines
3.7 KiB
Python

"""task tool result bridge
Revision ID: 20260831_task_tool_result_bridge
Revises: 20260830_task_admin_tools
"""
from alembic import op
import sqlalchemy as sa
revision = "20260831_task_tool_result_bridge"
down_revision = "20260830_task_admin_tools"
branch_labels = None
depends_on = None
def upgrade():
op.create_table(
"service_task_tool_runs",
sa.Column("id", sa.Integer(), autoincrement=True, nullable=False),
sa.Column("tenant_id", sa.Integer(), nullable=False),
sa.Column("task_instance_id", sa.Integer(), nullable=False),
sa.Column("client_id", sa.Integer(), nullable=False),
sa.Column("tool_code", sa.String(length=80), nullable=False),
sa.Column("status", sa.String(length=30), server_default="started", nullable=False),
sa.Column("summary", sa.Text(), nullable=True),
sa.Column("result_json", sa.Text(), nullable=True),
sa.Column("exception_count", sa.Integer(), nullable=True),
sa.Column("clean_pass", sa.Boolean(), nullable=True),
sa.Column("output_filename", sa.String(length=255), nullable=True),
sa.Column("output_content_type", sa.String(length=120), nullable=True),
sa.Column("output_bytes", sa.LargeBinary(), nullable=True),
sa.Column("started_by_user_id", sa.Integer(), nullable=True),
sa.Column("completed_by_user_id", sa.Integer(), nullable=True),
sa.Column("started_at_utc", sa.DateTime(timezone=True), nullable=False),
sa.Column("completed_at_utc", sa.DateTime(timezone=True), nullable=True),
sa.Column("created_at_utc", sa.DateTime(timezone=True), nullable=False),
sa.Column("updated_at_utc", sa.DateTime(timezone=True), nullable=False),
sa.ForeignKeyConstraint(["tenant_id"], ["tenants.id"], ondelete="CASCADE"),
sa.ForeignKeyConstraint(["task_instance_id"], ["client_service_task_instances.id"], ondelete="CASCADE"),
sa.ForeignKeyConstraint(["client_id"], ["clients.id"], ondelete="CASCADE"),
sa.ForeignKeyConstraint(["started_by_user_id"], ["users.id"], ondelete="SET NULL"),
sa.ForeignKeyConstraint(["completed_by_user_id"], ["users.id"], ondelete="SET NULL"),
sa.PrimaryKeyConstraint("id"),
)
op.create_index("ix_service_task_tool_runs_tenant_id", "service_task_tool_runs", ["tenant_id"], unique=False)
op.create_index("ix_service_task_tool_runs_task_instance_id", "service_task_tool_runs", ["task_instance_id"], unique=False)
op.create_index("ix_service_task_tool_runs_client_id", "service_task_tool_runs", ["client_id"], unique=False)
op.create_index("ix_service_task_tool_runs_tool_code", "service_task_tool_runs", ["tool_code"], unique=False)
op.create_index("ix_service_task_tool_runs_status", "service_task_tool_runs", ["status"], unique=False)
op.create_index("ix_service_task_tool_runs_started_by_user_id", "service_task_tool_runs", ["started_by_user_id"], unique=False)
op.create_index("ix_service_task_tool_runs_completed_by_user_id", "service_task_tool_runs", ["completed_by_user_id"], unique=False)
def downgrade():
op.drop_index("ix_service_task_tool_runs_completed_by_user_id", table_name="service_task_tool_runs")
op.drop_index("ix_service_task_tool_runs_started_by_user_id", table_name="service_task_tool_runs")
op.drop_index("ix_service_task_tool_runs_status", table_name="service_task_tool_runs")
op.drop_index("ix_service_task_tool_runs_tool_code", table_name="service_task_tool_runs")
op.drop_index("ix_service_task_tool_runs_client_id", table_name="service_task_tool_runs")
op.drop_index("ix_service_task_tool_runs_task_instance_id", table_name="service_task_tool_runs")
op.drop_index("ix_service_task_tool_runs_tenant_id", table_name="service_task_tool_runs")
op.drop_table("service_task_tool_runs")