From 70c965cdee681e72aadfe62634e60d4f9c05e6c3 Mon Sep 17 00:00:00 2001 From: Cizz22 Date: Mon, 23 Dec 2024 14:53:59 +0700 Subject: [PATCH] fix overhaul overview --- src/overhaul/router.py | 2 +- src/overhaul/schema.py | 3 ++- 2 files changed, 3 insertions(+), 2 deletions(-) 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]