cayden/auto-virtual-tryon
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Implement FastAPI Temporal MVP pipeline

Browse Source
This commit is contained in:
Codex
2026-03-27 00:10:28 +08:00
commit cc03da8a94
52 changed files with 3619 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

214
alembic/versions/20260326_0001_initial.py Normal file
View File

@@ -0,0 +1,214 @@
"""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")