90. Python中装饰器的应用示例

Lesson 90 Python中装饰器的应用示例

1. 示例1：比较for循环和jion方法哪个更快

import random
import time
import functools
import string

li = [random.choice(string.ascii_letters) for i in range(100)]
# print(li)
def timecal(fun):
    """这是一个装饰器"""
    @functools.wraps(fun)
    def wrapper(*args,**kwargs): #函数的功能是计算函数运行时间
        """这是wrapper函数"""
        start_time = time.time() #起始时间
        res = fun(*args,**kwargs)
        end_time = time.time()  #结束时间
        print('运行时间为: %.8f' %(end_time - start_time))
        return res
    return wrapper

@timecal
#定义两个函数分别输出li列表中的元素，定义装饰器用来比较两个函数运行的时间
def con_add():
    """这是con_add函数"""
    s = ''
    for i in li:
        s += (i + ',')
    return s

@timecal
def join_add():
    return ','.join(li)

# a = con_add()
# print(a)
con_add()
join_add()
#print(con_add.__name__)
#print(con_add.__doc__)

可以看到，在打印列表时，jion方法比使用for循环累加要快

2. 示例2：创建add_log装饰器， 被装饰的函数打印日志信息

创建装饰器， 要求如下：
# 1. 创建add_log装饰器， 被装饰的函数打印日志信息；
# 2. 日志格式为: [字符串时间] 函数名: xxx， 运行时间：xxx,
 运行返回值结果:xxx

functools 模块中主要包含了一些函数装饰器和便捷的功能函数
自定义python装饰器时一定要记住使用@functools.wraps(fun)修饰wrapper

import time
import functools

def add_log(fun):
    @functools.wraps(fun)
    def wrapper(*args,**kwargs):
        start_time = time.time()
        res = fun(*args,**kwargs)
        end_time = time.time()
        print('[%s]函数名: %s,运行时间: %.6f,运行返回值结果: %d'
            %(time.ctime(),fun.__name__,end_time-start_time,res))
        return res
    return wrapper

@add_log
def add(x,y):
    time.sleep(1)
    return x+y

add(1,2)

可以看到，能够实现装饰器的功能，按照日志格式要求打印日志信息