xeBuz/Flask-Validator

""" Comparision Others 

- ValidateRegex
- ValidateUUID
- ValidateISBN
- ValidateRange
"""

import re
from uuid import UUID
from isbnlib import is_isbn10, is_isbn13
from flask_validator import Validator


class ValidateRegex(Validator):
    """ Validate Regex

    Compare a value against a regular expresion

    Args:
        field: SQLAlchemy column to validate
        regex: Regular expresion
        allow_null: (bool) Allow null values
        throw_exception: (bool) Throw a ValidateError if the validation fails

    """

    regex = None

    def __init__(self, field, regex, allow_null=False, throw_exception=False, message=None):
        try:
            self.regex = regex
            re.compile(regex)
        except re.error:
            raise AttributeError('Invalid Regex')

        Validator.__init__(self, field, allow_null, throw_exception, message)

    def check_value(self, value):
        if re.match(self.regex, value):
            return True
        else:
            return False


class ValidateUUID(Validator):
    """ Validate UUID

    Validate if the value is a valid UUID

    Args:
        field: SQLAlchemy column to validate
        version: UUUID version. Default 4
        allow_null: (bool) Allow null values
        throw_exception: (bool) Throw a ValidateError if the validation fails

    """
    version = 4

    def __init__(self, field, version=4, allow_null=True, throw_exception=False, message=None):
        self.version = version

        Validator.__init__(self, field, allow_null, throw_exception, message)

    def check_value(self, value):
        try:
            val = UUID(value, version=self.version)
            if val:
                return True
            else:
                return False
        except (ValueError, TypeError):
            return False


class ValidateISBN(Validator):
    """ Validate ISBN

    Validate if the value is a valid ISBN10 or ISB13

    Args:
        field: SQLAlchemy column to validate
        allow_null: (bool) Allow null values
        throw_exception: (bool) Throw a ValidateError if the validation fails

    """
    def check_value(self, value):
        return is_isbn10(value) or is_isbn13(value)


class ValidateRange(Validator):
    """ Validate Range

    Validate if the value is in a specific value range

    Args:
        field: SQLAlchemy column to validate
        allow_null: (bool) Allow null values
        throw_exception: (bool) Throw a ValidateError if the validation fails

    """
    range_valid = None

    def __init__(self, field, range_valid, allow_null=True, throw_exception=False, message=None):
        self.range_valid = range_valid

        Validator.__init__(self, field, allow_null, throw_exception, message)

    def check_value(self, value):
        try:
            return value in self.range_valid
        except TypeError:
            return False