Files
@ 8ad265a65a08
Branch filter:
Location: symposion_app/registrasion/controllers/discount.py - annotation
8ad265a65a08
3.2 KiB
text/x-python
Fixes tests now that $0 invoices pay themselves
fb3878ce2e44 fb3878ce2e44 fb3878ce2e44 fb3878ce2e44 fb3878ce2e44 fb3878ce2e44 fb3878ce2e44 fb3878ce2e44 fb3878ce2e44 fb3878ce2e44 fb3878ce2e44 fb3878ce2e44 fb3878ce2e44 fb3878ce2e44 cf85af771983 cf85af771983 cf85af771983 cf85af771983 cf85af771983 fb3878ce2e44 fb3878ce2e44 fb3878ce2e44 fb3878ce2e44 fb3878ce2e44 fb3878ce2e44 fb3878ce2e44 fb3878ce2e44 fb3878ce2e44 fb3878ce2e44 fb3878ce2e44 fb3878ce2e44 fb3878ce2e44 fb3878ce2e44 fb3878ce2e44 fb3878ce2e44 fb3878ce2e44 fb3878ce2e44 fb3878ce2e44 fb3878ce2e44 fb3878ce2e44 fb3878ce2e44 fb3878ce2e44 fb3878ce2e44 fb3878ce2e44 fb3878ce2e44 fb3878ce2e44 fb3878ce2e44 fb3878ce2e44 fb3878ce2e44 fb3878ce2e44 fb3878ce2e44 fb3878ce2e44 fb3878ce2e44 fb3878ce2e44 fb3878ce2e44 fb3878ce2e44 fb3878ce2e44 fb3878ce2e44 fb3878ce2e44 fb3878ce2e44 fb3878ce2e44 fb3878ce2e44 fb3878ce2e44 fb3878ce2e44 cf85af771983 fb3878ce2e44 fb3878ce2e44 fb3878ce2e44 fb3878ce2e44 fb3878ce2e44 fb3878ce2e44 fb3878ce2e44 fb3878ce2e44 fb3878ce2e44 fb3878ce2e44 fb3878ce2e44 fb3878ce2e44 1e7a2abc7f92 fb3878ce2e44 fb3878ce2e44 fb3878ce2e44 fb3878ce2e44 fb3878ce2e44 fb3878ce2e44 fb3878ce2e44 fb3878ce2e44 fb3878ce2e44 fb3878ce2e44 fb3878ce2e44 | import itertools
from conditions import ConditionController
from registrasion import models as rego
from django.db.models import Sum
class DiscountAndQuantity(object):
def __init__(self, discount, clause, quantity):
self.discount = discount
self.clause = clause
self.quantity = quantity
def __repr__(self):
print "(discount=%s, clause=%s, quantity=%d)" % (
self.discount, self.clause, self.quantity,
)
def available_discounts(user, categories, products):
''' Returns all discounts available to this user for the given categories
and products. The discounts also list the available quantity for this user,
not including products that are pending purchase. '''
# discounts that match provided categories
category_discounts = rego.DiscountForCategory.objects.filter(
category__in=categories
)
# discounts that match provided products
product_discounts = rego.DiscountForProduct.objects.filter(
product__in=products
)
# discounts that match categories for provided products
product_category_discounts = rego.DiscountForCategory.objects.filter(
category__in=(product.category for product in products)
)
# (Not relevant: discounts that match products in provided categories)
# The set of all potential discounts
potential_discounts = set(itertools.chain(
product_discounts,
category_discounts,
product_category_discounts,
))
discounts = []
# Markers so that we don't need to evaluate given conditions more than once
accepted_discounts = set()
failed_discounts = set()
for discount in potential_discounts:
real_discount = rego.DiscountBase.objects.get_subclass(
pk=discount.discount.pk,
)
cond = ConditionController.for_condition(real_discount)
# Count the past uses of the given discount item.
# If this user has exceeded the limit for the clause, this clause
# is not available any more.
past_uses = rego.DiscountItem.objects.filter(
cart__user=user,
cart__active=False, # Only past carts count
cart__released=False, # You can reuse refunded discounts
discount=discount.discount,
)
agg = past_uses.aggregate(Sum("quantity"))
past_use_count = agg["quantity__sum"]
if past_use_count is None:
past_use_count = 0
if past_use_count >= discount.quantity:
# This clause has exceeded its use count
pass
elif real_discount not in failed_discounts:
# This clause is still available
if real_discount in accepted_discounts or cond.is_met(user):
# This clause is valid for this user
discounts.append(DiscountAndQuantity(
discount=real_discount,
clause=discount,
quantity=discount.quantity - past_use_count,
))
accepted_discounts.add(real_discount)
else:
# This clause is not valid for this user
failed_discounts.add(real_discount)
return discounts
|