""" Router "aggregate" provides: - listing aggregate calendars - finding aggregate calendars by name prefix - getting aggregate calendar events by name prefix """ from logging import getLogger from typing import Iterator from fastapi import APIRouter, Depends from ovdashboard_api.config import Config from ...dav_calendar import CalEvent, DavCalendar from ._common import AggregateNameLister, PrefixFinder, PrefixUnique from .calendar import calendar_unique _logger = getLogger(__name__) router = APIRouter(prefix="/aggregate", tags=["calendar"]) aggregate_lister = AggregateNameLister() aggregate_finder = PrefixFinder(aggregate_lister) aggregate_unique = PrefixUnique(aggregate_finder) @router.on_event("startup") async def start_router() -> None: _logger.debug(f"{router.prefix} router starting.") @router.get("/list", response_model=list[str]) async def list_aggregate_calendars( names: Iterator[str] = Depends(aggregate_lister), ) -> list[str]: return list(names) @router.get("/find/{prefix}", response_model=list[str]) async def find_aggregate_calendars( names: Iterator[str] = Depends(aggregate_finder), ) -> list[str]: return list(names) @router.get("/get/{prefix}", response_model=list[CalEvent]) async def get_aggregate_calendar( name: str = Depends(aggregate_unique), ) -> list[CalEvent]: cfg = await Config.get() aggregate = cfg.calendar.aggregates[name] calendars = ( DavCalendar(await calendar_unique(cal_prefix)) for cal_prefix in aggregate ) return sorted([ event async for calendar in calendars # type: ignore for event in (await calendar.events) ])