python - 计时装饰器引发 "' NoneType' 对象不可调用“异常

Question

我有我的计时功能和我的主要功能。当我只使用 main 函数时它运行良好，但是当我使用计时函数作为装饰器时它会引发异常。

定时功能代码:

def timing(function):
    import time
    t = time.time()
    function()
    t = time.time() - t
    print('Program has been running for {} seconds.'.format(t))

我是这样使用的:

@timing
def main():
    #some code

Answer 1

装饰器需要返回装饰后的函数:

def timing(function):
  def wrapped():
    import time
    t = time.time()
    function()
    t = time.time() - t
    print('Program has been running for {} seconds.'.format(t))
  return wrapped

@timing
def main():
  # some code