feat: add resource library and real image workflow

This commit is contained in:
afei A
2026-03-29 00:24:29 +08:00
parent eeaff269eb
commit 04da401ab4
38 changed files with 3033 additions and 117 deletions

View 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
View 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]

View File

@@ -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