Source code for shuup.campaigns.models.contact_group_sales_ranges

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

from shuup.campaigns.utils.sales_range import get_contacts_in_sales_range
from shuup.core.fields import MoneyValueField
from shuup.core.models import ContactGroup, Shop




class SalesRangeQuerySet(models.QuerySet):


    def active(self, shop):
        query = Q(shop=shop)
        # Min value needs to be set
        query &= Q(min_value__isnull=False)
        # If max value is not null it can't be zero
        query &= ~Q(Q(max_value__isnull=False) & Q(max_value=0))
        return self.filter(query)






class ContactGroupSalesRange(models.Model):
    group = models.ForeignKey(
        ContactGroup,
        related_name="+",
        on_delete=models.CASCADE,
        verbose_name=_("group"),
    )
    shop = models.ForeignKey(on_delete=models.CASCADE, to=Shop, related_name="+", verbose_name=_("shop"))
    min_value = MoneyValueField(verbose_name=_("min amount"), blank=True, null=True)
    max_value = MoneyValueField(verbose_name=_("max amount"), blank=True, null=True)

    objects = SalesRangeQuerySet.as_manager()

    class Meta:
        unique_together = ("group", "shop")



    def save(self, *args, **kwargs):
        self.clean()
        super().save(*args, **kwargs)
        if self.is_active():  # Update group members only if the range is still active
            contact_ids = get_contacts_in_sales_range(self.shop, self.min_value, self.max_value)
            self.group.members.set(contact_ids)




    def clean(self):
        super().clean()
        if self.group.is_protected:
            raise ValidationError(_("Can not add sales limits for default contact groups."))




    def is_active(self):
        return bool(self.min_value is not None and (self.max_value is None or self.max_value > 0))