98 lines
2.0 KiB
Python
98 lines
2.0 KiB
Python
from abc import ABC, abstractmethod
|
|
from functools import reduce
|
|
from operator import or_
|
|
import re
|
|
|
|
|
|
class ValidatorException(Exception):
|
|
pass
|
|
|
|
|
|
class Validator(ABC):
|
|
@staticmethod
|
|
def of(x):
|
|
if isinstance(x, Validator):
|
|
return x
|
|
if callable(x):
|
|
return LambdaValidator(x)
|
|
if x in validator_types:
|
|
return validator_types[x]()
|
|
raise ValidatorException("Cannot create validator")
|
|
|
|
@abstractmethod
|
|
def isValid(self, value):
|
|
pass
|
|
|
|
|
|
class LambdaValidator(Validator):
|
|
def __init__(self, c):
|
|
self.callable = c
|
|
|
|
def isValid(self, value):
|
|
return self.callable(value)
|
|
|
|
|
|
class TypeValidator(Validator):
|
|
def __init__(self, type):
|
|
self.type = type
|
|
super().__init__()
|
|
|
|
def isValid(self, value):
|
|
return isinstance(value, self.type)
|
|
|
|
|
|
class IntegerValidator(TypeValidator):
|
|
def __init__(self):
|
|
super().__init__(int)
|
|
|
|
|
|
class FloatValidator(TypeValidator):
|
|
def __init__(self):
|
|
super().__init__(float)
|
|
|
|
|
|
class StringValidator(TypeValidator):
|
|
def __init__(self):
|
|
super().__init__(str)
|
|
|
|
|
|
class BoolValidator(TypeValidator):
|
|
def __init__(self):
|
|
super().__init__(bool)
|
|
|
|
|
|
class OrValidator(Validator):
|
|
def __init__(self, *validators):
|
|
self.validators = validators
|
|
super().__init__()
|
|
|
|
def isValid(self, value):
|
|
return reduce(
|
|
or_,
|
|
[v.isValid(value) for v in self.validators],
|
|
False
|
|
)
|
|
|
|
|
|
class NumberValidator(OrValidator):
|
|
def __init__(self):
|
|
super().__init__(IntegerValidator(), FloatValidator())
|
|
|
|
|
|
class RegexValidator(Validator):
|
|
def __init__(self, regex: re.Pattern):
|
|
self.regex = regex
|
|
|
|
def isValid(self, value):
|
|
return isinstance(value, str) and self.regex.match(value) is not None
|
|
|
|
|
|
validator_types = {
|
|
"string": StringValidator,
|
|
"str": StringValidator,
|
|
"integer": IntegerValidator,
|
|
"int": IntegerValidator,
|
|
"number": NumberValidator,
|
|
"num": NumberValidator,
|
|
}
|