首页 > > 网络编程 > 其它 >

day 11 - 1 装饰器

2018-12-06 07:35:12来源：博客园阅读 ()

装饰器

装饰器形成的过程：最简单的装饰器——有返回值的——有一个参数——万能参数
装饰器的作用：不想修改函数的调用方式但是还想在原来的函数前后添加功能
原则：开放封闭原则
语法糖：@装饰器函数名
装饰器的固定模式

原则：开放封闭原则
开放：对扩展是开放的
封闭：对修改是封闭的

首先我们来逐步实现这么一个功能吧

计算代码运行的时间

#先来看下需要的组件
import time  #time 模块

time.time()     　　#获取当前时间
time.sleep(5)    　 #让程序睡眠多少时间

print(time.time()) #来试下

我们来写一个可以计算时间的函数

def func():
    start=time.time()
    time.sleep(0.1)   #时间太短 系统会显示 0.0，所以我们让他休眠 0.1s
    print("计算程序执行的时间")
    end=time.time()
    print(end-start)

func()

但是这里有一个问题：如果要计算很多函数的运行时间，那不是要在函数中都加上几行计算的代码，这显然是不可行的，一般写好的没问题的函数，是不会去对它进行修改，而且还很麻烦

所以这里我们写一个计算时间的函数独立出来

def times(f):
    start=time.time()
    time.sleep(0.1)
    f()
    end=time.time()
    print(end-start)


def func():
    print("计算程序执行的时间")


def func2():
    print(time.sleep(5))

times(func)
times(func2)

但是这样貌似还有一个问题，我要计算时间的函数都要调用 times() 函数，这样其实是很麻烦的

最终还是要调用 func() 就可以计算运行时间，就是说你调用的是 func() 其实调用的 times() 这就很完美了

接着我们来了解下装饰器吧，解决这样的问题

先来写一个简单的装饰器

#一个简单的装饰器
import time
def func():
    time.sleep(0.1)
    print('你好，世界')

def timmer(f):　　#装饰器函数
    def inner():
        start = time.time()
        f()　　　 #被装饰的函数
        end = time.time()
        print(end - start)
    return inner

func = timmer(func)    #通过这一步就可以使用 func 调用 timmer 了
func()

我们这里做的事情就是：在不想修改函数的调用方式的情况下但还想在原来的函数前后添加功能

接着我们来看另一个甜甜的东西——语法糖

import time

def timmer(f):　　#装饰器函数
    def inner():
        start = time.time()
　　　　 time.sleep(0.1)　
        f()　　　     #被装饰的函数
        end = time.time()
        print(end - start)
    return inner

@timmer              #语法糖  替代下面 func = timmer(func) 这句话
def func():    
    time.sleep(0.1)
    print('你好，世界')

#func = timmer(func)    #通过这一步就可以使用 func 调用 timmer 了
func()

然后我们来看接收返回值的装饰器

import time

def timmer(f):      #装饰器函数
    def inner():
        start=time.time()
        time.sleep(0.1)
        ret = f()         #被装饰的函数
        end=time.time()
        print(end-start)
        return ret  #接收 func() 函数的返回值    
    return inner    #这个地方要穿函数的名字 而不是加括号

@timmer             #语法糖 替代下面 func = timmer(func) 这句话
def func():
    print('你好，世界')
    return 'Hello'

print(func())       #返回值打印  返回值为：Hello

最后我们来看带参数的装饰器，一个参数与万能参数是一样的，只是参数的不同罢了

#一个参数的装饰器
import time

def timmer(f):      #装饰器函数
    def inner(a):   #3.给 inner() 加上参数
        start=time.time()
        time.sleep(0.1)
        ret = f(a)  #4.由于 inner() 把参数传给了 f(),所以 f() 也要加上参数，最后测试传值
        end=time.time()
        print(end-start)
        return ret  #接收 func() 函数的返回值
    return inner

@timmer             #语法糖 替代下面 func = timmer(func) 这句话
def func(a):        #2.这里也要加上参数,由于执行 func 时，实际执行的 inner(),所以 inner() 也要加上参数
    print("计算程序执行的时间",a)
    return 'Hello'

print(func("123"))  #1.加入参数

#输出结果
'''
计算程序执行的时间 123
0.10937380790710449
Hello
'''

###万能参数的装饰器
#可以看出 只是在 a 的位置更换了 *args **kwargs 来接收参数
import time

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

@timmer             
def func(*args,**kwargs):        
    print("计算程序执行的时间",args,kwargs)    #2. 就是这个位置的 kwargs，如果写了，打印时会出现一个 ' {} ' 空字典，并不会报错
    return 'Hello'

print(func('1','2','3',k=100))    #1. 不需要关键字传参时，kwargs 可以不写

#输出结果
'''
计算程序执行的时间 ('1', '2', '3') {'k': 100}
0.1093745231628418
Hello
'''

好了，以上就是我们对于装饰器完善的最终结果了

装饰器的固定模式

def wrapper(f):     #装饰器函数，f是被装饰的函数
    def inner(*args,**kwargs):    #定义一个内置函数，且和下面统级别的返回值名字一致且不加括号
                                  #动态参数一定要原封不动的传给 被装饰的函数 f()
        '''在被装饰函数之前要做的事'''
        ret = f(*args,**kwargs)   #被装饰的函数  函数执行完的返回值要原封不动传给 return
        '''在被装饰函数之后要做的事'''
        return ret
    return inner

@wrapper         #语法糖 @装饰器函数名 等价于 func = wrapper(func)
def func(a,b):     #被装饰的函数
    time.sleep(0.01)
    print('你好：世界',a,b)
    return '新年好'
#print(func(23,45))  #接收返回值
ret = func(23,45)   #接收返回值
print(ret)