Create decorator to validate types of a method

thitkhotauhp9x — Wed, 16 Aug 2023 08:40:44 +0000

def validate_types(func):
    def wrapper(*args, **kwargs):
        annotations = func.__annotations__
        param_types = [annotations[param_name] for param_name in annotations]
        for arg, param_type in zip(args, param_types):
            if not isinstance(arg, param_type):
                raise ValueError(f"Expected type '{param_type.__name__}', got '{type(arg).__name__}' instead")
        for param_name in kwargs:
            param_value = kwargs[param_name]
            param_type = annotations[param_name]
            if not isinstance(param_value, param_type):
                raise ValueError(f"Expected type '{param_type.__name__}', got '{type(param_value).__name__}' instead")
        return func(*args, **kwargs)
    return wrapper

How to use

@validate_types
def add(a: int, b: int) -> int:
    return a + b

DEV Community: thitkhotauhp9x

Create decorator to validate types of a method