felicity-lims/felicity/apps/storage/services.py

from felicity.apps.abstract.service import BaseService
from felicity.apps.analysis.entities.analysis import Sample
from felicity.apps.analysis.services.analysis import SampleService
from felicity.apps.storage.entities import (
    StorageContainer,
    StorageLocation,
    StorageSection,
    StoreRoom,
)
from felicity.apps.storage.repository import (
    StorageContainerRepository,
    StorageLocationRepository,
    StorageSectionRepository,
    StoreRoomRepository,
)
from felicity.apps.storage.schemas import (
    StorageContainerCreate,
    StorageContainerUpdate,
    StorageLocationCreate,
    StorageLocationUpdate,
    StorageSectionCreate,
    StorageSectionUpdate,
    StoreRoomCreate,
    StoreRoomUpdate,
)


class StoreRoomService(BaseService[StoreRoom, StoreRoomCreate, StoreRoomUpdate]):
    def __init__(self):
        super().__init__(StoreRoomRepository())


class StorageLocationService(
    BaseService[StorageLocation, StorageLocationCreate, StorageLocationUpdate]
):
    def __init__(self):
        super().__init__(StorageLocationRepository())


class StorageSectionService(
    BaseService[StorageSection, StorageSectionCreate, StorageSectionUpdate]
):
    def __init__(self):
        self.sample_service = SampleService()
        super().__init__(StorageSectionRepository())


class StorageContainerService(
    BaseService[StorageContainer, StorageContainerCreate, StorageContainerUpdate]
):
    def __init__(self):
        self.sample_service = SampleService()
        super().__init__(StorageContainerRepository())

    async def get_samples(self, storage_container_uid: str) -> list[Sample]:
        return await self.sample_service.get_all(
            storage_container_uid=storage_container_uid
        )

    async def reset_stored_count(self, uid: str) -> None:
        samples = await self.get_samples(uid)
        await self.update(uid, {"stored_count": len(samples)})