From 22d2ccc70724bf6f524d26c66c3c0c7b05001959 Mon Sep 17 00:00:00 2001
From: Cizz22 <cisatraa@gmail.com>
Date: Mon, 23 Jun 2025 16:47:27 +0700
Subject: [PATCH] add new endpoint

---
 src/aeros_equipment/router.py | 16 ++++++++++------
 src/aeros_equipment/schema.py |  4 ++--
 src/api.py                    |  3 ++-
 3 files changed, 14 insertions(+), 9 deletions(-)

diff --git a/src/aeros_equipment/router.py b/src/aeros_equipment/router.py
index 70d8114..58db7fe 100644
--- a/src/aeros_equipment/router.py
+++ b/src/aeros_equipment/router.py
@@ -7,21 +7,25 @@ from src.database.core import DbSession
 from src.database.service import CommonParameters
 from src.models import StandardResponse
 
-from .service import save_default_equipment
+from .schema import EquipmentPagination
+from .service import save_default_equipment, get_all
 
 # from .schema import (OverhaulScheduleCreate, OverhaulSchedulePagination, OverhaulScheduleUpdate)
 
 router = APIRouter()
 
 
-# @router.post("", response_model=StandardResponse[SimulationPagination])
-# async def get_all_simulation(db_session: DbSession, common: CommonParameters):
-#     """Get all simulation."""
+@router.post("", response_model=StandardResponse[EquipmentPagination])
+async def get_all_simulation(db_session: DbSession, common: CommonParameters):
+    """Get all simulation."""
 
-#     results = await get_all(common)
+    results = await get_all(common = common)
 
+    return {"data": results, "status": "success", "message": "Success"}
 
-@router.get("", response_model=StandardResponse[None])
+
+
+@router.get("/save_default", response_model=StandardResponse[None])
 async def save_default_equipments(
     db_session: DbSession, project_name: str = "trialapi"
 ):
diff --git a/src/aeros_equipment/schema.py b/src/aeros_equipment/schema.py
index 75687d1..83df2fb 100644
--- a/src/aeros_equipment/schema.py
+++ b/src/aeros_equipment/schema.py
@@ -42,8 +42,8 @@ class EquipmentBase(DefultBase):
 #     remark: Optional[str]
 
 
-# class OverhaulSchedulePagination(Pagination):
-#     items: List[OverhaulScheduleRead] = []
+class EquipmentPagination(Pagination):
+    items: List[dict] = []
 class FlowrateUnit(str, Enum):
     PER_DAY = "PerDay"
     PER_HOUR = "PerHour"
diff --git a/src/api.py b/src/api.py
index 91aba07..8f77220 100644
--- a/src/api.py
+++ b/src/api.py
@@ -8,7 +8,7 @@ from src.aeros_project.router import router as aeros_project_router
 from src.aeros_simulation.router import router as aeros_simulation_router
 from src.auth.service import JWTBearer
 from src.dashboard_model.router import router as dashboard_model_router
-
+from src.aeros_equipment.router import router as aeros_equipment_router
 
 class ErrorMessage(BaseModel):
     msg: str
@@ -47,6 +47,7 @@ aeros_routes = APIRouter(prefix="/aeros")
 
 aeros_routes.include_router(aeros_simulation_router, prefix="/simulation")
 aeros_routes.include_router(aeros_project_router, prefix="/project")
+aeros_routes.include_router(aeros_equipment_router, prefix="/equipment")
 
 authenticated_api_router.include_router(aeros_routes)