Source code for shuup.campaigns.admin_module.form_sets

from typing import Any, Optional, Type

from django.conf import settings
from django.forms import BaseModelFormSet

from shuup.campaigns.models import (
    BasketCondition,
    BasketDiscountEffect,
    BasketLineEffect,
    CatalogFilter,
    ContextCondition,
    ProductDiscountEffect,
)
from shuup.utils.multilanguage_model_form import TranslatableModelForm




class BaseFormset(BaseModelFormSet):
    form_class: Optional[Type[Any]] = None  # Set in initialization
    model = None  # type: ignore[assignment]  # Override this in subclass

    validate_min = False
    min_num = 0
    validate_max = False
    max_num = 20
    absolute_max = 20
    can_delete = True
    can_order = False
    extra = 0



    def __init__(self, *args, **kwargs):
        self.form_class = kwargs.pop("form")
        self.owner = kwargs.pop("owner")
        super().__init__(*args, **kwargs)




    def get_name(self):
        return getattr(self._get_actual_model(), "name", None)


    def _get_actual_model(self):
        return self.form_class._meta.model



    def get_queryset(self):
        raise NotImplementedError("Override this in subclass")




    def form(self, **kwargs):
        if issubclass(self.form_class, TranslatableModelForm):
            kwargs.setdefault("languages", settings.LANGUAGES)
            kwargs.setdefault("default_language", settings.PARLER_DEFAULT_LANGUAGE_CODE)
        return self.form_class(**kwargs)


    @property
    def empty_form(self):
        form = self._construct_form(self.total_form_count())
        form.empty_permitted = True
        return form

    @property
    def can_delete_extra(self):
        return self.can_delete





class BasketConditionsFormSet(BaseFormset):
    model = BasketCondition  # type: ignore[assignment]



    def get_queryset(self):
        return self.owner.conditions.instance_of(self._get_actual_model())


    @property
    def empty_form(self):
        form = self._construct_form(self.total_form_count())
        form.empty_permitted = True
        return form

    @property
    def can_delete_extra(self):
        return False





class EffectsFormset(BaseFormset):


    def form(self, **kwargs):
        kwargs.setdefault("initial", {"campaign": self.owner})
        return super().form(**kwargs)


    @property
    def empty_form(self):
        form = self._construct_form(self.total_form_count())
        form.empty_permitted = True
        return form

    @property
    def can_delete_extra(self):
        return self.can_delete





class BasketDiscountEffectsFormSet(EffectsFormset):
    model = BasketDiscountEffect  # type: ignore[assignment]



    def get_queryset(self):
        return self.owner.discount_effects.instance_of(self._get_actual_model())






class BasketLineEffectsFormSet(EffectsFormset):
    model = BasketLineEffect  # type: ignore[assignment]



    def get_queryset(self):
        return self.owner.line_effects.instance_of(self._get_actual_model())






class CatalogConditionsFormSet(BaseFormset):
    model = ContextCondition  # type: ignore[assignment]



    def get_queryset(self):
        return self.owner.conditions.instance_of(self._get_actual_model())


    @property
    def empty_form(self):
        form = self._construct_form(self.total_form_count())
        form.empty_permitted = True
        return form

    @property
    def can_delete_extra(self):
        return False





class CatalogFiltersFormSet(BaseFormset):
    model = CatalogFilter  # type: ignore[assignment]



    def get_queryset(self):
        return self.owner.filters.instance_of(self._get_actual_model())


    @property
    def empty_form(self):
        form = self._construct_form(self.total_form_count())
        form.empty_permitted = True
        return form

    @property
    def can_delete_extra(self):
        return False





class CatalogEffectsFormSet(EffectsFormset):
    model = ProductDiscountEffect  # type: ignore[assignment]



    def get_queryset(self):
        return self.owner.effects.instance_of(self._get_actual_model())