refactor: get equipment by assetnum

src/equipment/schema.py

@@ -1,4 +1,3 @@
-
 from datetime import datetime
 from typing import List, Optional
 from uuid import UUID
@@ -30,6 +29,8 @@ class EquipmentUpdate(EquipmentBase):
 
 class EquipmentRead(EquipmentBase):
     id: UUID
+    equipment_id: str
+
 
 class EquipmentPagination(Pagination):
     items: List[EquipmentRead] = []

src/equipment/service.py

@@ -1,5 +1,3 @@
-
-
 from sqlalchemy import Select, Delete
 from .model import Equipment
 from .schema import EquipmentCreate, EquipmentUpdate
@@ -11,7 +9,7 @@ from src.auth.service import CurrentUser
 
 async def get(*, db_session: DbSession, equipment_id: str) -> Optional[Equipment]:
     """Returns a document based on the given document id."""
-    query = Select(Equipment).filter(Equipment.id == equipment_id)
+    query = Select(Equipment).filter(Equipment.equipment_id == equipment_id)
     result = await db_session.execute(query)
     return result.scalars().one_or_none()
 
@@ -31,7 +29,9 @@ async def create(*, db_session: DbSession, equipment_in: EquipmentCreate):
     return equipment
 
 
-async def update(*, db_session: DbSession, equipment: Equipment, equipment_in: EquipmentUpdate):
+async def update(
+    *, db_session: DbSession, equipment: Equipment, equipment_in: EquipmentUpdate
+):
     """Updates a document."""
     data = equipment_in.model_dump()