Ticket #4833: 4833-3.diff

django/core/validators.py

@@ -138 +138 @@
         raise ValidationError('')
 
 class EmailValidator(RegexValidator):
+    """
+    EmailValidator takes a tuple of regexes on init, one for user part and one
+    for domain part
+    """
+    domain_whitelist = ['localhost']
 
     def __call__(self, value):
-        try:
-            super(EmailValidator, self).__call__(value)
-        except ValidationError, e:
-            # Trivial case failed. Try for possible IDN domain-part
-            if value and u'@' in value:
-                parts = value.split(u'@')
+        if value and u'@' in value:
+            user_part, domain_part = value.split(u'@', 1)
+            res = self.regex[0].search(smart_unicode(user_part))
+            if not self.regex[0].search(smart_unicode(user_part)):
+                raise ValidationError(self.message, code=self.code)
+            if (not smart_unicode(domain_part) in self.domain_whitelist and
+                    not self.regex[1].search(smart_unicode(domain_part))):
+                # Try for possible IDN domain-part
                 try:
-                    parts[-1] = parts[-1].encode('idna')
+                    domain_part = domain_part.encode('idna')
+                    if not self.regex[1].search(smart_unicode(domain_part)):
+                        raise ValidationError(self.message, code=self.code)
+                    else:
+                        return
                 except UnicodeError:
-                    raise e
-                super(EmailValidator, self).__call__(u'@'.join(parts))
-            else:
-                raise
+                    pass
+                raise ValidationError(self.message, code=self.code)
+        else:
+            raise ValidationError(self.message, code=self.code)
 
-email_re = re.compile(
-    r"(^[-!#$%&'*+/=?^_`{}|~0-9A-Z]+(\.[-!#$%&'*+/=?^_`{}|~0-9A-Z]+)*"  # dot-atom
-    r'|^"([\001-\010\013\014\016-\037!#-\[\]-\177]|\\[\001-011\013\014\016-\177])*"' # quoted-string
-    r')@((?:[A-Z0-9](?:[A-Z0-9-]{0,61}[A-Z0-9])?\.)+[A-Z]{2,6}\.?$)'  # domain
-    r'|\[(25[0-5]|2[0-4]\d|[0-1]?\d?\d)(\.(25[0-5]|2[0-4]\d|[0-1]?\d?\d)){3}\]$', re.IGNORECASE)  # literal form, ipv4 address (SMTP 4.1.3)
+email_re = (
+    re.compile(
+        r"(^[-!#$%&'*+/=?^_`{}|~0-9A-Z]+(\.[-!#$%&'*+/=?^_`{}|~0-9A-Z]+)*$"  # dot-atom
+        r'|^"([\001-\010\013\014\016-\037!#-\[\]-\177]|\\[\001-011\013\014\016-\177])*"$)', # quoted-string
+        re.IGNORECASE
+    ),
+    re.compile(
+        r'(^(?:[A-Z0-9](?:[A-Z0-9-]{0,61}[A-Z0-9])?\.)+[A-Z]{2,6}\.?$)'  # domain
+        r'|^\[(25[0-5]|2[0-4]\d|[0-1]?\d?\d)(\.(25[0-5]|2[0-4]\d|[0-1]?\d?\d)){3}\]$', # literal form, ipv4 address (SMTP 4.1.3)
+        re.IGNORECASE
+    )
+)
 validate_email = EmailValidator(email_re, _(u'Enter a valid e-mail address.'), 'invalid')
 
 slug_re = re.compile(r'^[-\w]+$')

tests/modeltests/validators/tests.py

@@ -23 +23 @@
     (validate_email, 'email@here.com', None),
     (validate_email, 'weirder-email@here.and.there.com', None),
     (validate_email, 'email@[127.0.0.1]', None),
+    (validate_email, 'email@localhost', None),
 
     (validate_email, None, ValidationError),
     (validate_email, '', ValidationError),