diff --git a/src/aeros_equipment/router.py b/src/aeros_equipment/router.py index aaa7ba9..004d78c 100644 --- a/src/aeros_equipment/router.py +++ b/src/aeros_equipment/router.py @@ -1,13 +1,15 @@ from typing import List, Optional from fastapi import APIRouter, HTTPException, status +from sqlalchemy import select +from src.aeros_equipment.model import AerosEquipmentCustomParameterData from src.auth.service import CurrentUser from src.database.core import DbSession from src.database.service import CommonParameters from src.models import StandardResponse -from .schema import EquipmentPagination +from .schema import CustomParameter, EquipmentPagination from .service import save_default_equipment, get_all # from .schema import (OverhaulScheduleCreate, OverhaulSchedulePagination, OverhaulScheduleUpdate) @@ -33,3 +35,18 @@ async def save_default_equipments( await save_default_equipment(db_session=db_session, project_name=project_name) return {"data": None, "status": "success", "message": "Success"} + + +@router.get("/parameter-oreda/{location_tag}", response_model=StandardResponse[CustomParameter]) +async def get_parameter_oreda( + db_session: DbSession, location_tag:str +): + query = select(AerosEquipmentCustomParameterData).where(AerosEquipmentCustomParameterData.location_tag == location_tag) + result = await db_session.execute(query) + + + return { + "data" : result.scalars().first(), + "status": "success", + "message": "Success" + } \ No newline at end of file diff --git a/src/aeros_simulation/router.py b/src/aeros_simulation/router.py index 2645aff..1fd4528 100644 --- a/src/aeros_simulation/router.py +++ b/src/aeros_simulation/router.py @@ -313,7 +313,7 @@ async def get_custom_parameters_controller(db_session: DbSession): @router.get("/metrics/{simulation_id}", response_model=StandardResponse[list]) -async def get_custom_parameters_controller(db_session: DbSession, simulation_id:UUID): +async def get_metric_controller(db_session: DbSession, simulation_id:UUID): """Get simulation result.""" results = await calculate_plant_eaf(db_session, simulation_id, 0, 1200, 15000, 0)