import logging from datetime import datetime, timedelta from typing import Dict, List, Optional import strawberry # noqa from felicity.api.gql import OperationError, auth_from_info, verify_user_auth from felicity.api.gql.inventory import types from felicity.apps.inventory import models, schemas from felicity.apps.inventory.conf import order_states from felicity.core.uid_gen import FelicityID logging.basicConfig(level=logging.INFO) logger = logging.getLogger(__name__) StockItemResponse = strawberry.union( "StockItemResponse", (types.StockItemType, OperationError), description="" # noqa ) @strawberry.input class StockItemInputType: name: str description: str department_uid: Optional[FelicityID] = None StockCategoryResponse = strawberry.union( "StockCategoryResponse", (types.StockCategoryType, OperationError), description="", # noqa ) @strawberry.input class StockCategoryInputType: name: str description: str HazardResponse = strawberry.union( "HazardResponse", (types.HazardType, OperationError), description="" # noqa ) @strawberry.input class HazardInputType: name: str description: str StockUnitResponse = strawberry.union( "StockUnitResponse", (types.StockUnitType, OperationError), description="" # noqa ) @strawberry.input class StockUnitInputType: name: str StockPackagingResponse = strawberry.union( "StockPackagingResponse", (types.StockPackagingType, OperationError), description="", # noqa ) @strawberry.input class StockPackagingInputType: name: str StockProductResponse = strawberry.union( "StockProductResponse", (types.StockProductType, OperationError), description="", # noqa ) @strawberry.input class StockProductInputType: name: str stock_item_uid: Optional[FelicityID] = None department_uid: Optional[FelicityID] = None supplier_uid: Optional[FelicityID] = None category_uid: Optional[FelicityID] = None hazard_uid: Optional[FelicityID] = None store_room_uid: Optional[FelicityID] = None lot_number: Optional[str] = None batch: Optional[str] = None size: Optional[int] = None unit_uid: Optional[FelicityID] = None packaging_uid: Optional[FelicityID] = None price: Optional[int] = None quantity_received: Optional[int] = None date_received: Optional[datetime] = None expiry_date: Optional[datetime] = None received_by_uid: Optional[FelicityID] = None @strawberry.type class StockOrderLineType: stock_order: types.StockOrderType order_products: List[types.StockOrderProductType] StockOrderResponse = strawberry.union( "StockOrderResponse", (StockOrderLineType, types.StockOrderType, OperationError), description="", # noqa ) @strawberry.input class StockOrderProductLineInputType: product_uid: FelicityID quantity: int remarks: Optional[str] = None @strawberry.input class StockOrderInputType: order_products: List[StockOrderProductLineInputType] department_uid: Optional[FelicityID] = None @strawberry.input class StockOrderApprovalInputType: remarks: str status: str StockTransactionResponse = strawberry.union( "StockTransactionResponse", (types.StockTransactionType, OperationError), description="", # noqa ) @strawberry.input class StockTransactionInputType: product_uid: FelicityID issued: int department_uid: Optional[FelicityID] = None StockAdjustmentResponse = strawberry.union( "StockAdjustmentResponse", (types.StockAdjustmentType, OperationError), description="", # noqa ) @strawberry.input class StockAdjustmentInputType: product_uid: FelicityID adjustment_type: str adjust: int remarks: Optional[str] = None @strawberry.type class InventoryMutations: @strawberry.mutation async def create_stock_item( self, info, payload: StockItemInputType ) -> StockItemResponse: is_authenticated, felicity_user = await auth_from_info(info) auth_success, auth_error = verify_user_auth( is_authenticated, felicity_user, "Only Authenticated user can create stock item", ) if not auth_success: return auth_error exists = await models.StockItem.get(name=payload.name) if exists: return OperationError(error=f"StockItem with this name already exists") incoming: Dict = { "created_by_uid": felicity_user.uid, "updated_by_uid": felicity_user.uid, } for k, v in payload.__dict__.items(): incoming[k] = v obj_in = schemas.StockItemCreate(**incoming) stock_item: models.StockItem = await models.StockItem.create(obj_in) return types.StockItemType(**stock_item.marshal_simple()) @strawberry.mutation async def update_stock_item( self, info, uid: FelicityID, payload: StockItemInputType ) -> StockItemResponse: is_authenticated, felicity_user = await auth_from_info(info) verify_user_auth( is_authenticated, felicity_user, "Only Authenticated user can update stock item", ) if not uid: return OperationError(error="No uid provided to identity update obj") stock_item: models.StockItem = await models.StockItem.get(uid=uid) if not stock_item: return OperationError( error=f"StockItem with uid {uid} not found. Cannot update obj ..." ) obj_data = stock_item.to_dict() for field in obj_data: if field in payload.__dict__: try: setattr(stock_item, field, payload.__dict__[field]) except Exception as e: # noqa pass setattr(stock_item, "updated_by_uid", felicity_user.uid) obj_in = schemas.StockItemUpdate(**stock_item.to_dict()) stock_item = await stock_item.update(obj_in) return types.StockItemType(**stock_item.marshal_simple()) @strawberry.mutation async def create_stock_category( self, info, payload: StockCategoryInputType ) -> StockCategoryResponse: is_authenticated, felicity_user = await auth_from_info(info) auth_success, auth_error = verify_user_auth( is_authenticated, felicity_user, "Only Authenticated user can create stock categories", ) if not auth_success: return auth_error exists = await models.StockCategory.get(name=payload.name) if exists: return OperationError(error=f"StockCategory with this name already exists") incoming: Dict = { "created_by_uid": felicity_user.uid, "updated_by_uid": felicity_user.uid, } for k, v in payload.__dict__.items(): incoming[k] = v obj_in = schemas.StockCategoryCreate(**incoming) stock_category: models.StockCategory = await models.StockCategory.create(obj_in) return types.StockCategoryType(**stock_category.marshal_simple()) @strawberry.mutation async def update_stock_category( self, info, uid: FelicityID, payload: StockCategoryInputType ) -> StockCategoryResponse: is_authenticated, felicity_user = await auth_from_info(info) verify_user_auth( is_authenticated, felicity_user, "Only Authenticated user can update stock category", ) if not uid: return OperationError(error="No uid provided to identity update obj") stock_category: models.StockCategory = await models.StockCategory.get(uid=uid) if not stock_category: return OperationError( error=f"StockCategory with uid {uid} not found. Cannot update obj ..." ) obj_data = stock_category.to_dict() for field in obj_data: if field in payload.__dict__: try: setattr(stock_category, field, payload.__dict__[field]) except Exception as e: # noqa pass setattr(stock_category, "updated_by_uid", felicity_user.uid) obj_in = schemas.StockCategoryUpdate(**stock_category.to_dict()) stock_category = await stock_category.update(obj_in) return types.StockCategoryType(**stock_category.marshal_simple()) @strawberry.mutation async def create_hazard(self, info, payload: HazardInputType) -> HazardResponse: is_authenticated, felicity_user = await auth_from_info(info) auth_success, auth_error = verify_user_auth( is_authenticated, felicity_user, "Only Authenticated user can create hazard", ) if not auth_success: return auth_error exists = await models.Hazard.get(name=payload.name) if exists: return OperationError(error=f"Hazard with this name already exists") incoming: Dict = { "created_by_uid": felicity_user.uid, "updated_by_uid": felicity_user.uid, } for k, v in payload.__dict__.items(): incoming[k] = v obj_in = schemas.HazardCreate(**incoming) hazard: models.Hazard = await models.Hazard.create(obj_in) return types.HazardType(**hazard.marshal_simple()) @strawberry.mutation async def update_hazard( self, info, uid: FelicityID, payload: HazardInputType ) -> HazardResponse: is_authenticated, felicity_user = await auth_from_info(info) verify_user_auth( is_authenticated, felicity_user, "Only Authenticated user can update hazard", ) if not uid: return OperationError(error="No uid provided to identity update obj") hazard: models.Hazard = await models.Hazard.get(uid=uid) if not hazard: return OperationError( error=f"Hazard with uid {uid} not found. Cannot update obj ..." ) obj_data = hazard.to_dict() for field in obj_data: if field in payload.__dict__: try: setattr(hazard, field, payload.__dict__[field]) except Exception as e: # noqa pass setattr(hazard, "updated_by_uid", felicity_user.uid) obj_in = schemas.HazardUpdate(**hazard.to_dict()) hazard = await hazard.update(obj_in) return types.HazardType(**hazard.marshal_simple()) @strawberry.mutation async def create_stock_unit( self, info, payload: StockUnitInputType ) -> StockUnitResponse: is_authenticated, felicity_user = await auth_from_info(info) auth_success, auth_error = verify_user_auth( is_authenticated, felicity_user, "Only Authenticated user can create stock_unit", ) if not auth_success: return auth_error exists = await models.StockUnit.get(name=payload.name) if exists: return OperationError(error=f"StockUnit with this name already exists") incoming: Dict = { "created_by_uid": felicity_user.uid, "updated_by_uid": felicity_user.uid, } for k, v in payload.__dict__.items(): incoming[k] = v obj_in = schemas.StockUnitCreate(**incoming) stock_unit: models.StockUnit = await models.StockUnit.create(obj_in) return types.StockUnitType(**stock_unit.marshal_simple()) @strawberry.mutation async def update_stock_unit( self, info, uid: FelicityID, payload: StockUnitInputType ) -> StockUnitResponse: is_authenticated, felicity_user = await auth_from_info(info) verify_user_auth( is_authenticated, felicity_user, "Only Authenticated user can update stock_unit", ) if not uid: return OperationError(error="No uid provided to identity update obj") stock_unit: models.StockUnit = await models.StockUnit.get(uid=uid) if not stock_unit: return OperationError( error=f"StockUnit with uid {uid} not found. Cannot update obj ..." ) obj_data = stock_unit.to_dict() for field in obj_data: if field in payload.__dict__: try: setattr(stock_unit, field, payload.__dict__[field]) except Exception as e: # noqa pass setattr(stock_unit, "updated_by_uid", felicity_user.uid) obj_in = schemas.StockUnitUpdate(**stock_unit.to_dict()) stock_unit = await stock_unit.update(obj_in) return types.StockUnitType(**stock_unit.marshal_simple()) @strawberry.mutation async def create_stock_packaging( self, info, payload: StockPackagingInputType ) -> StockPackagingResponse: is_authenticated, felicity_user = await auth_from_info(info) auth_success, auth_error = verify_user_auth( is_authenticated, felicity_user, "Only Authenticated user can create stock_packaging", ) if not auth_success: return auth_error exists = await models.StockPackaging.get(name=payload.name) if exists: return OperationError(error=f"StockPackaging with this name already exists") incoming: Dict = { "created_by_uid": felicity_user.uid, "updated_by_uid": felicity_user.uid, } for k, v in payload.__dict__.items(): incoming[k] = v obj_in = schemas.StockPackagingCreate(**incoming) stock_packaging: models.StockPackaging = await models.StockPackaging.create( obj_in ) return types.StockPackagingType(**stock_packaging.marshal_simple()) @strawberry.mutation async def update_stock_packaging( self, info, uid: FelicityID, payload: StockPackagingInputType ) -> StockPackagingResponse: is_authenticated, felicity_user = await auth_from_info(info) verify_user_auth( is_authenticated, felicity_user, "Only Authenticated user can update stock_packaging", ) if not uid: return OperationError(error="No uid provided to identity update obj") stock_packaging: models.StockPackaging = await models.StockPackaging.get( uid=uid ) if not stock_packaging: return OperationError( error=f"StockPackaging with uid {uid} not found. Cannot update obj ..." ) obj_data = stock_packaging.to_dict() for field in obj_data: if field in payload.__dict__: try: setattr(stock_packaging, field, payload.__dict__[field]) except Exception as e: # noqa pass setattr(stock_packaging, "updated_by_uid", felicity_user.uid) obj_in = schemas.StockPackagingUpdate(**stock_packaging.to_dict()) stock_packaging = await stock_packaging.update(obj_in) return types.StockPackagingType(**stock_packaging.marshal_simple()) @strawberry.mutation async def create_stock_product( self, info, payload: StockProductInputType ) -> StockProductResponse: is_authenticated, felicity_user = await auth_from_info(info) auth_success, auth_error = verify_user_auth( is_authenticated, felicity_user, "Only Authenticated user can create stock_product", ) if not auth_success: return auth_error incoming: Dict = { "date_received": datetime.today(), "expiry_date": datetime.today() + timedelta(days=10), "created_by_uid": felicity_user.uid, "updated_by_uid": felicity_user.uid, } for k, v in payload.__dict__.items(): incoming[k] = v incoming["remaining"] = payload.quantity_received obj_in = schemas.StockProductCreate(**incoming) stock_product: models.StockProduct = await models.StockProduct.create(obj_in) return types.StockProductType(**stock_product.marshal_simple()) @strawberry.mutation async def update_stock_product( self, info, uid: FelicityID, payload: StockProductInputType ) -> StockProductResponse: is_authenticated, felicity_user = await auth_from_info(info) verify_user_auth( is_authenticated, felicity_user, "Only Authenticated user can update stock_product", ) if not uid: return OperationError(error="No uid provided to identity update obj") stock_product: models.StockProduct = await models.StockProduct.get(uid=uid) if not stock_product: return OperationError( error=f"StockProduct with uid {uid} not found. Cannot update obj ..." ) obj_data = stock_product.to_dict() for field in obj_data: if field in payload.__dict__: try: setattr(stock_product, field, payload.__dict__[field]) except Exception as e: # noqa pass setattr(stock_product, "updated_by_uid", felicity_user.uid) obj_in = schemas.StockProductUpdate(**stock_product.to_dict()) stock_product = await stock_product.update(obj_in) return types.StockProductType(**stock_product.marshal_simple()) @strawberry.mutation async def create_stock_order( self, info, payload: StockOrderInputType ) -> StockOrderResponse: is_authenticated, felicity_user = await auth_from_info(info) auth_success, auth_error = verify_user_auth( is_authenticated, felicity_user, "Only Authenticated user can create stock_order", ) if not auth_success: return auth_error incoming: Dict = { "created_by_uid": felicity_user.uid, "updated_by_uid": felicity_user.uid, } for k, v in payload.__dict__.items(): incoming[k] = v obj_in = schemas.StockOrderCreate(**incoming) stock_order: models.StockOrder = await models.StockOrder.create(obj_in) # create Order Products for prod in payload.order_products: product: models.StockProduct = await models.StockProduct.get( uid=prod.product_uid ) op_in = schemas.StockOrderProductCreate( product_uid=prod.product_uid, order_uid=stock_order.uid, price=product.price, quantity=prod.quantity, ) await models.StockOrderProduct.create(op_in) order_products = await models.StockOrderProduct.get_all( order_uid=stock_order.uid ) return StockOrderLineType( stock_order=stock_order, order_products=order_products ) @strawberry.mutation async def update_stock_order( self, info, uid: FelicityID, payload: List[StockOrderProductLineInputType] ) -> StockOrderResponse: is_authenticated, felicity_user = await auth_from_info(info) auth_success, auth_error = verify_user_auth( is_authenticated, felicity_user, "Only Authenticated user can create stock_order", ) if not auth_success: return auth_error stock_order: models.StockOrder = await models.StockOrder.get(uid=uid) if stock_order.status != order_states.PREPARATION: return OperationError( error=f"You can only update a StockOrder under preparation" ) # add Order Products old_products = await models.StockOrderProduct.get_all(order_uid=uid) _pr_uids = [p.product_uid for p in old_products] for prod in payload: # New product if prod.product_uid not in _pr_uids: product: models.StockProduct = await models.StockProduct.get( uid=prod.product_uid ) op_in = schemas.StockOrderProductCreate( product_uid=prod.product_uid, order_uid=uid, price=product.price, quantity=prod.quantity, ) await models.StockOrderProduct.create(op_in) else: # update existing products so_product = await models.StockOrderProduct.get( product_uid=prod.product_uid, order_uid=uid ) await so_product.update({"quantity": prod.quantity}) # delete removed products order_products_uids = [p.product_uid for p in payload] for _op in old_products: if _op.product_uid not in order_products_uids: await _op.delete() # re-fetch updated o_products = await models.StockOrderProduct.get_all(order_uid=uid) stock_order: models.StockOrder = await models.StockOrder.get(uid=uid) return StockOrderLineType(stock_order=stock_order, order_products=o_products) @strawberry.mutation async def submit_stock_order(self, info, uid: FelicityID) -> StockOrderResponse: is_authenticated, felicity_user = await auth_from_info(info) auth_success, auth_error = verify_user_auth( is_authenticated, felicity_user, "Only Authenticated user can submit stock orders", ) if not auth_success: return auth_error stock_order: models.StockOrder = await models.StockOrder.get(uid=uid) if stock_order.status not in [order_states.PREPARATION]: return OperationError( error=f"You can only submit a StockOrder under preperation" ) stock_order = await stock_order.update( {"status": order_states.SUBMITTED} ) # noqa return types.StockOrderType(**stock_order.marshal_simple()) @strawberry.mutation async def approve_stock_order( self, info, uid: FelicityID, payload: StockOrderApprovalInputType ) -> StockOrderResponse: is_authenticated, felicity_user = await auth_from_info(info) auth_success, auth_error = verify_user_auth( is_authenticated, felicity_user, "Only Authenticated user can approve stock orders", ) if not auth_success: return auth_error stock_order: models.StockOrder = await models.StockOrder.get(uid=uid) if stock_order.status not in [order_states.SUBMITTED]: return OperationError( error=f"You can only approve/revert a submitted StockOrder" ) stock_order = await stock_order.update( {"status": payload.status, "remarks": payload.remarks} ) # noqa return types.StockOrderType(**stock_order.marshal_simple()) @strawberry.mutation async def issue_stock_order( self, info, uid: FelicityID, payload: List[StockOrderProductLineInputType] ) -> StockOrderResponse: is_authenticated, felicity_user = await auth_from_info(info) auth_success, auth_error = verify_user_auth( is_authenticated, felicity_user, "Only Authenticated user can delete stock orders", ) if not auth_success: return auth_error stock_order: models.StockOrder = await models.StockOrder.get(uid=uid) if stock_order.status not in [order_states.PENDING, order_states.SUBMITTED]: return OperationError(error=f"You can only issue a pending StockOrder") # issuance for order_p in payload: # init transaction incoming: Dict = { "created_by_uid": felicity_user.uid, "updated_by_uid": felicity_user.uid, "date_issued": datetime.now(), "product_uid": order_p.product_uid, "issued": order_p.quantity, } obj_in = schemas.StockTransactionCreate(**incoming) stock_transaction: models.StockTransaction = ( await models.StockTransaction.create(obj_in) ) # product = await models.StockProduct.get(uid=order_p.product_uid) quantity = product.remaining - order_p.quantity if quantity < 0: return OperationError( error=f"Sorry you cannot issue beyond whats available" ) else: await product.update({"remaining": quantity}) stock_order = await stock_order.update( {"status": order_states.PROCESSED, "remarks": ""} ) # noqa o_products = await models.StockOrderProduct.get_all(order_uid=uid) return StockOrderLineType(stock_order=stock_order, order_products=o_products) @strawberry.mutation async def delete_stock_order(self, info, uid: FelicityID) -> StockOrderResponse: is_authenticated, felicity_user = await auth_from_info(info) auth_success, auth_error = verify_user_auth( is_authenticated, felicity_user, "Only Authenticated user can delete stock orders", ) if not auth_success: return auth_error stock_order: models.StockOrder = await models.StockOrder.get(uid=uid) if stock_order.status != order_states.PREPARATION: return OperationError( error=f"You can only delete a StockOrder under preparation" ) order_products = await models.StockOrderProduct.get_all( order_uid=stock_order.uid ) for op in order_products: await op.delete() await stock_order.delete() return types.StockOrderType(**stock_order.marshal_simple()) @strawberry.mutation async def create_stock_transaction( self, info, payload: StockTransactionInputType ) -> StockTransactionResponse: is_authenticated, felicity_user = await auth_from_info(info) auth_success, auth_error = verify_user_auth( is_authenticated, felicity_user, "Only Authenticated user can create stock_transaction", ) if not auth_success: return auth_error incoming: Dict = { "created_by_uid": felicity_user.uid, "updated_by_uid": felicity_user.uid, "date_issued": datetime.now(), } for k, v in payload.__dict__.items(): incoming[k] = v obj_in = schemas.StockTransactionCreate(**incoming) stock_transaction: models.StockTransaction = ( await models.StockTransaction.create(obj_in) ) product = await models.StockProduct.get(uid=stock_transaction.product_uid) # Transact remaining = product.remaining - stock_transaction.issued if remaining < 0: await stock_transaction.update( {"remarks": "Sustained: Sorry you cannot issue beyond whats available"} ) # noqa return OperationError( error=f"Sorry you cannot issue beyond whats available" ) else: await product.update({"remaining": remaining}) return types.StockTransactionType(**stock_transaction.marshal_simple()) @strawberry.mutation async def create_stock_adjustment( self, info, payload: StockAdjustmentInputType ) -> StockAdjustmentResponse: is_authenticated, felicity_user = await auth_from_info(info) auth_success, auth_error = verify_user_auth( is_authenticated, felicity_user, "Only Authenticated user can create stock_adjustment", ) if not auth_success: return auth_error incoming: Dict = { "created_by_uid": felicity_user.uid, "updated_by_uid": felicity_user.uid, "adjustment_date": datetime.now(), } for k, v in payload.__dict__.items(): incoming[k] = v obj_in = schemas.StockAdjustmentCreate(**incoming) stock_adjustment: models.StockAdjustment = await models.StockAdjustment.create( obj_in ) product = await models.StockProduct.get(uid=stock_adjustment.product_uid) # Adjust if stock_adjustment.adjustment_type == "transfer in": remaining = product.remaining + stock_adjustment.adjust await product.update({"remaining": remaining}) else: remaining = product.remaining - stock_adjustment.adjust if remaining < 0: await stock_adjustment.update( { "remarks": "Sustained: Sorry you cant transact beyond what you have" } ) # noqa return OperationError( error="Sorry you cant transact beyond what you have" ) else: await product.update({"remaining": remaining}) return types.StockAdjustmentType(**stock_adjustment.marshal_simple())