mirror of
https://github.com/beak-insights/felicity-lims.git
synced 2025-02-24 08:53:00 +08:00
53 lines
2.2 KiB
Python
53 lines
2.2 KiB
Python
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)})
|