Source code for shuup.campaigns.models.contact_group_sales_ranges

# -*- coding: utf-8 -*-
# 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.db import models
from django.db.models import Q
from django.utils.translation import ugettext_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(ContactGroupSalesRange, self).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(ContactGroupSalesRange, self).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))