From 78dfbd3227ea1979ee9c29bb04a63bd84f89cb8c Mon Sep 17 00:00:00 2001 From: Cizz22 Date: Thu, 16 Oct 2025 09:28:40 +0700 Subject: [PATCH] fix --- src/overhaul/router.py | 6 +++--- src/overhaul/service.py | 7 ++++--- 2 files changed, 7 insertions(+), 6 deletions(-) diff --git a/src/overhaul/router.py b/src/overhaul/router.py index 42158cf..dbe0621 100644 --- a/src/overhaul/router.py +++ b/src/overhaul/router.py @@ -3,7 +3,7 @@ from typing import List from fastapi import APIRouter, HTTPException, status from src.auth.service import Token -from src.database.core import DbSession +from src.database.core import CollectorDbSession, DbSession from src.models import StandardResponse from src.overhaul.service import (get_overhaul_critical_parts, get_overhaul_overview, @@ -18,11 +18,11 @@ router = APIRouter() @router.get("", response_model=StandardResponse[OverhaulRead]) -async def get_overhaul(db_session: DbSession, token:Token): +async def get_overhaul(db_session: DbSession, token:Token, collector_db_session:CollectorDbSession): """Get all scope pagination.""" overview = await get_overhaul_overview(db_session=db_session) schedules = await get_overhaul_schedules(db_session=db_session) - criticalParts = await get_overhaul_critical_parts(db_session=db_session, session_id=overview["overhaul"]["id"], token=token) + criticalParts = await get_overhaul_critical_parts(db_session=db_session, session_id=overview["overhaul"]["id"], token=token, collector_db_session=collector_db_session) systemComponents = get_overhaul_system_components() return StandardResponse( diff --git a/src/overhaul/service.py b/src/overhaul/service.py index d847278..db71621 100644 --- a/src/overhaul/service.py +++ b/src/overhaul/service.py @@ -55,11 +55,12 @@ async def get_simulation_results(*, simulation_id: str, token: str): "plant_result": plant_data } -async def get_overhaul_critical_parts(db_session, session_id, token): +async def get_overhaul_critical_parts(db_session, session_id, token, collector_db_session): """Get all overhaul critical parts.""" - equipments, _ = await get_by_oh_session_id( + equipments = await get_standard_scope_by_session_id( db_session=db_session, - oh_session_id=session_id, + overhaul_session_id=session_id, + collector_db=collector_db_session )