1def identity_function(function):
 2    return function
 3
 4def type_check(arg_type):
 5    if type(arg_type) is not str or (arg_type != "in" and arg_type != "out"):
 6        print("Error!!!")
 7        return identity_function
 8
 9    def type_decorator(*expected_types):
10
11        def type_wrapper(function):
12
13            def type_validator(*args, **kwargs):
14
15                def print_error():
16                    expected_types_str = ', '.join(map(str, expected_types))
17                    if arg_type == "in":
18                        print(f"Invalid input arguments, expected {expected_types_str}!")
19                    elif arg_type == "out":
20                        print(f"Invalid output value, expected {expected_types_str}!")
21
22                if arg_type == "in":
23                    invalid_args = [
24                        current_arg for current_arg in args
25                        if type(current_arg) not in expected_types
26                    ]
27                    invalid_kwargs = [
28                        current_value for current_value in kwargs.values()
29                        if type(current_value) not in expected_types
30                    ]
31                    if invalid_args or invalid_kwargs:
32                        print_error()
33
34                result = function(*args, **kwargs)
35
36                if arg_type == "out" and type(result) not in expected_types:
37                    print_error()
38
39                return result
40
41            return type_validator
42
43        return type_wrapper
44
45    return type_decorator