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

from django import forms

from shuup.admin.forms.fields import WeekdayField
from shuup.admin.forms.widgets import TimeInput
from shuup.campaigns.models.basket_conditions import (
    BasketMaxTotalAmountCondition,
    BasketMaxTotalProductAmountCondition,
    BasketTotalAmountCondition,
    BasketTotalProductAmountCondition,
    BasketTotalUndiscountedProductAmountCondition,
    CategoryProductsBasketCondition,
    ChildrenProductCondition,
    ContactBasketCondition,
    ContactGroupBasketCondition,
    HourBasketCondition,
    ProductsInBasketCondition,
)
from shuup.core.models import Category

from ._base import BaseRuleModelForm




class BasketTotalProductAmountConditionForm(BaseRuleModelForm):


    class Meta(BaseRuleModelForm.Meta):
        model = BasketTotalProductAmountCondition






class BasketTotalAmountConditionForm(BaseRuleModelForm):


    class Meta(BaseRuleModelForm.Meta):
        model = BasketTotalAmountCondition






class BasketTotalUndiscountedProductAmountConditionForm(BaseRuleModelForm):


    class Meta(BaseRuleModelForm.Meta):
        model = BasketTotalUndiscountedProductAmountCondition






class ProductsInBasketConditionForm(BaseRuleModelForm):


    class Meta(BaseRuleModelForm.Meta):
        model = ProductsInBasketCondition




    def __init__(self, *args, **kwargs):
        super().__init__(*args, **kwargs)
        self.fields["products"].widget = forms.SelectMultiple(attrs={"data-model": "shuup.product"})






class ContactGroupBasketConditionForm(BaseRuleModelForm):


    class Meta(BaseRuleModelForm.Meta):
        model = ContactGroupBasketCondition






class ContactBasketConditionForm(BaseRuleModelForm):


    class Meta(BaseRuleModelForm.Meta):
        model = ContactBasketCondition






class BasketMaxTotalProductAmountConditionForm(BaseRuleModelForm):


    class Meta(BaseRuleModelForm.Meta):
        model = BasketMaxTotalProductAmountCondition






class BasketMaxTotalAmountConditionForm(BaseRuleModelForm):


    class Meta(BaseRuleModelForm.Meta):
        model = BasketMaxTotalAmountCondition






class ChildrenProductConditionForm(BaseRuleModelForm):


    def __init__(self, *args, **kwargs):
        super().__init__(*args, **kwargs)
        self.fields["product"].widget = forms.Select(
            attrs={"data-model": "shuup.product", "data-search-mode": "parent_product"}
        )




    class Meta(BaseRuleModelForm.Meta):
        model = ChildrenProductCondition






class CategoryProductsBasketConditionForm(BaseRuleModelForm):


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




    class Meta(BaseRuleModelForm.Meta):
        model = CategoryProductsBasketCondition






class HourBasketConditionForm(BaseRuleModelForm):
    days = WeekdayField()



    class Meta(BaseRuleModelForm.Meta):
        model = HourBasketCondition
        widgets = {
            "hour_start": TimeInput(),
            "hour_end": TimeInput(),
        }