1.5.3 Python装饰器

1、简单的装饰器

为何要使用装饰器呢？   装饰器的功能：在不修改原函数及其调用方式的状况下对原函数功能进行扩展   装饰器的本质：就是一个闭包函数 那么咱们先来看一个简单的装饰器：实现计算每一个函数的执行时间的功能

import time


def wrapper(func):
    def inner():
        start = time.time()
        func()
        end = time.time()
        print(end - start)

    return inner


def hahaha():
    time.sleep(1)
    print('aaaaa')


hahaha = wrapper(hahaha)
hahaha()
复制代码

上面的功能有点不简介，不完美，下面就引进了语法糖。

import time


def wrapper(func):
    def inner():
        start = time.time()
        func()
        end = time.time()
        print(end - start)

    return inner


@wrapper
def kkk():  ## 至关于kkk=wrapper(kkk)
    print('aaaaa')


kkk()

复制代码

以上的装饰器都是不带参数的函数，如今装饰一个带参数的该怎么办呢 原函数带多个参数的装饰器

import time
def timer(func):
    def inner(*args,**kwargs):
        start = time.time()
        re = func(*args,**kwargs)
        end=time.time()
        print(end- start)
        return re
    return inner

@timer   #**> func1 = timer(func1)
def func1(a,b):
    print('in func1')
    print(a,b)

@timer   #**> func1 = timer(func1)
def func2(a):
    print('in func2 and get a:%s'%(a))
    return 'fun2 over'

func1(1,2)
print(func2('aaaaaa'))

复制代码

带返回值的装饰器

import time
def timer(func):
    def inner(*args,**kwargs):
        start = time.time()
        re = func(*args,**kwargs)
        end=time.time()
        print(end - start)
        return re
    return inner

@timer   #**> func1 = timer(func1)
def jjj(a):
    print('in jjj and get a:%s'%(a))
    return 'fun2 over'

jjj('aaaaaa')
print(jjj('aaaaaa'))

复制代码

2、开放封闭原则

1.对扩展是开放的 2.对修改是封闭的

3、装饰器的固定结构

import time
def wrapper(func):  ## 装饰器
    def inner(*args, **kwargs):
        '''函数执行以前的内容扩展'''
        ret = func(*args, **kwargs)
         '''函数执行以前的内容扩展'''
        return ret
    return inner

@wrapper  ## ****=>aaa=timmer(aaa)
def aaa():
    time.sleep(1)
    print('fdfgdg')
aaa()
复制代码

4、带参数的装饰器

带参数的装饰器：就是给装饰器传参 用处： 就是当加了不少装饰器的时候，如今突然又不想加装饰器了，想把装饰器给去掉了，可是那么多的代码，一个一个的去闲的麻烦，那么，咱们能够利用带参数的装饰器去装饰它，这就他就像一个开关同样，要的时候就调用了，不用的时候就去掉了。给装饰器里面传个参数，那么那个语法糖也要带个括号。在语法糖的括号内传参。在这里，咱们能够用三层嵌套，弄一个标识为去标识。以下面的代码示例

## 带参数的装饰器：（至关于开关）为了给装饰器传参
## F=True#为True时就把装饰器给加上了
F=False#为False时就把装饰器给去掉了
def outer(flag):
    def wrapper(func):
        def inner(*args,**kwargs):
            if flag:
                print('before')
                ret=func(*args,**kwargs)
                print('after')
            else:
                ret = func(*args, **kwargs)
            return ret
        return inner
    return wrapper

@outer(F)#@wrapper
def hahaha():
    print('hahaha')

@outer(F)
def shuangwaiwai():
    print('shuangwaiwai')

hahaha()
shuangwaiwai()

给装饰器加参数
复制代码

5、多个装饰器装饰一个函数

def qqqxing(fun):
    def inner(*args,**kwargs):
        print('in qqxing: before')
        ret = fun(*args,**kwargs)
        print('in qqxing: after')
        return ret
    return inner

def pipixia(fun):
    def inner(*args,**kwargs):
        print('in qqxing: before')
        ret = fun(*args,**kwargs)
        print('in qqxing: after')
        return ret
    return inner
@qqqxing
@pipixia
def dapangxie():
    print('饿了吗')
dapangxie()

''' @qqqxing和@pipixia的执行顺序：先执行qqqxing里面的 print('in qqxing: before')，而后跳到了pipixia里面的 print('in qqxing: before') ret = fun(*args,**kwargs) print('in qqxing: after')，完了又回到了qqqxing里面的 print('in qqxing: after')。因此就以下面的运行结果截图同样 '''

多个装饰器装饰一个函数
复制代码

6、统计多少个函数被装饰了的小应用

l=[]
def wrapper(fun):
    l.append(fun)#统计当前程序中有多少个函数被装饰了
    def inner(*args,**kwargs):
        ## l.append(fun)#统计本次程序执行有多少个带装饰器的函数被调用了
        ret = fun(*args,**kwargs)
        return ret
    return inner

@wrapper
def f1():
    print('in f1')

@wrapper
def f2():
    print('in f2')

@wrapper
def f3():
    print('in f3')
print(l)

复制代码