python装饰器计算函数运行时间

import time
from functools import wraps
import random

def fn_timer(function):
@wraps(function)
def function_timer(*args, **kwargs):
t0 = time.time()
result = function(*args, **kwargs)
t1 = time.time()
print ("Total time running %s: %s seconds" %
(function.func_name, str(t1-t0))
)
return result
return function_timer

@fn_timer
def random_sort(n):
return sorted([random.random() for i in range(n)])

if __name__ == "__main__":
random_sort(2000000)