You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
54 lines
1.6 KiB
Python
54 lines
1.6 KiB
Python
|
|
|
|
from sqlalchemy import Select, Delete
|
|
from .model import Equipment
|
|
from .schema import EquipmentCreate, EquipmentUpdate
|
|
from typing import Optional
|
|
|
|
from src.database.core import DbSession
|
|
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)
|
|
result = await db_session.execute(query)
|
|
return result.scalars().one_or_none()
|
|
|
|
|
|
async def get_all(*, db_session: DbSession):
|
|
"""Returns all documents."""
|
|
query = Select(Equipment)
|
|
result = await db_session.execute(query)
|
|
return result.scalars().all()
|
|
|
|
|
|
async def create(*, db_session: DbSession, equipment_in: EquipmentCreate):
|
|
"""Creates a new document."""
|
|
equipment = Equipment(**equipment_in.model_dump())
|
|
db_session.add(equipment)
|
|
await db_session.commit()
|
|
return equipment
|
|
|
|
|
|
async def update(*, db_session: DbSession, equipment: Equipment, equipment_in: EquipmentUpdate):
|
|
"""Updates a document."""
|
|
data = equipment_in.model_dump()
|
|
|
|
update_data = equipment_in.model_dump(exclude_defaults=True)
|
|
|
|
for field in data:
|
|
if field in update_data:
|
|
setattr(equipment, field, update_data[field])
|
|
|
|
await db_session.commit()
|
|
|
|
return equipment
|
|
|
|
|
|
async def delete(*, db_session: DbSession, equipment_id: str):
|
|
"""Deletes a document."""
|
|
query = Delete(Equipment).where(Equipment.id == equipment_id)
|
|
await db_session.execute(query)
|
|
await db_session.commit()
|