felicity-lims/backend/felicity_lims/felicity/api/gql/inventory/mutations.py

import logging
from typing import Dict, Optional, List
from datetime import datetime, date, timedelta

import strawberry  # noqa
from felicity.api.gql import OperationError, auth_from_info, verify_user_auth
from felicity.apps.inventory import models, schemas
from felicity.api.gql.inventory import types
from felicity.apps.inventory.conf import order_states

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[int] = 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
    department_uid: Optional[int] = None
    supplier_uid: Optional[int] = None
    category_uid: Optional[int] = None
    hazard_uid: Optional[int] = None
    store_room_uid: Optional[int] = None
    lot_number: Optional[str] = None
    batch: Optional[str] = None
    size: Optional[int] = None
    unit_uid: Optional[int] = None
    packaging_uid: Optional[int] = None
    price: Optional[int] = None
    quantity_received: Optional[int] = None
    minimum_level: Optional[int] = None
    expiry_date: Optional[datetime] = 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: int
    quantity: int
    remarks: Optional[str] = None


@strawberry.input
class StockOrderInputType:
    order_products: List[StockOrderProductLineInputType]
    department_uid: Optional[int] = None


@strawberry.input
class StockOrderApprovalInputType:
    remarks: str
    status: str


StockTransactionResponse = strawberry.union(
    "StockTransactionResponse", (types.StockTransactionType, OperationError), description=""  # noqa
)


@strawberry.input
class StockTransactionInputType:
    product_uid: int
    issued: int
    department_uid: Optional[int] = None


StockAdjustmentResponse = strawberry.union(
    "StockAdjustmentResponse", (types.StockAdjustmentType, OperationError), description=""  # noqa
)


@strawberry.input
class StockAdjustmentInputType:
    product_uid: int
    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: int, 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: int, 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: int, 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: int, 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: int, 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

        exists = await models.StockProduct.get(name=payload.name)
        if exists:
            return OperationError(error=f"StockProduct with this name already exists")

        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: int, 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: int,
                                 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_all(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 approve_stock_order(self, info, uid: int, 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 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.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: int,
                                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]:
            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:
                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': quantity})

        stock_order = await stock_order.update({'status': payload.status, 'remarks': payload.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: int) -> 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")

        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
        test_value = product.remaining
        test_value -= stock_transaction.issued
        if test_value < 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:
            remaining = product.remaining - stock_transaction.issued
            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:
            test_value = product.remaining
            test_value -= stock_adjustment.adjust
            if test_value < 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:
                remaining = product.remaining - stock_adjustment.adjust
                await product.update({'remaining': remaining})

        return types.StockAdjustmentType(**stock_adjustment.marshal_simple())