AiDA_Python/app/schemas/design.py

from pydantic import BaseModel


# class BodyPointModel(BaseModel):
#     waistband_right: list[int]
#     hand_point_right: list[int]
#     waistband_left: list[int]
#     hand_point_left: list[int]
#     shoulder_left: list[int]
#     shoulder_right: list[int]
#
#
# class BasicModel(BaseModel):
#     body_point: BodyPointModel
#     layer_order: bool
#     scale_bag: float
#     scale_earrings: float
#     self_template: bool
#     single_overall: str
#     switch_category: str
#     body_path: str
#
#
# class PrintModel(BaseModel):
#     if_single: bool
#     print_path_list: list[str]
#
#
# class ItemModel(BaseModel):
#     color: str
#     image_id: str
#     offset: list[int]
#     path: str
#     print: PrintModel
#     resize_scale: float
#     type: str
#
#
# class CollocationModel(BaseModel):
#     basic: BasicModel
#     item: list[ItemModel]
#
#
# class DesignModel(BaseModel):
#     object: list[CollocationModel]
#     process_id: str

class DesignModel(BaseModel):
    objects: list[dict]
    process_id: str


class DesignProgressModel(BaseModel):
    process_id: str