装饰器

Posted 2021-01-24 leo-tail-x

一个简单的装饰器

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‘)

 

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

 

 1 def wrapper1(func):
 2     def inner(*args, **kwargs):
 3         print(‘wrapper1 前‘)  # 2
 4         ret = func(*args, **kwargs)
 5         print(‘wrapper1 后‘)  # 4
 6         return ret
 7     return inner
 8 
 9 def wrapper2(func):
10     def inner(*args, **kwargs):
11         print(‘wrapper2 前‘)  # 1
12         ret = func(*args, **kwargs)
13         print(‘wrapper2 后‘)  # 5
14         return ret
15     return inner
16 
17 @wrapper2  # func1 = wrapper2(func1)  wrapper2.inner   func=wrapper1.inner
18 @wrapper1  # func1 = wrapper1(func1)  wrapper1.inner   func=func1
19 def func1():
20     print(‘func1‘)  # 3
21     return ‘func1的返回值‘
22 
23 print(func1())  # 6

 

装饰器修复技术

1 from functools import wraps
2 def timer(func):
3     @wraps(func)
4     def inner():
5         print(time.time())
6         ret = func()  # 原来的函数
7         return ret
8     return inner