| 123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760 |
- from decimal import Decimal, ROUND_DOWN, ROUND_UP
- import math
- import datetime
-
- from django.core import exceptions
- from django.template.defaultfilters import slugify
- from django.db import models
- from django.utils.translation import ungettext, ugettext as _
- from django.core.exceptions import ValidationError
- from django.core.urlresolvers import reverse
- from django.conf import settings
-
- from oscar.apps.offer.managers import ActiveOfferManager
- from oscar.models.fields import PositiveDecimalField, ExtendedURLField
-
-
- class ConditionalOffer(models.Model):
- """
- A conditional offer (eg buy 1, get 10% off)
- """
- name = models.CharField(_('Name'), max_length=128, unique=True,
- help_text=_("""This is displayed within the customer's
- basket"""))
- slug = models.SlugField(_('Slug'), max_length=128, unique=True, null=True)
- description = models.TextField(_('Description'), blank=True, null=True)
-
- # Offers come in a few different types:
- # (a) Offers that are available to all customers on the site. Eg a
- # 3-for-2 offer.
- # (b) Offers that are linked to a voucher, and only become available once
- # that voucher has been applied to the basket
- # (c) Offers that are linked to a user. Eg, all students get 10% off. The code
- # to apply this offer needs to be coded
- # (d) Session offers - these are temporarily available to a user after some trigger
- # event. Eg, users coming from some affiliate site get 10% off.
- SITE, VOUCHER, USER, SESSION = ("Site", "Voucher", "User", "Session")
- TYPE_CHOICES = (
- (SITE, _("Site offer - available to all users")),
- (VOUCHER, _("Voucher offer - only available after entering the appropriate voucher code")),
- (USER, _("User offer - available to certain types of user")),
- (SESSION, _("Session offer - temporary offer, available for a user for the duration of their session")),
- )
- offer_type = models.CharField(_('Type'), choices=TYPE_CHOICES, default=SITE, max_length=128)
-
- condition = models.ForeignKey('offer.Condition')
- benefit = models.ForeignKey('offer.Benefit')
-
- # Range of availability. Note that if this is a voucher offer, then these
- # dates are ignored and only the dates from the voucher are used to determine
- # availability.
- start_date = models.DateField(_('Start Date'), blank=True, null=True)
- end_date = models.DateField(_('End Date'), blank=True, null=True,
- help_text=_("""Offers are not active on their end
- date, only the days preceding"""))
-
- # Some complicated situations require offers to be applied in a set order.
- priority = models.IntegerField(_('Priority'), default=0,
- help_text=_("The highest priority offers are applied first"))
-
- # We track some information on usage
- total_discount = models.DecimalField(_('Total Discount'), decimal_places=2, max_digits=12, default=Decimal('0.00'))
- num_orders = models.PositiveIntegerField(_('Number of Orders'), default=0)
-
- date_created = models.DateTimeField(auto_now_add=True)
-
- objects = models.Manager()
- active = ActiveOfferManager()
-
- redirect_url = ExtendedURLField(_('URL redirect (optional)'), blank=True)
-
- # We need to track the voucher that this offer came from (if it is a voucher offer)
- _voucher = None
-
- class Meta:
- ordering = ['-priority']
- verbose_name = _("Conditional Offer")
- verbose_name_plural = _("Conditional Offers")
-
- def save(self, *args, **kwargs):
- if not self.slug:
- self.slug = slugify(self.name)
- return super(ConditionalOffer, self).save(*args, **kwargs)
-
- def get_absolute_url(self):
- return reverse('offer:detail', kwargs={'slug': self.slug})
-
- def __unicode__(self):
- return self.name
-
- def clean(self):
- if self.start_date and self.end_date and self.start_date > self.end_date:
- raise exceptions.ValidationError(_('End date should be later than start date'))
-
- def is_active(self, test_date=None):
- if not test_date:
- test_date = datetime.date.today()
- return self.start_date <= test_date and test_date < self.end_date
-
- def is_condition_satisfied(self, basket):
- return self._proxy_condition().is_satisfied(basket)
-
- def is_condition_partially_satisfied(self, basket):
- return self._proxy_condition().is_partially_satisfied(basket)
-
- def get_upsell_message(self, basket):
- return self._proxy_condition().get_upsell_message(basket)
-
- def apply_benefit(self, basket):
- """
- Applies the benefit to the given basket and returns the discount.
- """
- if not self.is_condition_satisfied(basket):
- return Decimal('0.00')
- return self._proxy_benefit().apply(basket, self._proxy_condition())
-
- def set_voucher(self, voucher):
- self._voucher = voucher
-
- def get_voucher(self):
- return self._voucher
-
- def _proxy_condition(self):
- """
- Returns the appropriate proxy model for the condition
- """
- field_dict = dict(self.condition.__dict__)
- if '_state' in field_dict:
- del field_dict['_state']
- if '_range_cache' in field_dict:
- del field_dict['_range_cache']
- if self.condition.type == self.condition.COUNT:
- return CountCondition(**field_dict)
- elif self.condition.type == self.condition.VALUE:
- return ValueCondition(**field_dict)
- elif self.condition.type == self.condition.COVERAGE:
- return CoverageCondition(**field_dict)
- return self.condition
-
- def _proxy_benefit(self):
- """
- Returns the appropriate proxy model for the condition
- """
- field_dict = dict(self.benefit.__dict__)
- if '_state' in field_dict:
- del field_dict['_state']
- if self.benefit.type == self.benefit.PERCENTAGE:
- return PercentageDiscountBenefit(**field_dict)
- elif self.benefit.type == self.benefit.FIXED:
- return AbsoluteDiscountBenefit(**field_dict)
- elif self.benefit.type == self.benefit.MULTIBUY:
- return MultibuyDiscountBenefit(**field_dict)
- elif self.benefit.type == self.benefit.FIXED_PRICE:
- return FixedPriceBenefit(**field_dict)
- return self.benefit
-
- def record_usage(self, discount):
- self.num_orders += 1
- self.total_discount += discount
- self.save()
-
-
- class Condition(models.Model):
- COUNT, VALUE, COVERAGE = ("Count", "Value", "Coverage")
- TYPE_CHOICES = (
- (COUNT, _("Depends on number of items in basket that are in condition range")),
- (VALUE, _("Depends on value of items in basket that are in condition range")),
- (COVERAGE, _("Needs to contain a set number of DISTINCT items from the condition range"))
- )
- range = models.ForeignKey('offer.Range')
- type = models.CharField(_('Type'), max_length=128, choices=TYPE_CHOICES)
- value = PositiveDecimalField(_('Value'), decimal_places=2, max_digits=12)
-
- class Meta:
- verbose_name = _("Condition")
- verbose_name_plural = _("Conditions")
-
- def __unicode__(self):
- if self.type == self.COUNT:
- return _("Basket includes %(count)d item(s) from %(range)s") % {
- 'count': self.value, 'range': unicode(self.range).lower()}
- elif self.type == self.COVERAGE:
- return _("Basket includes %(count)d distinct products from %(range)s") % {
- 'count': self.value, 'range': unicode(self.range).lower()}
- return _("Basket includes %(count)d value from %(range)s") % {
- 'count': self.value, 'range': unicode(self.range).lower()}
-
- description = __unicode__
-
- def consume_items(self, basket, lines=None):
- raise NotImplementedError("This method should never be called - "
- "ensure you are using the correct proxy model")
-
- def is_satisfied(self, basket):
- """
- Determines whether a given basket meets this condition. This is
- stubbed in this top-class object. The subclassing proxies are
- responsible for implementing it correctly.
- """
- return False
-
- def is_partially_satisfied(self, basket):
- """
- Determine if the basket partially meets the condition. This is useful
- for up-selling messages to entice customers to buy something more in
- order to qualify for an offer.
- """
- return False
-
- def get_upsell_message(self, basket):
- return None
-
- def can_apply_condition(self, product):
- """
- Determines whether the condition can be applied to a given product
- """
- return (self.range.contains_product(product)
- and product.is_discountable)
-
-
- class Benefit(models.Model):
- PERCENTAGE, FIXED, MULTIBUY, FIXED_PRICE = ("Percentage", "Absolute", "Multibuy", "Fixed price")
- TYPE_CHOICES = (
- (PERCENTAGE, _("Discount is a % of the product's value")),
- (FIXED, _("Discount is a fixed amount off the product's value")),
- (MULTIBUY, _("Discount is to give the cheapest product for free")),
- (FIXED_PRICE, _("Get the products that meet the condition for a fixed price")),
- )
- range = models.ForeignKey('offer.Range', null=True, blank=True)
- type = models.CharField(_('Type'), max_length=128, choices=TYPE_CHOICES)
- value = PositiveDecimalField(_('Value'), decimal_places=2, max_digits=12,
- null=True, blank=True)
-
- price_field = 'price_incl_tax'
-
- # If this is not set, then there is no upper limit on how many products
- # can be discounted by this benefit.
- max_affected_items = models.PositiveIntegerField(_('Max Affected Items'), blank=True, null=True,
- help_text=_("""Set this to prevent the discount consuming all items within the range that are in the basket."""))
-
- class Meta:
- verbose_name = _("Benefit")
- verbose_name_plural = _("Benefits")
-
- def __unicode__(self):
- if self.type == self.PERCENTAGE:
- desc = _("%(value)s%% discount on %(range)s") % {'value': self.value, 'range': unicode(self.range).lower()}
- elif self.type == self.MULTIBUY:
- desc = _("Cheapest product is free from %s") % unicode(self.range).lower()
- elif self.type == self.FIXED_PRICE:
- desc = _("The products that meet the condition are sold for %s") % self.value
- else:
- desc = _("%(value).2f discount on %(range)s") % {'value': float(self.value),
- 'range': unicode(self.range).lower()}
-
- if self.max_affected_items:
- desc += ungettext(" (max 1 item)", " (max %d items)", self.max_affected_items) % self.max_affected_items
-
- return desc
-
- description = __unicode__
-
- def apply(self, basket, condition=None):
- return Decimal('0.00')
-
- def clean(self):
- if self.value is None:
- if not self.type:
- raise ValidationError(_("Benefit requires a value"))
- elif self.type != self.MULTIBUY:
- raise ValidationError(_("Benefits of type %s need a value") % self.type)
- elif self.value > 100 and self.type == 'Percentage':
- raise ValidationError(_("Percentage benefit value can't be greater than 100"))
- # All benefits need a range apart from FIXED_PRICE
- if self.type and self.type != self.FIXED_PRICE and not self.range:
- raise ValidationError(_("Benefits of type %s need a range") % self.type)
-
- def round(self, amount):
- """
- Apply rounding to discount amount
- """
- if hasattr(settings, 'OSCAR_OFFER_ROUNDING_FUNCTION'):
- return settings.OSCAR_OFFER_ROUNDING_FUNCTION(amount)
- return amount.quantize(Decimal('.01'), ROUND_DOWN)
-
- def _effective_max_affected_items(self):
- if not self.max_affected_items:
- max_affected_items = 10000
- else:
- max_affected_items = self.max_affected_items
- return max_affected_items
-
- def can_apply_benefit(self, product):
- """
- Determines whether the benefit can be applied to a given product
- """
- return product.is_discountable
-
-
- class Range(models.Model):
- """
- Represents a range of products that can be used within an offer
- """
- name = models.CharField(_("Name"), max_length=128, unique=True)
- includes_all_products = models.BooleanField(_('Includes All Products'), default=False)
- included_products = models.ManyToManyField('catalogue.Product', related_name='includes', blank=True)
- excluded_products = models.ManyToManyField('catalogue.Product', related_name='excludes', blank=True)
- classes = models.ManyToManyField('catalogue.ProductClass', related_name='classes', blank=True)
- included_categories = models.ManyToManyField('catalogue.Category', related_name='includes', blank=True)
- date_created = models.DateTimeField(auto_now_add=True)
-
- __included_product_ids = None
- __excluded_product_ids = None
- __class_ids = None
-
- class Meta:
- verbose_name = _("Range")
- verbose_name_plural = _("Ranges")
-
- def __unicode__(self):
- return self.name
-
- def contains_product(self, product):
- """
- Check whether the passed product is part of this range
- """
- # We look for shortcircuit checks first before
- # the tests that require more database queries.
-
- if settings.OSCAR_OFFER_BLACKLIST_PRODUCT and \
- settings.OSCAR_OFFER_BLACKLIST_PRODUCT(product):
- return False
- excluded_product_ids = self._excluded_product_ids()
- if product.id in excluded_product_ids:
- return False
- if self.includes_all_products:
- return True
- if product.product_class_id in self._class_ids():
- return True
- included_product_ids = self._included_product_ids()
- if product.id in included_product_ids:
- return True
- test_categories = self.included_categories.all()
- if test_categories:
- for category in product.categories.all():
- for test_category in test_categories:
- if category == test_category or category.is_descendant_of(test_category):
- return True
- return False
-
- def _included_product_ids(self):
- if None == self.__included_product_ids:
- self.__included_product_ids = [row['id'] for row in self.included_products.values('id')]
- return self.__included_product_ids
-
- def _excluded_product_ids(self):
- if None == self.__excluded_product_ids:
- self.__excluded_product_ids = [row['id'] for row in self.excluded_products.values('id')]
- return self.__excluded_product_ids
-
- def _class_ids(self):
- if None == self.__class_ids:
- self.__class_ids = [row['id'] for row in self.classes.values('id')]
- return self.__class_ids
-
- def num_products(self):
- if self.includes_all_products:
- return None
- return self.included_products.all().count()
-
-
- class CountCondition(Condition):
- """
- An offer condition dependent on the NUMBER of matching items from the basket.
- """
-
- class Meta:
- proxy = True
- verbose_name = _("Count Condition")
- verbose_name_plural = _("Count Conditions")
-
- def is_satisfied(self, basket):
- """
- Determines whether a given basket meets this condition
- """
- num_matches = 0
- for line in basket.all_lines():
- if (self.can_apply_condition(line.product)
- and line.quantity_without_discount > 0):
- num_matches += line.quantity_without_discount
- if num_matches >= self.value:
- return True
- return False
-
- def _get_num_matches(self, basket):
- if hasattr(self, '_num_matches'):
- return getattr(self, '_num_matches')
- num_matches = 0
- for line in basket.all_lines():
- if (self.can_apply_condition(line.product)
- and line.quantity_without_discount > 0):
- num_matches += line.quantity_without_discount
- self._num_matches = num_matches
- return num_matches
-
- def is_partially_satisfied(self, basket):
- num_matches = self._get_num_matches(basket)
- return 0 < num_matches < self.value
-
- def get_upsell_message(self, basket):
- num_matches = self._get_num_matches(basket)
- delta = self.value - num_matches
- return ungettext('Buy %(delta)d more product from %(range)s',
- 'Buy %(delta)d more products from %(range)s', delta) % {
- 'delta': delta, 'range': self.range}
-
- def consume_items(self, basket, lines=None, value=None):
- """
- Marks items within the basket lines as consumed so they
- can't be reused in other offers.
- """
- lines = lines or basket.all_lines()
- consumed_products = []
- value = self.value if value is None else value
- for line in lines:
-
- if self.can_apply_condition(line.product):
- quantity_to_consume = min(line.quantity_without_discount,
- value - len(consumed_products))
- line.consume(quantity_to_consume)
- consumed_products.extend((line.product,)*int(quantity_to_consume))
- if len(consumed_products) == value:
- break
- return consumed_products
-
-
- class CoverageCondition(Condition):
- """
- An offer condition dependent on the number of DISTINCT matching items from the basket.
- """
- class Meta:
- proxy = True
- verbose_name = _("Coverage Condition")
- verbose_name_plural = _("Coverage Conditions")
-
-
- def is_satisfied(self, basket):
- """
- Determines whether a given basket meets this condition
- """
- covered_ids = []
- for line in basket.all_lines():
- if not line.is_available_for_discount:
- continue
- product = line.product
- if (self.can_apply_condition(product) and product.id not in covered_ids):
- covered_ids.append(product.id)
- if len(covered_ids) >= self.value:
- return True
- return False
-
- def _get_num_covered_products(self, basket):
- covered_ids = []
- for line in basket.all_lines():
- if not line.is_available_for_discount:
- continue
- product = line.product
- if (self.can_apply_condition(product) and product.id not in covered_ids):
- covered_ids.append(product.id)
- return len(covered_ids)
-
- def get_upsell_message(self, basket):
- delta = self.value - self._get_num_covered_products(basket)
- return ungettext('Buy %(delta)d more product from %(range)s',
- 'Buy %(delta)d more products from %(range)s', delta) % {
- 'delta': delta, 'range': self.range}
-
- def is_partially_satisfied(self, basket):
- return 0 < self._get_num_covered_products(basket) < self.value
-
- def consume_items(self, basket, lines=None, value=None):
- """
- Marks items within the basket lines as consumed so they
- can't be reused in other offers.
- """
- lines = lines or basket.all_lines()
- consumed_products = []
- value = self.value if value is None else value
- for line in basket.all_lines():
- product = line.product
- if (line.is_available_for_discount and self.can_apply_condition(product)
- and product not in consumed_products):
- line.consume(1)
- consumed_products.append(line.product)
- if len(consumed_products) >= value:
- break
- return consumed_products
-
- def get_value_of_satisfying_items(self, basket):
- covered_ids = []
- value = Decimal('0.00')
- for line in basket.all_lines():
- if (self.can_apply_condition(line.product) and line.product.id not in covered_ids):
- covered_ids.append(line.product.id)
- value += line.unit_price_incl_tax
- if len(covered_ids) >= self.value:
- return value
- return value
-
-
- class ValueCondition(Condition):
- """
- An offer condition dependent on the VALUE of matching items from the basket.
- """
- price_field = 'price_incl_tax'
-
- class Meta:
- proxy = True
- verbose_name = _("Value Condition")
- verbose_name_plural = _("Value Conditions")
-
-
- def is_satisfied(self, basket):
- """Determines whether a given basket meets this condition"""
- value_of_matches = Decimal('0.00')
- for line in basket.all_lines():
- product = line.product
- if (self.can_apply_condition(product) and product.has_stockrecord
- and line.quantity_without_discount > 0):
- price = getattr(product.stockrecord, self.price_field)
- value_of_matches += price * int(line.quantity_without_discount)
- if value_of_matches >= self.value:
- return True
- return False
-
- def _get_value_of_matches(self, basket):
- if hasattr(self, '_value_of_matches'):
- return getattr(self, '_value_of_matches')
- value_of_matches = Decimal('0.00')
- for line in basket.all_lines():
- product = line.product
- if (self.can_apply_condition(product) and product.has_stockrecord
- and line.quantity_without_discount > 0):
- price = getattr(product.stockrecord, self.price_field)
- value_of_matches += price * int(line.quantity_without_discount)
- self._value_of_matches = value_of_matches
- return value_of_matches
-
- def is_partially_satisfied(self, basket):
- value_of_matches = self._get_value_of_matches(basket)
- return Decimal('0.00') < value_of_matches < self.value
-
- def get_upsell_message(self, basket):
- value_of_matches = self._get_value_of_matches(basket)
- return _('Spend %(value)s more from %(range)s') % {'value': value_of_matches, 'range': self.range}
-
- def consume_items(self, basket, lines=None, value=None):
- """
- Marks items within the basket lines as consumed so they
- can't be reused in other offers.
-
- We allow lines to be passed in as sometimes we want them sorted
- in a specific order.
- """
- value_of_matches = Decimal('0.00')
- lines = lines or basket.all_lines()
- consumed_products = []
- value = self.value if value is None else value
- for line in basket.all_lines():
- product = line.product
- if (self.can_apply_condition(product) and product.has_stockrecord):
- price = getattr(product.stockrecord, self.price_field)
- if not price:
- continue
- quantity_to_consume = min(line.quantity_without_discount,
- int(((value - value_of_matches)/price).quantize(Decimal(1),
- ROUND_UP)))
- value_of_matches += price * quantity_to_consume
- line.consume(quantity_to_consume)
- consumed_products.extend((line.product,)*int(quantity_to_consume))
- if value_of_matches >= value:
- break
- return consumed_products
-
- # ========
- # Benefits
- # ========
-
- class PercentageDiscountBenefit(Benefit):
- """
- An offer benefit that gives a percentage discount
- """
-
- class Meta:
- proxy = True
- verbose_name = _("Percentage Discount Benefit")
- verbose_name_plural = _("Percentage Discount Benefits")
-
- def apply(self, basket, condition=None):
- discount = Decimal('0.00')
- affected_items = 0
- max_affected_items = self._effective_max_affected_items()
-
- for line in basket.all_lines():
- if affected_items >= max_affected_items:
- break
- product = line.product
- if (self.range.contains_product(product) and product.has_stockrecord
- and self.can_apply_benefit(product)):
- price = getattr(product.stockrecord, self.price_field)
- quantity = min(line.quantity_without_discount,
- max_affected_items - affected_items)
- line_discount = self.round(self.value/100 * price * int(quantity))
- line.discount(line_discount, quantity)
- affected_items += quantity
- discount += line_discount
-
- if discount > 0 and condition:
- condition.consume_items(basket)
- return discount
-
-
- class AbsoluteDiscountBenefit(Benefit):
- """
- An offer benefit that gives an absolute discount
- """
-
- class Meta:
- proxy = True
- verbose_name = _("Absolute Discount Benefit")
- verbose_name_plural = _("Absolute Discount Benefits")
-
- def apply(self, basket, condition=None):
- discount = Decimal('0.00')
- affected_items = 0
- max_affected_items = self._effective_max_affected_items()
-
- for line in basket.all_lines():
- if affected_items >= max_affected_items:
- break
- product = line.product
- if (self.range.contains_product(product) and product.has_stockrecord
- and self.can_apply_benefit(product)):
- price = getattr(product.stockrecord, self.price_field)
- if not price:
- # Avoid zero price products
- continue
- remaining_discount = self.value - discount
- quantity_affected = int(min(line.quantity_without_discount,
- max_affected_items - affected_items,
- math.ceil(remaining_discount / price)))
-
- # Update line with discounts
- line_discount = self.round(min(remaining_discount, quantity_affected * price))
- if condition:
- # Pass zero as quantity to avoid double consumption
- line.discount(line_discount, 0)
- else:
- line.discount(line_discount, quantity_affected)
-
- # Update loop vars
- affected_items += quantity_affected
- remaining_discount -= line_discount
- discount += line_discount
-
- if discount > 0 and condition:
- condition.consume_items(basket)
-
- return discount
-
-
- class FixedPriceBenefit(Benefit):
- """
- An offer benefit that gives the items in the condition for a
- fixed price. This is useful for "bundle" offers.
-
- Note that we ignore the benefit range here and only give a fixed price
- for the products in the condition range.
-
- The condition should be a count condition
- """
- class Meta:
- proxy = True
- verbose_name = _("Fixed Price Benefit")
- verbose_name_plural = _("Fixed Price Benefits")
-
- def apply(self, basket, condition=None):
- num_covered = 0
- num_permitted = int(condition.value)
- covered_lines = []
- product_total = Decimal('0.00')
- for line in basket.all_lines():
- product = line.product
- if (condition.range.contains_product(product) and line.quantity_without_discount > 0
- and self.can_apply_benefit(product)):
- # Line is available - determine quantity to consume and
- # record the total of the consumed products
- if isinstance(condition, CoverageCondition):
- quantity = 1
- else:
- quantity = min(line.quantity_without_discount, num_permitted - num_covered)
- num_covered += quantity
- product_total += quantity*line.unit_price_incl_tax
- covered_lines.append((line, quantity))
- if num_covered >= num_permitted:
- break
- discount = max(product_total - self.value, Decimal('0.00'))
-
- if not discount:
- return discount
-
- # Apply discount weighted by original value of line
- discount_applied = Decimal('0.00')
- last_line = covered_lines[-1][0]
- for line, quantity in covered_lines:
- if line == last_line:
- # If last line, we just take the difference to ensure that
- # a rounding doesn't lead to an off-by-one error
- line_discount = discount - discount_applied
- else:
- line_discount = self.round(discount * (line.unit_price_incl_tax * quantity) / product_total)
- line.discount(line_discount, quantity)
- discount_applied += line_discount
- return discount
-
-
- class MultibuyDiscountBenefit(Benefit):
-
- class Meta:
- proxy = True
- verbose_name = _("Multibuy Discount Benefit")
- verbose_name_plural = _("Multibuy Discount Benefits")
-
- def apply(self, basket, condition=None):
- benefit_lines = [line for line in basket.all_lines() if (self.range.contains_product(line.product) and
- line.quantity_without_discount > 0 and
- line.product.has_stockrecord and
- self.can_apply_benefit(line.product))]
- if not benefit_lines:
- return self.round(Decimal('0.00'))
-
- # Determine cheapest line to give for free
- line_price_getter = lambda line: getattr(line.product.stockrecord,
- self.price_field)
- free_line = min(benefit_lines, key=line_price_getter)
- discount = line_price_getter(free_line)
-
- if condition:
- compare = lambda l1, l2: cmp(line_price_getter(l2),
- line_price_getter(l1))
- lines_with_price = [line for line in basket.all_lines() if line.product.has_stockrecord]
- sorted_lines = sorted(lines_with_price, compare)
- free_line.discount(discount, 1)
- if condition.range.contains_product(free_line.product):
- condition.consume_items(basket, lines=sorted_lines,
- value=condition.value-1)
- else:
- condition.consume_items(basket, lines=sorted_lines)
- else:
- free_line.discount(discount, 0)
- return self.round(discount)
|