refactor: get equipment by assetnum

main
MrWaradana 1 year ago
parent 277d48d03b
commit 30e7679a18

@ -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] = []

@ -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()

Loading…
Cancel
Save