Source code for shuup.campaigns.admin_module.forms._basket_effects

from django.core.exceptions import ValidationError
from django.utils.translation import gettext_lazy as _

from shuup.admin.forms.fields import PercentageField
from shuup.campaigns.models.basket_effects import (
    BasketDiscountAmount,
    BasketDiscountPercentage,
    DiscountPercentageFromUndiscounted,
)
from shuup.campaigns.models.basket_line_effects import (
    DiscountFromCategoryProducts,
    DiscountFromProduct,
    FreeProductLine,
)
from shuup.core.models import Category, Product, ShopProduct

from ._base import BaseEffectModelForm



[docs]
class BasketDiscountAmountForm(BaseEffectModelForm):

[docs]
    class Meta(BaseEffectModelForm.Meta):
        model = BasketDiscountAmount





[docs]
class BasketDiscountPercentageForm(BaseEffectModelForm):
    discount_percentage = PercentageField(
        max_digits=6,
        decimal_places=5,
        label=_("discount percentage"),
        help_text=_("The discount percentage for this campaign."),
    )


[docs]
    class Meta(BaseEffectModelForm.Meta):
        model = BasketDiscountPercentage





[docs]
class DiscountPercentageFromUndiscountedForm(BaseEffectModelForm):
    discount_percentage = PercentageField(
        max_digits=6,
        decimal_places=5,
        label=_("discount percentage"),
        help_text=_("The discount percentage for this campaign."),
    )


[docs]
    class Meta(BaseEffectModelForm.Meta):
        model = DiscountPercentageFromUndiscounted





[docs]
class FreeProductLineForm(BaseEffectModelForm):

[docs]
    class Meta(BaseEffectModelForm.Meta):
        model = FreeProductLine



[docs]
    def clean(self):
        super().clean()
        # Don't validate data is form is deleted
        if self.cleaned_data.get("DELETE"):
            return
        campaign = self.cleaned_data["campaign"]
        products = self.cleaned_data.get("products", [])
        for product_id in products:
            product = Product.objects.get(pk=product_id)
            try:
                shop_product = product.get_shop_instance(campaign.shop)
            except ShopProduct.DoesNotExist as exc:
                raise ValidationError(
                    _("Product %(product)s is not available in the %(shop)s shop.")
                    % {"product": product, "shop": campaign.shop.name}
                ) from exc
            for error in shop_product.get_quantity_errors(self.cleaned_data["quantity"], False):
                raise ValidationError({"quantity": error.message})





[docs]
class DiscountFromProductForm(BaseEffectModelForm):

[docs]
    class Meta(BaseEffectModelForm.Meta):
        model = DiscountFromProduct





[docs]
class DiscountFromCategoryProductsForm(BaseEffectModelForm):
    discount_percentage = PercentageField(
        max_digits=6,
        decimal_places=5,
        label=_("discount percentage"),
        required=False,
        help_text=_("The discount percentage for this campaign."),
    )


[docs]
    class Meta(BaseEffectModelForm.Meta):
        model = DiscountFromCategoryProducts



[docs]
    def __init__(self, **kwargs):
        super().__init__(**kwargs)
        self.fields["category"].queryset = Category.objects.all_except_deleted()



[docs]
    def clean(self):
        data = self.cleaned_data
        if data["discount_amount"] and data["discount_percentage"]:
            msg = _("Only amount or percentage can be set, not both.")
            self.add_error("discount_amount", msg)
            self.add_error("discount_percentage", msg)
        return data