Python基础之装饰器

装饰器

装饰器就是装修、装饰的意思，它不会对原来的函数造成改变，还能添加新的功能，调用函数时的接口不会变化。
比如，我们要在函数基础上，增加一个程序效率检测功能，也就是记录函数执行的时间。

例：
首先，这是一个测试脚本运行时间的函数

import time

def index():
    time.sleep(2)

start_time = time.time()
index()
end_time = time.time()
print(end_time - start_time)

如果我们用装饰器的形式来完成，就会是这样。

import time

# 首先，我们定义一个装饰器函数，切记装饰器一定要先定义于被装饰的对象。
# 装饰器函数的功能是实现对函数运行时间的检测
def calucate_time(f):
    def inner(*args, **kwargs):
        start_time = time.time()
        ret = f(*args, **kwargs) 
        end_time = time.time()
        print(end_time - start_time)
        return ret
    return inner

# 然后，我们把运行时间检测功能用装饰器装饰到加法函数上。
@calucate_time
def add(m, n):
    time.sleep(2.2)
    return m + n
print(add(1,2))