python-not-none/not_none.py at master · unitedsoftwork/python-not-none · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
def not_none(nullable_parameters=None):

    def the_actual_test(f, args, filter_array):
        has_none = False
        bad_parameters = []

        if type(filter_array) is str:

            filter_array = [filter_array]

        if not filter_array:

            if any(arg[1] is None for arg in args):
                raise ValueError('function {}: Parameters cannot be None. '.format(f.__name__))

        elif type(filter_array) is list:
            for a in args:
                for ff in filter_array:
                    if a[0] != ff:
                        if a[1] is None:
                            has_none = True
                            bad_parameters.append(a[0])
                            break

        if has_none:
            raise ValueError('function {}: Parameters {} cannot be None. '.format(f.__name__, bad_parameters))

    def real_decorator(f):


        v_names = f.__code__.co_varnames

        def wrapper(*args, **kwargs):
            n_args = []

            for a in range(0, len(args)):
                n_args.append((v_names[a], args[a]))

            the_actual_test(f, n_args, nullable_parameters)
            result = f(*args, **kwargs)

            return result
        return wrapper

    return real_decorator


# tests
@not_none()
def _test_no_filter(a, b):
    pass


@not_none(nullable_parameters=["a"])
def _test_yes_filter(a, b):
    pass


@not_none(nullable_parameters="a")
def _test_string_filter(a,b):
    pass


@not_none(nullable_parameters="a")
def _test_default_filter(a,b=1):
    pass


try:

    _test_no_filter(None,None)

except Exception as e:

    print(e)


try:

    _test_no_filter(None, 1)

except Exception as e:

    print(e)


try:

    _test_yes_filter(None, None)

except Exception as e:

    print(e)


try:

    _test_yes_filter(None, 2)
    print("Function Success")

except Exception as e:

    print(e)

try:

    _test_string_filter(None, 2)
    print("Function Success")

except Exception as e:

    print(e)

try:

    _test_string_filter(None, None)


except Exception as e:

    print(e)


try:

    _test_default_filter(None)


except Exception as e:

    print(e)

try:

    _test_default_filter(None,None)


except Exception as e:

    print(e)