[Python] @NotNull装饰器如何写比较好?

Question

实际现象

1. Java中的 @NotNull, @Nullable (by jetbrains): https://www.jetbrains.com/hel...

2. Python中可以用装饰器实现, 但是目前没找到细粒度控制的: https://www.google.com/search...

预期现象

1. 希望提供一种细粒度控制的, 即: 具体到每一个参数的控制

2. 只希望检查空指针即可, 类型限定不做

相关代码

• 手动的, 繁琐的写法

def func(param1, param2, param3):
    # 明确表示, 不接受空指针(这里写法很繁琐, 是否有简化方法?)
    if not ((param1 is not None) and (param2 is not None) and (param3 is not None)):
        raise TypeError('%s or %s or %s is None' % (param1, param2, param3))

    # 正常函数逻辑

• 装饰器的写法(未实现)

@not_null(param1, param2)
def func(param1, param2, param3, *args, **kw):

上下文环境

• Python 3.5

Answer 1

from functools import wraps
def allow_jsonp(fun):
    @wraps(fun)
    def wrapper_fun(*args, **kwargs):
        for a in args:
            if not a:
                return 'args is null'
        return fun(*args, **kwargs)
    return wrapper_fun

@allow_jsonp
def a(a):
    print (111111111)

print (a(0))

Answer 2

from functools import wraps

def not_none(func):
    @wraps(func)
    def wrapper(*args, **kwargs):
        if (None in args) or (None in kwargs.values()):
            raise ValueError('Arguments must be not None')
        else:
            return func(*args, **kwargs)
    return wrapper