diff --git a/src/aeros_equipment/router.py b/src/aeros_equipment/router.py index 004d78c..9831f47 100644 --- a/src/aeros_equipment/router.py +++ b/src/aeros_equipment/router.py @@ -41,12 +41,17 @@ async def save_default_equipments( async def get_parameter_oreda( db_session: DbSession, location_tag:str ): - query = select(AerosEquipmentCustomParameterData).where(AerosEquipmentCustomParameterData.location_tag == location_tag) + query = select(AerosEquipmentCustomParameterData).where(AerosEquipmentCustomParameterData.location_tag == location_tag).where(AerosEquipmentCustomParameterData.level == "Critical") result = await db_session.execute(query) + + res = result.scalars().first() + + if not res: + raise HTTPException(status_code=404, detail="Data not found") return { - "data" : result.scalars().first(), + "data" : , "status": "success", "message": "Success" } \ No newline at end of file