mirror of
https://github.com/comfyanonymous/ComfyUI.git
synced 2025-12-17 01:52:59 +08:00
74 lines
2.4 KiB
Python
74 lines
2.4 KiB
Python
from typing import Optional
|
|
|
|
from aiohttp import web
|
|
from pydantic import ValidationError
|
|
|
|
from .. import assets_manager
|
|
from .schemas_in import ListAssetsQuery, UpdateAssetBody
|
|
|
|
|
|
ROUTES = web.RouteTableDef()
|
|
|
|
|
|
@ROUTES.get("/api/assets")
|
|
async def list_assets(request: web.Request) -> web.Response:
|
|
query_dict = dict(request.rel_url.query)
|
|
|
|
try:
|
|
q = ListAssetsQuery.model_validate(query_dict)
|
|
except ValidationError as ve:
|
|
return _validation_error_response("INVALID_QUERY", ve)
|
|
|
|
payload = await assets_manager.list_assets(
|
|
include_tags=q.include_tags,
|
|
exclude_tags=q.exclude_tags,
|
|
name_contains=q.name_contains,
|
|
metadata_filter=q.metadata_filter,
|
|
limit=q.limit,
|
|
offset=q.offset,
|
|
sort=q.sort,
|
|
order=q.order,
|
|
)
|
|
return web.json_response(payload)
|
|
|
|
|
|
@ROUTES.put("/api/assets/{id}")
|
|
async def update_asset(request: web.Request) -> web.Response:
|
|
asset_info_id_raw = request.match_info.get("id")
|
|
try:
|
|
asset_info_id = int(asset_info_id_raw)
|
|
except Exception:
|
|
return _error_response(400, "INVALID_ID", f"AssetInfo id '{asset_info_id_raw}' is not a valid integer.")
|
|
|
|
try:
|
|
body = UpdateAssetBody.model_validate(await request.json())
|
|
except ValidationError as ve:
|
|
return _validation_error_response("INVALID_BODY", ve)
|
|
except Exception:
|
|
return _error_response(400, "INVALID_JSON", "Request body must be valid JSON.")
|
|
|
|
try:
|
|
result = await assets_manager.update_asset(
|
|
asset_info_id=asset_info_id,
|
|
name=body.name,
|
|
tags=body.tags,
|
|
user_metadata=body.user_metadata,
|
|
)
|
|
except ValueError as ve:
|
|
return _error_response(404, "ASSET_NOT_FOUND", str(ve), {"id": asset_info_id})
|
|
except Exception:
|
|
return _error_response(500, "INTERNAL", "Unexpected server error.")
|
|
return web.json_response(result, status=200)
|
|
|
|
|
|
def register_assets_routes(app: web.Application) -> None:
|
|
app.add_routes(ROUTES)
|
|
|
|
|
|
def _error_response(status: int, code: str, message: str, details: Optional[dict] = None) -> web.Response:
|
|
return web.json_response({"error": {"code": code, "message": message, "details": details or {}}}, status=status)
|
|
|
|
|
|
def _validation_error_response(code: str, ve: ValidationError) -> web.Response:
|
|
return _error_response(400, code, "Validation failed.", {"errors": ve.errors()})
|