No6.装饰器

大家好，欢迎来到IT知识分享网。

1、装饰器的作用

    不想修改函数的调用方式，但是还想在原来的函数前后添加功能

2、原则

    开放封闭原则

3、装饰器的固定模式

def wrapper(f): #装饰器函数，f是被装饰的函数
    def inner(*args, **kwargs):
        #在被装饰函数之前要做的事
        ret = f(*args, **kwargs)  #被装饰的函数，ret是返回值
        #在被装饰函数之后要做的事
        return ret
    return inner

4、装饰器举例

最基本的装饰器
import time

def timmer(f):
    def inner():
        start = time.time()
        f()
        end = time.time()
        print(end - start)
    return inner

@timmer
def func():
    time.sleep(0.01)
    print("in the func")

func()
>>>in the func
0.010112762451171875

#带参数的装饰器
#装饰器可能装饰不止一个函数，每个函数的参数可能不止一个，所以使用万能参数合适一点
import time

def timmer(f):
    def inner(*args, **kwargs):
        start = time.time()
        f(*args, **kwargs)
        end = time.time()
        print(end - start)
    return inner

@timmer
def func(s):
    print(s)

func("fdfd")

#带返回值的装饰器
import time
def timmer(f):
    def inner(*args, **kwargs):
        start = time.time()
        ret = f(*args, **kwargs)  #用于接收原函数的返回值
        end = time.time()
        print(end - start)
        return ret
    return inner

@timmer
def func(s):
    print(s)
    return 10

ret = func("fdfdf")
print(ret)
>>>fdfdf
0.0
10