45 lines
1.8 KiB
Python
45 lines
1.8 KiB
Python
from typing import List, Any, Coroutine
|
|
|
|
from fastapi import APIRouter, Depends
|
|
from pydantic import BaseModel
|
|
from starlette.exceptions import HTTPException
|
|
from starlette.requests import Request
|
|
|
|
from api.models.AssetDTO import AssetsResponse
|
|
from api.models.GenerationRequest import GenerationRequest, GenerationResponse
|
|
from models.Asset import Asset
|
|
from models.Character import Character
|
|
from repos.dao import DAO
|
|
from api.dependency import get_dao
|
|
|
|
router = APIRouter(prefix="/api/characters", tags=["Characters"])
|
|
|
|
|
|
@router.get("/", response_model=List[Character])
|
|
async def get_characters(request: Request, dao: DAO = Depends(get_dao), ) -> List[Character]:
|
|
characters = await dao.chars.get_all_characters()
|
|
return characters
|
|
|
|
|
|
@router.get("/{character_id}/assets", response_model=AssetsResponse)
|
|
async def get_character_assets(character_id: str, dao: DAO = Depends(get_dao), limit: int = 10,
|
|
offset: int = 0, ) -> AssetsResponse:
|
|
character = await dao.chars.get_character(character_id)
|
|
if character is None:
|
|
raise HTTPException(status_code=404, detail="Character not found")
|
|
assets = await dao.assets.get_assets_by_char_id(character_id, limit, offset)
|
|
total_count = await dao.assets.get_asset_count(character_id)
|
|
return AssetsResponse(assets=assets, total_count=total_count)
|
|
|
|
|
|
@router.get("/{character_id}", response_model=Character)
|
|
async def get_character_by_id(character_id: str, request: Request, dao: DAO = Depends(get_dao)) -> Character:
|
|
character = await dao.chars.get_character(character_id)
|
|
return character
|
|
|
|
|
|
@router.post("/{character_id}/_run", response_model=Asset)
|
|
async def post_character_generation(character_id: str, generation: GenerationRequest,
|
|
request: Request) -> GenerationResponse:
|
|
generation_service = request.app.state.generation_service
|