Python世界中, 有Java中那种 @NotNull 吗( JetBrains提供的 ) ?

Question

2017/2/6

描述

在写Java的过程中, 这两个东西非常有用( 对付null )

那么, Python世界中, 可否有类似实现 ?

• 检查函数的返回值 ( @NotNull 提供 )

• 检查函数的形参( @NotNull 提供 )

• 提醒对返回值进行check ( @Nullable提供 )

最厉害的使这个

public String someMethod(@NotNull String var1, @NotNull int var2, float var3)

相关代码

• 检查函数的返回值, python装饰器应该不难做到

def return_value_not_none(fn):
    @functools.wraps(fn)
    def wrapper(*arg, **kw):
        return_value = fn(*args, **kw)
        
        if return_value is None:
            raise TypeError('return value can not be None')
        else:
            return return_value
   return wrapper


@return_value_not_none
def test(a):
    if a > 0:
        return a
    else:
        return None

上下文环境

• Jetbrains提供的 @NotNull: https://blog.jetbrains.com/id...

• 对应的blog: https://blog.jetbrains.com/id...

Answer 1

I’m not very familiar with java, but I think Python’s decorator can solve it.

def value_not_none(fn, **kwargs):
    @functools.wraps(fn)
    def wrapper(*arg, **kw):
        for k,v in kw.items():
            if kw[k] is None:
                raise TypeError('%s value can not be None' % )
        return_value = fn(*args, **kw)
        return return_value
   return wrapper

Answer 2

Weak question, can I solve it directly using if return_value is None:?