Ticket #3250: brian_validators.diff

.py

@@ -296 +296 @@
         if not all_data.get(self.other, False) and not field_data:
             raise ValidationError, self.error_message
 
+def RequiredIfOtherFieldsNotGiven(other_field_names, error_message=None):
+    if error_message is None:
+        from django.utils.text import get_text_list
+        fields = get_text_list(other_field_names)
+        error_message = gettext_lazy("This field must be given if %s are not given." % fields)
+    validators = [RequiredIfOtherFieldNotGiven(field) for field in other_field_names]
+    return AllValidators(validators, error_message)
+
+def RequiredIfAllOtherFieldsNotGiven(other_field_names, error_message=None):
+    if error_message is None:
+        from django.utils.text import get_text_list
+        fields = get_text_list(other_field_names, 'and')
+        error_message = gettext_lazy("This field must be given if %s are not given." % fields)
+    validators = [RequiredIfOtherFieldNotGiven(field) for field in other_field_names]
+    return AnyValidator(validators, error_message)
+
 class RequiredIfOtherFieldsGiven(object):
     def __init__(self, other_field_names, error_message=gettext_lazy("Please enter both fields or leave them both empty.")):
         self.other, self.error_message = other_field_names, error_message
@@ -481 +497 @@
             except ValidationError, e:
                 pass
         raise ValidationError(self.error_message)
+
+
+class AllValidators(object):
+    """
+    This validator tries all given validators. If all of them succeed,
+    validation passes. If any of them do not succeed, the given message is
+    thrown as a validation error. The message is rather unspecific, so it's
+    best to specify one on instantiation.
+    """
+    def __init__(self, validator_list=None, error_message=gettext_lazy("This field is invalid.")):
+        if validator_list is None: validator_list = []
+        self.validator_list = validator_list
+        self.error_message = error_message
+        for v in validator_list:
+            if hasattr(v, 'always_test'):
+                self.always_test = True
+
+    def __call__(self, field_data, all_data):
+        for v in self.validator_list:
+            try:
+                v(field_data, all_data)
+            except ValidationError, e:
+                raise ValidationError(self.error_message)
 
 class URLMimeTypeCheck(object):
     "Checks that the provided URL points to a document with a listed mime type"