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

[wentian]

问题详情: https://segmentfault.com/q/1010000009853967

[XYxe]

一个比较粗略的实现：
def not_null(*varnames):
def outer(func):
def inner(*args, **kwargs):
args_name = func.__code__.co_varnames
args_cnt = func.__code__.co_argcount # 没用到
flag = True
for name in varnames: # 这里还要考虑 varnames 中有 args 中没有的参数
index = args_name.index(name) # 还可以加上考虑 kwargs 参数
flag = flag and (args[index] is not None)
if not flag:
raise TypeError('something is None')

result = func(*args, **kwargs)
return result
return inner
return outer

[XYxe]

格式乱了，贴个图吧



另外在用的时候要：@not_null('param1', 'param2')

[araraloren]

@XYxe 返回值也要判断一下，严格的定义肯定实现不了～～

[XYxe]

@araraloren #3 哎？判断返回值那检查一下倒数第四行的 result 不就可以了吗？

[araraloren]

@XYxe 是的，不要在意后面那句话，我看错了