from uuid import uuid4
from django.utils.translation import gettext_lazy as _
from shuup.campaigns.models.campaigns import BasketCampaign, CatalogCampaign, CouponUsage
from shuup.core.models import OrderLineType, ShopProduct
from shuup.core.order_creator import OrderSourceModifierModule
from shuup.core.order_creator._source import LineSource
from shuup.core.pricing import DiscountModule, PriceInfo
# DEPRECATED: 3.0
# This discount module will be completely removed on next major version
[docs]
class CatalogCampaignModule(DiscountModule):
identifier = "catalog_campaigns"
name = _("Campaigns")
[docs]
def discount_price(self, context, product, price_info: PriceInfo):
"""
Get the discounted price for context.
Best discount is selected.
Minimum price will be selected if the cheapest price is under that.
"""
create_price = context.shop.create_price
shop_product = ShopProduct.objects.filter(shop=context.shop, product=product).first()
if not shop_product:
return price_info
best_discount = None
for campaign in CatalogCampaign.get_matching(context, shop_product):
price = price_info.price
# get first matching effect
for effect in campaign.effects.all():
price -= effect.apply_for_product(context=context, product=product, price_info=price_info)
if best_discount is None:
best_discount = price
if price < best_discount:
best_discount = price
if not best_discount:
return price_info
if shop_product.minimum_price and best_discount < shop_product.minimum_price:
best_discount = shop_product.minimum_price
if best_discount < create_price("0"):
best_discount = create_price("0")
return PriceInfo(
best_discount,
base_price=price_info.base_price,
quantity=price_info.quantity,
expires_on=price_info.expires_on,
)
[docs]
class BasketCampaignModule(OrderSourceModifierModule):
identifier = "basket_campaigns"
name = _("Campaign Basket Discounts")
[docs]
def get_new_lines(self, order_source, lines):
matching_campaigns = BasketCampaign.get_matching(order_source, lines)
for line in self._handle_line_effects(matching_campaigns, order_source, lines):
yield line
# total discounts must be run after line effects since lines can be changed in place
for line in self._handle_total_discount_effects(matching_campaigns, order_source, lines):
yield line
def _get_campaign_line(self, campaign, highest_discount, order_source, supplier):
text = campaign.public_name
if campaign.coupon:
text += " ({} {})".format(_("Coupon Code:"), campaign.coupon.code)
return order_source.create_line(
line_id=f"discount_{uuid4().hex}",
type=OrderLineType.DISCOUNT,
quantity=1,
discount_amount=campaign.shop.create_price(highest_discount),
text=text,
line_source=LineSource.DISCOUNT_MODULE,
supplier=supplier,
)
[docs]
def can_use_code(self, order_source, code):
campaigns = BasketCampaign.objects.filter(
active=True,
shop=order_source.shop,
coupon__code__iexact=code,
coupon__active=True,
)
for campaign in campaigns:
if not campaign.is_available():
continue
coupon_code = campaign.coupon
suppliers = {supplier for supplier in (campaign.supplier, coupon_code.supplier) if supplier}
if suppliers:
has_supplier = False
# make sure there is at least one item in the order source that has this supplier
has_supplier = False
for line in order_source.get_final_lines():
if line.supplier in suppliers:
has_supplier = True
break
# there is no line that matches the coupon or the campaign supplier
if not has_supplier:
return False
return campaign.coupon.can_use_code(order_source.customer)
return False
[docs]
def use_code(self, order, code):
campaigns = BasketCampaign.objects.filter(
active=True, shop=order.shop, coupon__code__iexact=code, coupon__active=True
)
for campaign in campaigns:
campaign.coupon.use(order)
[docs]
def clear_codes(self, order):
CouponUsage.objects.filter(order=order).delete()
def _handle_total_discount_effects(self, matching_campaigns, order_source, original_lines):
price_so_far = sum((x.price for x in original_lines), order_source.zero_price)
def get_discount_line(campaign, amount, price_so_far, supplier):
new_amount = min(amount, price_so_far)
price_so_far -= new_amount
return self._get_campaign_line(campaign, new_amount, order_source, supplier)
best_discount_for_supplier = {}
lines = []
for campaign in matching_campaigns:
campaign_supplier = getattr(campaign, "supplier", None)
for effect in campaign.discount_effects.all():
discount_amount = min(price_so_far, effect.apply_for_basket(order_source=order_source))
# if campaign has coupon, match it to order_source.codes
if campaign.coupon:
# campaign was found because discount code matched. This line is always added
lines.append(get_discount_line(campaign, discount_amount, price_so_far, campaign_supplier))
else:
best_discount = best_discount_for_supplier.get(campaign_supplier)
if not best_discount or discount_amount > best_discount["discount_amount"]:
best_discount_for_supplier[campaign_supplier] = {
"discount_amount": discount_amount,
"campaign": campaign,
}
for supplier, best_discount_info in best_discount_for_supplier.items():
lines.append(
get_discount_line(
best_discount_info["campaign"],
best_discount_info["discount_amount"],
price_so_far,
supplier,
)
)
return lines
def _handle_line_effects(self, matching_campaigns, order_source, original_lines):
lines = []
for campaign in matching_campaigns:
campaign_supplier = getattr(campaign, "supplier", None)
for effect in campaign.line_effects.all():
lines += effect.get_discount_lines(
order_source=order_source,
original_lines=original_lines,
supplier=campaign_supplier,
)
return lines