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

浮生若梦的编程
  • 2.8k

实际现象

  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

回复
阅读 2.2k
2 个回答
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))
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
撰写回答
你尚未登录,登录后可以
  • 和开发者交流问题的细节
  • 关注并接收问题和回答的更新提醒
  • 参与内容的编辑和改进,让解决方法与时俱进
你知道吗?

宣传栏