edan_f0/oscar/apps/partner/availability.py

from django.utils.translation import ugettext_lazy as _


class Base(object):
    """
    Simple based availability class which defaults to everything being
    unavailable.
    """
    is_tax_known = False
    is_available_to_buy = False
    code = ''
    message = ''
    lead_time = None
    dispatch_date = None


class NoStockRecord(Base):
    availability_code = 'outofstock'
    availability = _("Unavailable")


class WrappedStockrecord(Base):

    def __init__(self, product, stockrecord=None):
        self.product = product
        self.stockrecord = stockrecord

    @property
    def is_available_to_buy(self):
        if self.stockrecord is None:
            return False
        if not self.product.get_product_class().track_stock:
            return True
        return self.stockrecord.is_available_to_buy

    @property
    def code(self):
        return self.stockrecord.availability_code

    @property
    def message(self):
        return self.stockrecord.availability

    @property
    def lead_time(self):
        return self.stockrecord.lead_time

    @property
    def dispatch_date(self):
        return self.stockrecord.dispatch_date