python装饰器

python 装饰器原理

装饰器的目的为在不改变被修饰函数的源码以及调用方式的情况下,对该函数进行功能扩展,符合"开闭原则",装饰器本质上是一个函数,只不过python提供了较为独特的调用方式:

• @装饰器名 ,如下例所示:@timer装饰了index函数,当python解释器解释到@timer时,首先将其下方的函数名(index)作为参数传入timer修饰器,然后返回一个inner对象并赋值给index,即index=timer(index),当执行到index()语句时候,实际上执行的是inner()语句

# 装饰器原理
import time
from functools import wraps
# 无参装饰器
def timmer(func):
    @wraps(func) #将inner的帮助文档，名字，参数等信息转换成被装饰的func的对应信息
    def inner(*args,**kwargs):
        start=time.time()
        res=func(*args,**kwargs)
        stop=time.time()
        return res # 被装饰函数的返回值
    return inner

# 带参数的装饰器
def auth(tmp=1):
    def timmer(func):
        def inner(*args, **kwargs):
            start = time.time()
            time.sleep(tmp)
            res = func(*args, **kwargs)
            stop = time.time()
            return res  # 被装饰函数的返回值
        return inner

@timmer  # 其作用等同于index=timmer(index)
def index(name):
    time.sleep(3)
    print('welcome %s to index'%name)

index('bob') # 此时的index本质上inner(不再是index，使用了@的语法糖),故此次为inner('bob')