python @func 装饰器函数 decorator

代码

作用：顾名思义就是对被修饰函数进行“修饰”的函数
关键：定义一个函数，再定义一个包装函数（def wrapper），其中包括装饰内容，和被修饰函数的形参。这样就实现了对被修饰函数的包装目的，还是很形象的。

import time

#计算耗时的装饰器函数
def display_time(func):	 #func指代被装饰的函数
    def wrapper(*args):  #wapper函数中存放装饰内容，即求func得求解时间
        t1 = time.time()
        result = func(*args) #指代def count_print_nums(maxnum):
        t2 = time.time()
        print('total time:{:.4} s'.format(t2-t1))
        return result
    return wrapper

#质数计算方法
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 count_print_nums(maxnum)耗时
@display_time      #@修饰符指定修饰函数，并紧跟被修饰函数
def count_print_nums(maxnum):
    count = 0
    for i in range(2, maxnum):
        if is_prime(i):
            count +=1
    return count
    
count = count_print_nums(10001)
print(count)

运行到@display_time，进入def display_time(func)，其中func即被修饰函数def count_print_nums(maxnum)。通过wrapper函数编写修饰内容，并最终返回。
被修饰函数def count_print_nums(maxnum)，嵌套进修饰函数中def display_time(func)。

总结，装饰器函数就是为了扩展被修饰函数得功能（比如以上代码得计算耗时），增加整个程序结构得简洁和可读性（修饰函数与被修饰函数之间功能模块得分割）

参考讲解视频

装饰器的统一模板

from functools import wraps
# 对函数的装饰器， 对类func最好为cls
def decorate(func):
    @wraps(func)
    # 增添或修改功能的函数
    def wrapper(*args,**kwargs):
				# 执行被装饰的函数
        result = func(*args,**kwargs) 
        # 返回结果
        return result
    # 返回内层函数
    return wrapper

原因可参考：
https://blog.csdn.net/qq_42874994/article/details/89192150