Synnefo

# Copyright 2013 GRNET S.A. All rights reserved.

#

# Redistribution and use in source and binary forms, with or

# without modification, are permitted provided that the following

# conditions are met:

#

#   1. Redistributions of source code must retain the above

#      copyright notice, this list of conditions and the following

#      disclaimer.

#

#   2. Redistributions in binary form must reproduce the above

#      copyright notice, this list of conditions and the following

#      disclaimer in the documentation and/or other materials

#      provided with the distribution.

#

# THIS SOFTWARE IS PROVIDED BY GRNET S.A. ``AS IS'' AND ANY EXPRESS

# OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED

# WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR

# PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL GRNET S.A OR

# CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,

# SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT

# LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF

# USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED

# AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT

# LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN

# ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE

# POSSIBILITY OF SUCH DAMAGE.

#

# The views and conclusions contained in the software and

# documentation are those of the authors and should not be

# interpreted as representing official policies, either expressed

# or implied, of GRNET S.A.

import re

from django.utils.translation import ugettext as _

from django.utils.encoding import smart_str, force_text

from django import forms

class EmailValidator(object):

    """

    Email validator. Backported from django 1.6

    """

    message = _('Enter a valid email address.')

    code = 'invalid'

    user_regex = 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)

    domain_regex = re.compile(

        r'(?:[A-Z0-9](?:[A-Z0-9-]{0,61}[A-Z0-9])?\.)+(?:[A-Z]{2,6}|[A-Z0-9-]{2,})$'  # domain

        # literal form, ipv4 address (SMTP 4.1.3)

        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)

    domain_whitelist = ['localhost']

    def __init__(self, message=None, code=None, whitelist=None):

        if message is not None:

            self.message = message

        if code is not None:

            self.code = code

        if whitelist is not None:

            self.domain_whitelist = whitelist

    def __call__(self, value):

        value = force_text(value)

        if not value or '@' not in value:

            raise forms.ValidationError(self.message, code=self.code)

        user_part, domain_part = value.rsplit('@', 1)

        if not self.user_regex.match(user_part):

            raise forms.ValidationError(self.message, code=self.code)

        if (not domain_part in self.domain_whitelist and

                not self.domain_regex.match(domain_part)):

            # Try for possible IDN domain-part

            try:

                domain_part = domain_part.encode('idna').decode('ascii')

                if not self.domain_regex.match(domain_part):

                    raise forms.ValidationError(self.message, code=self.code)

                else:

                    return

            except UnicodeError:

                pass

            raise forms.ValidationError(self.message, code=self.code)

class EmailField(forms.EmailField):

    default_validators = [EmailValidator()]