import six
from django.utils.functional import lazy
from django.utils.translation import gettext_lazy as _
from shuup.notify.base import Binding, Condition, ConstantUse
from shuup.notify.typology import Boolean, Integer, Language, Text
from shuup.utils.text import camel_case
[docs]
class NonEmpty(Condition):
identifier = "non_empty"
description = _("Check whether the bound value `value` exists and is non-empty and non-zero.")
name = _("Non-Empty")
v = Binding("Value")
[docs]
def test(self, context):
return bool(self.get_value(context, "v"))
[docs]
class Empty(Condition):
identifier = "empty"
description = _("Check whether the bound value `value` is empty or zero.")
name = _("Empty")
v = Binding("Value")
[docs]
def test(self, context):
return not bool(self.get_value(context, "v"))
[docs]
class BaseEqual(Condition):
identifier_suffix = "equal"
[docs]
def test(self, context):
value1 = self.get_value(context, "v1")
value2 = self.get_value(context, "v2")
if isinstance(value1, six.text_type) or isinstance(value2, six.text_type):
# When either value is a string, compare them stringly typed.
# (see http://c2.com/cgi/wiki?StringlyTyped)
return six.text_type(value1) == six.text_type(value2)
return value1 == value2
[docs]
class CaseInsensitiveStringEqual(Condition):
identifier_suffix = "equal"
[docs]
def test(self, context):
value1 = self.get_value(context, "v1")
value2 = self.get_value(context, "v2")
value1 = six.text_type(value1).lower().strip()
value2 = six.text_type(value2).lower().strip()
return value1 == value2
[docs]
class BooleanValuesEqual(Condition):
identifier_suffix = "equal"
[docs]
def test(self, context):
value1 = bool(self.get_value(context, "v1"))
value2 = bool(self.get_value(context, "v2"))
return value1 == value2
[docs]
def construct_simple(base, var_type):
identifier = f"{var_type.identifier}_{base.identifier_suffix}"
class_name = str(camel_case(identifier))
suffixed_type_name = lazy(lambda s: f"{var_type.name} {s}", six.text_type)
class_ns = {
"bindings": {
"v1": Binding(suffixed_type_name("1"), type=var_type),
"v2": Binding(
suffixed_type_name("2"),
type=var_type,
constant_use=ConstantUse.VARIABLE_OR_CONSTANT,
),
},
"identifier": identifier,
}
return type(class_name, (base,), class_ns)
LanguageEqual = construct_simple(CaseInsensitiveStringEqual, Language)
TextEqual = construct_simple(CaseInsensitiveStringEqual, Text)
IntegerEqual = construct_simple(BaseEqual, Integer)
BooleanEqual = construct_simple(BooleanValuesEqual, Boolean)