felicity-lims/felicity/api/gql/storage/query.py
2023-04-07 17:52:19 +02:00

60 lines
1.9 KiB
Python

from typing import List, Optional
import strawberry # noqa
from felicity.api.gql.storage import types
from felicity.apps.storage import models
from felicity.core.uid_gen import FelicityID
@strawberry.type
class StorageQuery:
@strawberry.field
async def store_room_all(self, info) -> List[types.StoreRoomType]:
return await models.StoreRoom.all()
@strawberry.field
async def store_room_by_uid(
self, info, uid: FelicityID
) -> Optional[types.StoreRoomType]:
return await models.StoreRoom.get(uid=uid)
@strawberry.field
async def storage_locations(
self, info, store_room_uid: FelicityID
) -> List[types.StorageLocationType]:
return await models.StorageLocation.get_all(store_room_uid=store_room_uid)
@strawberry.field
async def storage_location_by_uid(
self, info, uid: FelicityID
) -> Optional[types.StorageLocationType]:
return await models.StorageLocation.get(uid=uid)
@strawberry.field
async def storage_sections(
self, info, storage_location_uid: FelicityID
) -> List[types.StorageSectionType]:
return await models.StorageLocation.get_all(
storage_location_uid=storage_location_uid
)
@strawberry.field
async def storage_section_by_uid(
self, info, uid: FelicityID
) -> Optional[types.StorageSectionType]:
return await models.StorageSection.get(uid=uid)
@strawberry.field
async def storage_containers(
self, info, storage_section_uid: FelicityID
) -> List[types.StorageContainerType]:
return await models.StorageContainer.get_all(
storage_section_uid=storage_section_uid
)
@strawberry.field
async def storage_container_by_uid(
self, info, uid: FelicityID
) -> Optional[types.StorageContainerType]:
return await models.StorageContainer.get(uid=uid)