Python 在使用装饰器时在关闭时返回 None

Question

问题在于，如果第二个数字不为零，则内部函数返回 None 而不是产生输出！

为什么它不能充当普通的装饰器？

代码：

def check(fun):
    def inner_fun(a,b):
        if b == 0:
            print("0 is not valid")
            return
        fun(a,b)
    return inner_fun  #returns None

def divide(a,b):
    return a / b

divide = check(divide)

print(divide(5, 0))

输出： 0 is not valid None

Answer 1

内部函数没有返回您期望的值，因为...它没有return。相反，它隐式返回 None，就像 Python 函数在到达控制流结束时所做的那样。

您可以编辑内部函数以在正常情况下返回f(a, b)：

def check(fun):
   def inner_fun(a,b):
        if b == 0:
            print("0 is not valid")
            return       # implicitly returns None
        return fun(a,b)  # note return
    return inner_fun