一个例子搞定Python装饰器的使用

装饰器原理：将主函数A作为参数传入到装饰器函数B中，且装饰器函数中的参数与主函数保持一致。调用主函数的同时，实际上先调用装饰器函数，然后再在装饰器函数中调用主函数
装饰器作用：不改变函数代码和调用方式的情况下，给函数添加新的功能。
话不多说，直接上代码：

1.求出【2-10000】内的质数的个数，并计算出核心代码的执行时间

import time
def is_prime(num):
    if num<2:
        return False
    elif num == 2:
        return True
    else:
        for i in range(2,num):
            if num % i ==0:
                return False
        return True

def prime_nums():
    t1 = time.time()
    sum = 0
    for i in range(2,10000):
        if is_prime(i):
            sum += 1
    t2 = time.time()
    print(t2-t1)
    return sum
sum = prime_nums()
print(sum)

执行结果如下：

2.将时间计算的代码独立成单独的函数，在主函数中通过装饰器的方式引入

import time
def display_time(func):
    def wrap():
        t1 = time.time()
        sum = func()
        t2 = time.time()
        print(t2 - t1)
        return sum
    return wrap

def is_prime(num):
    if num<2:
        return False
    elif num == 2:
        return True
    else:
        for i in range(2,num):
            if num % i ==0:
                return False
        return True

@display_time
def prime_nums():
    sum = 0
    for i in range(2,10000):
        if is_prime(i):
            sum += 1
    return sum
sum = prime_nums()
print(sum)

输出结果如下：

注意：1.时间计算函数display_time()一定要带一个参数func,且将函数体wrap返回，其中返回的是wrap不带小括号()的2.主函数prime_nums()有返回参数，那么对应wrap()函数一定也是有返回参数的 3.主函数上的装饰器@display_time是不带小括号()的

3.在主函数内加入一个入参，对应装饰器修改代码如下：

import time
def display_time(func):
    def wrap(*args):
        t1 = time.time()
        sum = func(*args)
        t2 = time.time()
        print(t2 - t1)
        return sum
    return wrap

def is_prime(num):
    if num<2:
        return False
    elif num == 2:
        return True
    else:
        for i in range(2,num):
            if num % i ==0:
                return False
        return True

@display_time
def prime_nums(num):
    sum = 0
    for i in range(2,num):
        if is_prime(i):
            sum += 1
    return sum
sum = prime_nums(10000)
print(sum)

运行结果如下：

仔细的同学会发现，在prime_nums中，将10000参数化了，那么在装饰器函数wrap下也是要增加对应的入参，这里添加*arg代表任意多个入参。
这样子如果再有需要计算运算时间的函数，只需要在对应函数上面加上@display_time即可，这就是装饰器的好处，如果get到了请点个赞再走吧！