92 lines
2.7 KiB
Python
92 lines
2.7 KiB
Python
|
from django.core.exceptions import ValidationError
|
||
|
|
from django.core.validators import (
|
||
|
|
MaxLengthValidator,
|
||
|
|
MaxValueValidator,
|
||
|
|
MinLengthValidator,
|
||
|
|
MinValueValidator,
|
||
|
|
)
|
||
|
|
from django.utils.deconstruct import deconstructible
|
||
|
|
from django.utils.translation import gettext_lazy as _
|
||
|
|
from django.utils.translation import ngettext_lazy
|
||
|
|
|
||
|
|
|
||
|
|
class ArrayMaxLengthValidator(MaxLengthValidator):
|
||
|
|
message = ngettext_lazy(
|
||
|
|
"List contains %(show_value)d item, it should contain no more than "
|
||
|
|
"%(limit_value)d.",
|
||
|
|
"List contains %(show_value)d items, it should contain no more than "
|
||
|
|
"%(limit_value)d.",
|
||
|
|
"limit_value",
|
||
|
|
)
|
||
|
|
|
||
|
|
|
||
|
|
class ArrayMinLengthValidator(MinLengthValidator):
|
||
|
|
message = ngettext_lazy(
|
||
|
|
"List contains %(show_value)d item, it should contain no fewer than "
|
||
|
|
"%(limit_value)d.",
|
||
|
|
"List contains %(show_value)d items, it should contain no fewer than "
|
||
|
|
"%(limit_value)d.",
|
||
|
|
"limit_value",
|
||
|
|
)
|
||
|
|
|
||
|
|
|
||
|
|
@deconstructible
|
||
|
|
class KeysValidator:
|
||
|
|
"""A validator designed for HStore to require/restrict keys."""
|
||
|
|
|
||
|
|
messages = {
|
||
|
|
"missing_keys": _("Some keys were missing: %(keys)s"),
|
||
|
|
"extra_keys": _("Some unknown keys were provided: %(keys)s"),
|
||
|
|
}
|
||
|
|
strict = False
|
||
|
|
|
||
|
|
def __init__(self, keys, strict=False, messages=None):
|
||
|
|
self.keys = set(keys)
|
||
|
|
self.strict = strict
|
||
|
|
if messages is not None:
|
||
|
|
self.messages = {**self.messages, **messages}
|
||
|
|
|
||
|
|
def __call__(self, value):
|
||
|
|
keys = set(value)
|
||
|
|
missing_keys = self.keys - keys
|
||
|
|
if missing_keys:
|
||
|
|
raise ValidationError(
|
||
|
|
self.messages["missing_keys"],
|
||
|
|
code="missing_keys",
|
||
|
|
params={"keys": ", ".join(missing_keys)},
|
||
|
|
)
|
||
|
|
if self.strict:
|
||
|
|
extra_keys = keys - self.keys
|
||
|
|
if extra_keys:
|
||
|
|
raise ValidationError(
|
||
|
|
self.messages["extra_keys"],
|
||
|
|
code="extra_keys",
|
||
|
|
params={"keys": ", ".join(extra_keys)},
|
||
|
|
)
|
||
|
|
|
||
|
|
def __eq__(self, other):
|
||
|
|
return (
|
||
|
|
isinstance(other, self.__class__)
|
||
|
|
and self.keys == other.keys
|
||
|
|
and self.messages == other.messages
|
||
|
|
and self.strict == other.strict
|
||
|
|
)
|
||
|
|
|
||
|
|
|
||
|
|
class RangeMaxValueValidator(MaxValueValidator):
|
||
|
|
def compare(self, a, b):
|
||
|
|
return a.upper is None or a.upper > b
|
||
|
|
|
||
|
|
message = _(
|
||
|
|
"Ensure that this range is completely less than or equal to %(limit_value)s."
|
||
|
|
)
|
||
|
|
|
||
|
|
|
||
|
|
class RangeMinValueValidator(MinValueValidator):
|
||
|
|
def compare(self, a, b):
|
||
|
|
return a.lower is None or a.lower < b
|
||
|
|
|
||
|
|
message = _(
|
||
|
|
"Ensure that this range is completely greater than or equal to %(limit_value)s."
|
||
|
|
)
|