from lib_b2b.erp import ERPOrderLine


# TODO: - remove this class along with ship manager integration
class OrderShipmentStatus:
    def __init__(self, order: ERPOrderLine, request_count, shipped_count):
        self.request_count = request_count
        self.shipped_count = shipped_count
        self.order = order

    def is_shipped(self):
        return (self.order.order_qty - self.order.cancelled_qty - self.shipped_count) <= 0

    def unrequested_quantity(self):
        unrequested_count = (self.order.order_qty - self.order.cancelled_qty - self.shipped_count)
        if unrequested_count < 0:
            return 0
        else:
            return unrequested_count