py之装饰器（实现函数计时器）

“”"
装饰器实现一个函数计时器
1.被装饰的函数如果有 返回值的时候 怎么办？
2.如何保留被装饰函数的函数名字和帮助文档信息？
“”"

import functools
import random
import string
import time
li = [random.choice(string.ascii_letters) for i in range(100)]

def timeit(fun):
    #@functools.wraps(fun)
    def wrapper(*args,**kwargs):# 接收可变参数和关键字参数
        """这是一个装饰器timeit"""
        # 在函数执行之前
        start_time = time.time()
        # 执行函数
        res = fun(*args,**kwargs)
        # 在函数执行之后
        stop_time = time.time()
        print('运行事件为:%.6f' %(stop_time-start_time))
        return res
    return wrapper
# @timeit
# def con_add():
#     s = ''
#     for i in li:
#         s += (i + '')
#     print(s)
# @timeit
# def join_add():
#     print(','.join(li))
#
# con_add()
# join_add()


@timeit
def fun_list(n):
    """这是fun_list函数"""
    return [2 * i for i in range(n)]
@timeit
def fun_map(n):
    """这是fun_map函数 被装饰器timeis装饰"""
    return list(map(lambda x:x*2,range(n)))
# fun_list(10000)
# fun_map(10000)

print(fun_list.__name__)
print(fun_list.__doc__)

print(fun_map.__name__)
print(fun_map.__doc__)