Source code for shuup.core.utils.line_unit_mixin

from django.core.exceptions import ObjectDoesNotExist

from shuup.core.models._units import PiecesSalesUnit, UnitInterface




class LineWithUnit:
    @property
    def unit(self):
        """
        Unit of this line.

        :rtype: UnitInterface
        """
        # TODO: Store the sales unit and display unit to the line
        if not self.product or not self.product.sales_unit or not self.shop:
            return UnitInterface(PiecesSalesUnit())
        try:
            shop_product = self.product.get_shop_instance(self.shop)
        except ObjectDoesNotExist:
            return UnitInterface(self.product.sales_unit)
        return shop_product.unit