diff --git a/src/equipment/__pycache__/schema.cpython-311.pyc b/src/equipment/__pycache__/schema.cpython-311.pyc index 8d54480..ea19e43 100644 Binary files a/src/equipment/__pycache__/schema.cpython-311.pyc and b/src/equipment/__pycache__/schema.cpython-311.pyc differ diff --git a/src/equipment/__pycache__/service.cpython-311.pyc b/src/equipment/__pycache__/service.cpython-311.pyc index d576b57..571348e 100644 Binary files a/src/equipment/__pycache__/service.cpython-311.pyc and b/src/equipment/__pycache__/service.cpython-311.pyc differ diff --git a/src/equipment/schema.py b/src/equipment/schema.py index 864b35e..1785b03 100644 --- a/src/equipment/schema.py +++ b/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] = [] diff --git a/src/equipment/service.py b/src/equipment/service.py index 3415a0a..51e7255 100644 --- a/src/equipment/service.py +++ b/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()