68 lines
2.5 KiB
Python
68 lines
2.5 KiB
Python
from __future__ import annotations
|
|
|
|
from typing import TypeVar
|
|
|
|
from sqlalchemy.orm import Session
|
|
|
|
from fastapi import HTTPException, status
|
|
|
|
from app.models.contact import Contact
|
|
from app.models.customer import Customer
|
|
from app.models.device import Device
|
|
from app.models.equipment import Equipment
|
|
from app.models.location import Location
|
|
from app.models.validation import Validation
|
|
from app.repositories.base import Repository
|
|
from app.repositories.domain import (
|
|
ContactRepository,
|
|
CustomerRepository,
|
|
DeviceRepository,
|
|
EquipmentRepository,
|
|
LocationRepository,
|
|
ValidationRepository,
|
|
)
|
|
|
|
ModelT = TypeVar("ModelT")
|
|
|
|
|
|
class CrudService:
|
|
def __init__(self, repository: Repository[ModelT]) -> None:
|
|
self.repository = repository
|
|
|
|
def list(self, page: int = 1, page_size: int = 20, search: str | None = None) -> dict:
|
|
safe_page = max(page, 1)
|
|
safe_page_size = min(max(page_size, 1), 100)
|
|
offset = (safe_page - 1) * safe_page_size
|
|
return {
|
|
"items": self.repository.list(safe_page_size, offset, search),
|
|
"total": self.repository.count(search),
|
|
"page": safe_page,
|
|
"page_size": safe_page_size,
|
|
}
|
|
|
|
def create(self, data: dict) -> ModelT:
|
|
return self.repository.add(self.repository.model(**data))
|
|
|
|
def update(self, item_id: str, data: dict) -> ModelT:
|
|
item = self.repository.get(item_id)
|
|
if item is None:
|
|
raise HTTPException(status_code=status.HTTP_404_NOT_FOUND, detail="Resource not found")
|
|
for key, value in data.items():
|
|
setattr(item, key, value)
|
|
return item
|
|
|
|
def delete(self, item_id: str) -> None:
|
|
item = self.repository.get(item_id)
|
|
if item is None:
|
|
raise HTTPException(status_code=status.HTTP_404_NOT_FOUND, detail="Resource not found")
|
|
self.repository.delete(item)
|
|
|
|
|
|
class DomainServices:
|
|
def __init__(self, session: Session) -> None:
|
|
self.customers: CrudService[Customer] = CrudService(CustomerRepository(session))
|
|
self.locations: CrudService[Location] = CrudService(LocationRepository(session))
|
|
self.contacts: CrudService[Contact] = CrudService(ContactRepository(session))
|
|
self.devices: CrudService[Device] = CrudService(DeviceRepository(session))
|
|
self.equipment: CrudService[Equipment] = CrudService(EquipmentRepository(session))
|
|
self.validations: CrudService[Validation] = CrudService(ValidationRepository(session))
|