feat: add resource library and real image workflow
This commit is contained in:
118
app/api/schemas/library.py
Normal file
118
app/api/schemas/library.py
Normal file
@@ -0,0 +1,118 @@
|
||||
"""Library resource API schemas."""
|
||||
|
||||
from datetime import datetime
|
||||
|
||||
from pydantic import BaseModel, Field
|
||||
|
||||
from app.domain.enums import LibraryFileRole, LibraryResourceStatus, LibraryResourceType
|
||||
|
||||
|
||||
class PresignUploadRequest(BaseModel):
|
||||
"""Request payload for generating direct-upload metadata."""
|
||||
|
||||
resource_type: LibraryResourceType
|
||||
file_role: LibraryFileRole
|
||||
file_name: str
|
||||
content_type: str
|
||||
|
||||
|
||||
class PresignUploadResponse(BaseModel):
|
||||
"""Response returned when generating direct-upload metadata."""
|
||||
|
||||
method: str
|
||||
upload_url: str
|
||||
headers: dict[str, str] = Field(default_factory=dict)
|
||||
storage_key: str
|
||||
public_url: str
|
||||
|
||||
|
||||
class CreateLibraryResourceFileRequest(BaseModel):
|
||||
"""Metadata for one already-uploaded file."""
|
||||
|
||||
file_role: LibraryFileRole
|
||||
storage_key: str
|
||||
public_url: str
|
||||
mime_type: str
|
||||
size_bytes: int
|
||||
sort_order: int = 0
|
||||
width: int | None = None
|
||||
height: int | None = None
|
||||
|
||||
|
||||
class CreateLibraryResourceRequest(BaseModel):
|
||||
"""One-shot resource creation request."""
|
||||
|
||||
resource_type: LibraryResourceType
|
||||
name: str
|
||||
description: str | None = None
|
||||
tags: list[str] = Field(default_factory=list)
|
||||
gender: str | None = None
|
||||
age_group: str | None = None
|
||||
pose_id: int | None = None
|
||||
environment: str | None = None
|
||||
category: str | None = None
|
||||
files: list[CreateLibraryResourceFileRequest]
|
||||
|
||||
|
||||
class UpdateLibraryResourceRequest(BaseModel):
|
||||
"""Partial update request for a library resource."""
|
||||
|
||||
name: str | None = None
|
||||
description: str | None = None
|
||||
tags: list[str] | None = None
|
||||
gender: str | None = None
|
||||
age_group: str | None = None
|
||||
pose_id: int | None = None
|
||||
environment: str | None = None
|
||||
category: str | None = None
|
||||
cover_file_id: int | None = None
|
||||
|
||||
|
||||
class ArchiveLibraryResourceResponse(BaseModel):
|
||||
"""Response returned after archiving a resource."""
|
||||
|
||||
id: int
|
||||
|
||||
|
||||
class LibraryResourceFileRead(BaseModel):
|
||||
"""Serialized library file metadata."""
|
||||
|
||||
id: int
|
||||
file_role: LibraryFileRole
|
||||
storage_key: str
|
||||
public_url: str
|
||||
bucket: str
|
||||
mime_type: str
|
||||
size_bytes: int
|
||||
sort_order: int
|
||||
width: int | None = None
|
||||
height: int | None = None
|
||||
created_at: datetime
|
||||
|
||||
|
||||
class LibraryResourceRead(BaseModel):
|
||||
"""Serialized library resource."""
|
||||
|
||||
id: int
|
||||
resource_type: LibraryResourceType
|
||||
name: str
|
||||
description: str | None = None
|
||||
tags: list[str]
|
||||
status: LibraryResourceStatus
|
||||
gender: str | None = None
|
||||
age_group: str | None = None
|
||||
pose_id: int | None = None
|
||||
environment: str | None = None
|
||||
category: str | None = None
|
||||
cover_url: str | None = None
|
||||
original_url: str | None = None
|
||||
files: list[LibraryResourceFileRead]
|
||||
created_at: datetime
|
||||
updated_at: datetime
|
||||
|
||||
|
||||
class LibraryResourceListResponse(BaseModel):
|
||||
"""Paginated library resource response."""
|
||||
|
||||
total: int
|
||||
items: list[LibraryResourceRead]
|
||||
@@ -14,9 +14,9 @@ class CreateOrderRequest(BaseModel):
|
||||
customer_level: CustomerLevel
|
||||
service_mode: ServiceMode
|
||||
model_id: int
|
||||
pose_id: int
|
||||
pose_id: int | None = None
|
||||
garment_asset_id: int
|
||||
scene_ref_asset_id: int
|
||||
scene_ref_asset_id: int | None = None
|
||||
|
||||
|
||||
class CreateOrderResponse(BaseModel):
|
||||
@@ -35,9 +35,9 @@ class OrderDetailResponse(BaseModel):
|
||||
service_mode: ServiceMode
|
||||
status: OrderStatus
|
||||
model_id: int
|
||||
pose_id: int
|
||||
pose_id: int | None
|
||||
garment_asset_id: int
|
||||
scene_ref_asset_id: int
|
||||
scene_ref_asset_id: int | None
|
||||
final_asset_id: int | None
|
||||
workflow_id: str | None
|
||||
current_step: WorkflowStepName | None
|
||||
|
||||
Reference in New Issue
Block a user