装饰器

一个简单的装饰器

def wrapper(func):
    def inner(*args, **kwargs):
        # 执行被装饰函数之前进行的操作
        ret = func(*args, **kwargs)
        # 执行被装饰函数之后进行的操作
        return ret
    return inner


@wrapper
def func():
    print('func')

func()

装饰器带参数

def outer(flag):
    def timer(func):
        def inner(*args, **kwargs):
            if flag:
                print(time.time())
                ret = func(*args, **kwargs)  # 原来的函数
            else:
                ret = func(*args, **kwargs)  # 原来的函数
            return ret
        return inner
    return timer

@outer(True)  # func1 = timer(func1)  inner
def func1():
    print('func1')

多个装饰器装饰同一个函数

def wrapper1(func):
    def inner(*args, **kwargs):
        print('wrapper1 前')  # 2
        ret = func(*args, **kwargs)
        print('wrapper1 后')  # 4
        return ret
    return inner

def wrapper2(func):
    def inner(*args, **kwargs):
        print('wrapper2 前')  # 1
        ret = func(*args, **kwargs)
        print('wrapper2 后')  # 5
        return ret
    return inner

@wrapper2  # func1 = wrapper2(func1)  wrapper2.inner   func=wrapper1.inner
@wrapper1  # func1 = wrapper1(func1)  wrapper1.inner   func=func1
def func1():
    print('func1')  # 3
    return 'func1的返回值'

print(func1())  # 6

装饰器修复技术

from functools import wraps
def timer(func):
    @wraps(func)
    def inner():
        print(time.time())
        ret = func()  # 原来的函数
        return ret
    return inner