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

# This file is part of Shuup.
#
# Copyright (c) 2012-2021, Shuup Commerce Inc. All rights reserved.
#
# This source code is licensed under the OSL-3.0 license found in the
# LICENSE file in the root directory of this source tree.
from django.core.exceptions import ValidationError
from django.utils.translation import ugettext_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


class BasketDiscountAmountForm(BaseEffectModelForm):
    class Meta(BaseEffectModelForm.Meta):
        model = BasketDiscountAmount


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

    class Meta(BaseEffectModelForm.Meta):
        model = BasketDiscountPercentage


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

    class Meta(BaseEffectModelForm.Meta):
        model = DiscountPercentageFromUndiscounted


class FreeProductLineForm(BaseEffectModelForm):
    class Meta(BaseEffectModelForm.Meta):
        model = FreeProductLine

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


class DiscountFromProductForm(BaseEffectModelForm):
    class Meta(BaseEffectModelForm.Meta):
        model = DiscountFromProduct


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."),
    )

    class Meta(BaseEffectModelForm.Meta):
        model = DiscountFromCategoryProducts

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

    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