diff --git a/src/overhaul/router.py b/src/overhaul/router.py index 4271c7a..cb7e4a9 100644 --- a/src/overhaul/router.py +++ b/src/overhaul/router.py @@ -14,7 +14,7 @@ router = APIRouter() async def get_overhaul(db_session: DbSession): """Get all scope pagination.""" overview = get_overhaul_overview() - schedules = await get_overhaul_schedules() + schedules = await get_overhaul_schedules(db_session=db_session) criticalParts = get_overhaul_critical_parts() systemComponents = get_overhaul_system_components() diff --git a/src/overhaul/schema.py b/src/overhaul/schema.py index b2a59bc..4cc6b19 100644 --- a/src/overhaul/schema.py +++ b/src/overhaul/schema.py @@ -5,6 +5,7 @@ from uuid import UUID from pydantic import Field, BaseModel from src.models import DefultBase, Pagination +from src.overhaul_schedule.schema import OverhaulScheduleRead class OverhaulBase(BaseModel): @@ -28,7 +29,7 @@ class OverhaulSystemComponents(OverhaulBase): class OverhaulRead(OverhaulBase): overview: Dict[str, Any] criticalParts: List[str] - schedules: List[Dict[str, Any]] + schedules: List[OverhaulScheduleRead] systemComponents: Dict[str, Any]