import re

import six
from wtforms import validators


class EmailValidator(object):
    """Validates wtform email field"""

    def __init__(self, message="The field data was not an email"):
        self._message = message

    def __call__(self, form, email):
        if not isinstance(email.data, six.text_type):
            raise validators.ValidationError(
                "Email fields should be of text type. Found {0}".format(
                    type(email.data)
                )
            )
        elif not re.match(r"[^@]+@[^@]+\.[^@]+", email.data):
            raise validators.ValidationError(self._message)