初解装饰器

装饰器

装饰器就是一个函数，用来装饰另一个函数。为另一个函数添加新功能。

特点：①不改变原来函数的源代码     ②不改变原来函数的调用方式

装饰器=高阶函数+嵌套函数+函数闭包

加入目前我有一个这样的test函数，想给它增加一个新的功能，就是查看函数的运行时间，那我该怎么做呢？

import time
def test():
    for i in range(5):
        time.sleep(0.1)
    print("运行完毕")

test()

第一种方法也是最直接的方法：直接改动源代码

import time
def test():
    start_time = time.time()
    for i in range(5):
        time.sleep(0.1)
    print("运行完毕")
    stop_time = time.time()
    print("该函数运行的时间是%s"%(stop_time - start_time))

test()

结果是：

运行完毕
该函数运行的时间是0.5019335746765137

这样改动的话看起来十分简单的操作

然而如果你的手边有10000或者是更多这样的函数，每一个都要这样添加函数功能，那你还会一个一个的去修改这些函数么?

答案是当然不会的啦，这时你就会想到另一个方法，我能不能再另外定义一个函数用来专门统计时间呢？当然可以了

import time

def timer(func):
    start_time = time.time()
    func()
    stop_time = time.time()
    print("该函数运行的时间是%s"%(stop_time - start_time))

def test():
    for i in range(5):
        time.sleep(0.1)
    print("运行完毕")

timer(test)

结果：

运行完毕
该函数运行的时间是0.5027310848236084

这样一看是不是简单多了呢？

但是面你对成千上万的函数，每次都去这样调用是不是还是显得麻烦了呢？

假想一下：你调用函数

timer(test)

timer(test1)

timer(test2)

timer(test3)

.......

这样的话你又突然想到我把test函数的返回值写成return timer，然后定义一个test变量作为test()的返回值，再调用test()不就行了么？这样想一下，貌似是完美的。然而。。。

import time

def timer():
    start_time = time.time()
    test()
    stop_time = time.time()
    print("该函数运行的时间是%s"%(stop_time - start_time))

def test():
    for i in range(5):
        time.sleep(0.1)
    print("运行完毕")
    return timer
test = test()   #可以使用test()()的方式调用
test()

结果：

运行完毕
运行完毕
该函数运行的时间是0.5015039443969727

哇，结果是错的（调用了两次test()函数 ），天啊，而且好像又把它给搞的更加复杂了，那到底应该怎么做呢，真让人头皮发麻！

如果把定义的timer()放到另一个新定义的decorate()中然后decorate函数的返回值设置为timer是不是就可以了呢？来让我们试一下

import time

def decorate(func):
    def timer():
        start_time = time.time()
        func()
        stop_time = time.time()
        print("该函数运行的时间是%s"%(stop_time - start_time))
    return timer
def test():
    for i in range(5):
        time.sleep(0.1)
    print("运行完毕")

test = decorate(test)
test()

结果：

运行完毕
该函数运行的时间是0.5024490356445312

哇，完美。

然而在python中有一种语法糖就是我们常常用到的@符号

@装饰器函数就等价于（变量 = 装饰器函数（被装饰函数名字））

在这里既是@decorate = decorate（test）

怎么样，是不是很美妙呢？

让我们试一试是不是真的如我所说

import time

def decorate(func):
    def timer():
        start_time = time.time()
        func()
        stop_time = time.time()
        print("该函数运行的时间是%s"%(stop_time - start_time))
    return timer

@decorate

def test():
    for i in range(5):
        time.sleep(0.1)
    print("运行完毕")
    
test()

结果：

运行完毕
该函数运行的时间是0.5020065307617188

是了，语法糖@可用无疑

这样一个简单的装饰器就被我们创建出来了

但是新问题又来了，如果我的被装饰函数有返回值或者函数形参怎么办呢？

首先我们先看一下有函数返回值的情况怎么办？

import time

def decorate(func):
    def timer():
        start_time = time.time()
        res = func()     #此处调用被修饰函数，在此处定义一个变量接受被装饰函数的返回值就可以了
        stop_time = time.time()
        print("该函数运行的时间是%s"%(stop_time - start_time))
        return res    #此处返回函数的返回值
    return timer

@decorate

def test():
    for i in range(5):
        time.sleep(0.1)
    return "运行完毕"

print(test())

结果：

该函数运行的时间是0.5053601264953613
运行完毕

看来这样是可以的，我们又解决了一个问题

接下来让我们看看函数有形参的情况下怎么办？

这时候我们就要使用*args, **kwargs这两个可变长参数了，因为我们传入参数数量是不确定的，比如一个函数需要两个参数，另一个需要三个参数，另一个需要4个参数。。。。。

好了，那这样我们该怎么写呢？

import time

def decorate(func):
    def timer(*args, **kwargs):
        start_time = time.time()
        res = func(*args, **kwargs)
        stop_time = time.time()
        print("该函数运行的时间是%s"%(stop_time - start_time))
        return res
    return timer

@decorate

def test(name, age):
    for i in range(5):
        time.sleep(0.1)
    return "我的名字是%s，我的年龄是%s"%(name,age)

print(test("bob", 18))

结果：

该函数运行的时间是0.5026912689208984
我的名字是bob，我的年龄是18

这样一个装饰器的基本条件基本上是齐备了 。

最后一个问题，如果我想在每一个被修饰的函数前面天界一句“hello，world！”我该怎么办呢？

简单，就是在装饰函数外在套一个装饰器函数就好了，在这里我套了函数decorate1。

import time
def decorate1(s):
    def decorate(func):
        def timer(*args, **kwargs):
            print("%s"%s)
            start_time = time.time()
            res = func(*args, **kwargs)
            stop_time = time.time()
            print("该函数运行的时间是%s"%(stop_time - start_time))
            return res
        return timer
    return decorate
@decorate1("hello,world!")

def test(name, age):
    for i in range(5):
        time.sleep(0.1)
    return "我的名字是%s，我的年龄是%s"%(name,age)

print(test("bob", 18))

结果：

hello,world!
该函数运行的时间是0.5027992725372314
我的名字是bob，我的年龄是18

好了，关于装饰器，目前只能想到这么多了！！！