wehub-resource-sync
cddb07a176
docs / deploy (push) Has been cancelled
docs / changes (push) Has been cancelled
docs / check-and-build (push) Has been cancelled
build container image / cpu (push) Has been cancelled
build container image / cuda (push) Has been cancelled
build container image / rocm (push) Has been cancelled
frontend checks / frontend-checks (push) Has been cancelled
frontend tests / frontend-tests (push) Has been cancelled
lfs checks / lfs-check (push) Has been cancelled
python checks / python-checks (push) Has been cancelled
python tests / py3.12: macos-default (push) Has been cancelled
python tests / py3.11: windows-cpu (push) Has been cancelled
python tests / py3.12: windows-cpu (push) Has been cancelled
python tests / py3.11: linux-cpu (push) Has been cancelled
typegen checks / typegen-checks (push) Has been cancelled
uv lock checks / uv-lock-checks (push) Has been cancelled
openapi checks / openapi-checks (push) Has been cancelled
python tests / py3.11: macos-default (push) Has been cancelled
python tests / py3.12: linux-cpu (push) Has been cancelled
57 lines
2.4 KiB
Python
57 lines
2.4 KiB
Python
from fastapi import HTTPException, Path, Query
|
|
from fastapi.routing import APIRouter
|
|
|
|
from invokeai.app.api.auth_dependencies import CurrentUserOrDefault
|
|
from invokeai.app.api.dependencies import ApiDependencies
|
|
from invokeai.app.services.image_records.image_records_common import ImageCategory, ImageNamesResult
|
|
from invokeai.app.services.shared.sqlite.sqlite_common import SQLiteDirection
|
|
from invokeai.app.services.virtual_boards.virtual_boards_common import VirtualSubBoardDTO
|
|
|
|
virtual_boards_router = APIRouter(prefix="/v1/virtual_boards", tags=["virtual_boards"])
|
|
|
|
|
|
@virtual_boards_router.get(
|
|
"/by_date",
|
|
operation_id="list_virtual_boards_by_date",
|
|
response_model=list[VirtualSubBoardDTO],
|
|
)
|
|
async def list_virtual_boards_by_date(
|
|
current_user: CurrentUserOrDefault,
|
|
) -> list[VirtualSubBoardDTO]:
|
|
"""Gets a list of virtual sub-boards grouped by date."""
|
|
try:
|
|
return ApiDependencies.invoker.services.image_records.get_image_dates(
|
|
user_id=current_user.user_id,
|
|
is_admin=current_user.is_admin,
|
|
)
|
|
except Exception:
|
|
raise HTTPException(status_code=500, detail="Failed to get virtual boards by date")
|
|
|
|
|
|
@virtual_boards_router.get(
|
|
"/by_date/{date}/image_names",
|
|
operation_id="list_virtual_board_image_names_by_date",
|
|
response_model=ImageNamesResult,
|
|
)
|
|
async def list_virtual_board_image_names_by_date(
|
|
current_user: CurrentUserOrDefault,
|
|
date: str = Path(description="The ISO date string, e.g. '2026-03-18'"),
|
|
starred_first: bool = Query(default=True, description="Whether to sort starred images first"),
|
|
order_dir: SQLiteDirection = Query(default=SQLiteDirection.Descending, description="The sort direction"),
|
|
categories: list[ImageCategory] | None = Query(default=None, description="The categories of images to include"),
|
|
search_term: str | None = Query(default=None, description="Search term to filter images"),
|
|
) -> ImageNamesResult:
|
|
"""Gets ordered image names for a specific date."""
|
|
try:
|
|
return ApiDependencies.invoker.services.image_records.get_image_names_by_date(
|
|
date=date,
|
|
starred_first=starred_first,
|
|
order_dir=order_dir,
|
|
categories=categories,
|
|
search_term=search_term,
|
|
user_id=current_user.user_id,
|
|
is_admin=current_user.is_admin,
|
|
)
|
|
except Exception:
|
|
raise HTTPException(status_code=500, detail="Failed to get image names for date")
|