feat: add resource library and real image workflow
This commit is contained in:
89
app/api/routers/library.py
Normal file
89
app/api/routers/library.py
Normal file
@@ -0,0 +1,89 @@
|
||||
"""Library resource routes."""
|
||||
|
||||
from fastapi import APIRouter, Depends, Query, status
|
||||
from sqlalchemy.ext.asyncio import AsyncSession
|
||||
|
||||
from app.api.schemas.library import (
|
||||
ArchiveLibraryResourceResponse,
|
||||
CreateLibraryResourceRequest,
|
||||
LibraryResourceListResponse,
|
||||
LibraryResourceRead,
|
||||
PresignUploadRequest,
|
||||
PresignUploadResponse,
|
||||
UpdateLibraryResourceRequest,
|
||||
)
|
||||
from app.application.services.library_service import LibraryService
|
||||
from app.domain.enums import LibraryResourceType
|
||||
from app.infra.db.session import get_db_session
|
||||
|
||||
router = APIRouter(prefix="/library", tags=["library"])
|
||||
library_service = LibraryService()
|
||||
|
||||
|
||||
@router.post("/uploads/presign", response_model=PresignUploadResponse)
|
||||
async def create_library_upload_presign(payload: PresignUploadRequest) -> PresignUploadResponse:
|
||||
"""Create direct-upload metadata for a library file."""
|
||||
|
||||
return library_service.create_upload_presign(
|
||||
resource_type=payload.resource_type,
|
||||
file_name=payload.file_name,
|
||||
content_type=payload.content_type,
|
||||
)
|
||||
|
||||
|
||||
@router.post("/resources", response_model=LibraryResourceRead, status_code=status.HTTP_201_CREATED)
|
||||
async def create_library_resource(
|
||||
payload: CreateLibraryResourceRequest,
|
||||
session: AsyncSession = Depends(get_db_session),
|
||||
) -> LibraryResourceRead:
|
||||
"""Create a library resource with already-uploaded file metadata."""
|
||||
|
||||
return await library_service.create_resource(session, payload)
|
||||
|
||||
|
||||
@router.get("/resources", response_model=LibraryResourceListResponse)
|
||||
async def list_library_resources(
|
||||
resource_type: LibraryResourceType | None = Query(default=None),
|
||||
query: str | None = Query(default=None, min_length=1),
|
||||
gender: str | None = Query(default=None),
|
||||
age_group: str | None = Query(default=None),
|
||||
environment: str | None = Query(default=None),
|
||||
category: str | None = Query(default=None),
|
||||
page: int = Query(default=1, ge=1),
|
||||
limit: int = Query(default=20, ge=1, le=100),
|
||||
session: AsyncSession = Depends(get_db_session),
|
||||
) -> LibraryResourceListResponse:
|
||||
"""List library resources with basic filtering."""
|
||||
|
||||
return await library_service.list_resources(
|
||||
session,
|
||||
resource_type=resource_type,
|
||||
query=query,
|
||||
gender=gender,
|
||||
age_group=age_group,
|
||||
environment=environment,
|
||||
category=category,
|
||||
page=page,
|
||||
limit=limit,
|
||||
)
|
||||
|
||||
|
||||
@router.patch("/resources/{resource_id}", response_model=LibraryResourceRead)
|
||||
async def update_library_resource(
|
||||
resource_id: int,
|
||||
payload: UpdateLibraryResourceRequest,
|
||||
session: AsyncSession = Depends(get_db_session),
|
||||
) -> LibraryResourceRead:
|
||||
"""Update editable metadata on a library resource."""
|
||||
|
||||
return await library_service.update_resource(session, resource_id, payload)
|
||||
|
||||
|
||||
@router.delete("/resources/{resource_id}", response_model=ArchiveLibraryResourceResponse)
|
||||
async def archive_library_resource(
|
||||
resource_id: int,
|
||||
session: AsyncSession = Depends(get_db_session),
|
||||
) -> ArchiveLibraryResourceResponse:
|
||||
"""Archive a library resource instead of hard deleting it."""
|
||||
|
||||
return await library_service.archive_resource(session, resource_id)
|
||||
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