215 lines
7.2 KiB
Python
215 lines
7.2 KiB
Python
"""initial schema
|
|
|
|
Revision ID: 20260326_0001
|
|
Revises:
|
|
Create Date: 2026-03-26 23:59:00.000000
|
|
"""
|
|
|
|
from collections.abc import Sequence
|
|
|
|
from alembic import op
|
|
import sqlalchemy as sa
|
|
|
|
|
|
revision: str = "20260326_0001"
|
|
down_revision: str | None = None
|
|
branch_labels: Sequence[str] | None = None
|
|
depends_on: Sequence[str] | None = None
|
|
|
|
|
|
def upgrade() -> None:
|
|
"""Create the initial application tables."""
|
|
|
|
op.create_table(
|
|
"orders",
|
|
sa.Column("id", sa.Integer(), primary_key=True, autoincrement=True),
|
|
sa.Column("customer_level", sa.Enum("low", "mid", name="customerlevel", native_enum=False), nullable=False),
|
|
sa.Column("service_mode", sa.Enum("auto_basic", "semi_pro", name="servicemode", native_enum=False), nullable=False),
|
|
sa.Column(
|
|
"status",
|
|
sa.Enum(
|
|
"created",
|
|
"running",
|
|
"waiting_review",
|
|
"succeeded",
|
|
"failed",
|
|
"cancelled",
|
|
name="orderstatus",
|
|
native_enum=False,
|
|
),
|
|
nullable=False,
|
|
),
|
|
sa.Column("model_id", sa.Integer(), nullable=False),
|
|
sa.Column("pose_id", sa.Integer(), nullable=False),
|
|
sa.Column("garment_asset_id", sa.Integer(), nullable=False),
|
|
sa.Column("scene_ref_asset_id", sa.Integer(), nullable=False),
|
|
sa.Column("final_asset_id", sa.Integer(), nullable=True),
|
|
sa.Column("created_at", sa.DateTime(timezone=True), nullable=False),
|
|
sa.Column("updated_at", sa.DateTime(timezone=True), nullable=False),
|
|
)
|
|
|
|
op.create_table(
|
|
"assets",
|
|
sa.Column("id", sa.Integer(), primary_key=True, autoincrement=True),
|
|
sa.Column("order_id", sa.Integer(), sa.ForeignKey("orders.id"), nullable=False),
|
|
sa.Column(
|
|
"asset_type",
|
|
sa.Enum(
|
|
"prepared_model",
|
|
"tryon",
|
|
"scene",
|
|
"texture",
|
|
"face",
|
|
"fusion",
|
|
"qc_candidate",
|
|
"final",
|
|
name="assettype",
|
|
native_enum=False,
|
|
),
|
|
nullable=False,
|
|
),
|
|
sa.Column(
|
|
"step_name",
|
|
sa.Enum(
|
|
"prepare_model",
|
|
"tryon",
|
|
"scene",
|
|
"texture",
|
|
"face",
|
|
"fusion",
|
|
"qc",
|
|
"export",
|
|
"review",
|
|
name="workflowstepname",
|
|
native_enum=False,
|
|
),
|
|
nullable=True,
|
|
),
|
|
sa.Column("uri", sa.String(length=500), nullable=False),
|
|
sa.Column("metadata_json", sa.JSON(), nullable=True),
|
|
sa.Column("created_at", sa.DateTime(timezone=True), nullable=False),
|
|
sa.Column("updated_at", sa.DateTime(timezone=True), nullable=False),
|
|
)
|
|
op.create_index("ix_assets_order_id", "assets", ["order_id"])
|
|
|
|
op.create_table(
|
|
"review_tasks",
|
|
sa.Column("id", sa.Integer(), primary_key=True, autoincrement=True),
|
|
sa.Column("order_id", sa.Integer(), sa.ForeignKey("orders.id"), nullable=False),
|
|
sa.Column(
|
|
"status",
|
|
sa.Enum("pending", "submitted", name="reviewtaskstatus", native_enum=False),
|
|
nullable=False,
|
|
),
|
|
sa.Column(
|
|
"decision",
|
|
sa.Enum(
|
|
"approve",
|
|
"rerun_scene",
|
|
"rerun_face",
|
|
"rerun_fusion",
|
|
"reject",
|
|
name="reviewdecision",
|
|
native_enum=False,
|
|
),
|
|
nullable=True,
|
|
),
|
|
sa.Column("reviewer_id", sa.Integer(), nullable=True),
|
|
sa.Column("selected_asset_id", sa.Integer(), nullable=True),
|
|
sa.Column("comment", sa.Text(), nullable=True),
|
|
sa.Column("created_at", sa.DateTime(timezone=True), nullable=False),
|
|
sa.Column("updated_at", sa.DateTime(timezone=True), nullable=False),
|
|
)
|
|
op.create_index("ix_review_tasks_order_id", "review_tasks", ["order_id"])
|
|
|
|
op.create_table(
|
|
"workflow_runs",
|
|
sa.Column("id", sa.Integer(), primary_key=True, autoincrement=True),
|
|
sa.Column("order_id", sa.Integer(), sa.ForeignKey("orders.id"), nullable=False),
|
|
sa.Column("workflow_id", sa.String(length=255), nullable=False),
|
|
sa.Column("workflow_type", sa.String(length=255), nullable=False),
|
|
sa.Column(
|
|
"status",
|
|
sa.Enum(
|
|
"created",
|
|
"running",
|
|
"waiting_review",
|
|
"succeeded",
|
|
"failed",
|
|
"cancelled",
|
|
name="orderstatus",
|
|
native_enum=False,
|
|
),
|
|
nullable=False,
|
|
),
|
|
sa.Column(
|
|
"current_step",
|
|
sa.Enum(
|
|
"prepare_model",
|
|
"tryon",
|
|
"scene",
|
|
"texture",
|
|
"face",
|
|
"fusion",
|
|
"qc",
|
|
"export",
|
|
"review",
|
|
name="workflowstepname",
|
|
native_enum=False,
|
|
),
|
|
nullable=True,
|
|
),
|
|
sa.Column("created_at", sa.DateTime(timezone=True), nullable=False),
|
|
sa.Column("updated_at", sa.DateTime(timezone=True), nullable=False),
|
|
sa.UniqueConstraint("workflow_id"),
|
|
)
|
|
op.create_index("ix_workflow_runs_order_id", "workflow_runs", ["order_id"])
|
|
|
|
op.create_table(
|
|
"workflow_steps",
|
|
sa.Column("id", sa.Integer(), primary_key=True, autoincrement=True),
|
|
sa.Column("workflow_run_id", sa.Integer(), sa.ForeignKey("workflow_runs.id"), nullable=False),
|
|
sa.Column(
|
|
"step_name",
|
|
sa.Enum(
|
|
"prepare_model",
|
|
"tryon",
|
|
"scene",
|
|
"texture",
|
|
"face",
|
|
"fusion",
|
|
"qc",
|
|
"export",
|
|
"review",
|
|
name="workflowstepname",
|
|
native_enum=False,
|
|
),
|
|
nullable=False,
|
|
),
|
|
sa.Column(
|
|
"step_status",
|
|
sa.Enum("pending", "running", "waiting", "succeeded", "failed", name="stepstatus", native_enum=False),
|
|
nullable=False,
|
|
),
|
|
sa.Column("input_json", sa.JSON(), nullable=True),
|
|
sa.Column("output_json", sa.JSON(), nullable=True),
|
|
sa.Column("error_message", sa.Text(), nullable=True),
|
|
sa.Column("started_at", sa.DateTime(timezone=True), nullable=False),
|
|
sa.Column("ended_at", sa.DateTime(timezone=True), nullable=True),
|
|
)
|
|
op.create_index("ix_workflow_steps_workflow_run_id", "workflow_steps", ["workflow_run_id"])
|
|
|
|
|
|
def downgrade() -> None:
|
|
"""Drop the application tables."""
|
|
|
|
op.drop_index("ix_workflow_steps_workflow_run_id", table_name="workflow_steps")
|
|
op.drop_table("workflow_steps")
|
|
op.drop_index("ix_workflow_runs_order_id", table_name="workflow_runs")
|
|
op.drop_table("workflow_runs")
|
|
op.drop_index("ix_review_tasks_order_id", table_name="review_tasks")
|
|
op.drop_table("review_tasks")
|
|
op.drop_index("ix_assets_order_id", table_name="assets")
|
|
op.drop_table("assets")
|
|
op.drop_table("orders")
|