当前位置：首页 > news >正文

学习之什么是装饰器

news 来源：原创 2024/9/28 21:18:18

什么是装饰器

本质：就是一个函数，能为其它的函数增加额外功能
在这里插入图片描述

# 目标函数
def power_num(n):"""计算并返回1-n之间每个数的平方和:param n::return:"""total = 0for i in range(1, n+1):total += i ** 2

"""
额外的功能能统计目标函数执行的耗时
"""
def power_num(n):"""计算并返回1-n之间每个数的平方和:param n::return:"""start_time = time()total = 0for i in range(1, n+1):total += i ** 2end_time = time()print("函数执行的耗时：{}".format(end_time - start_time))return totalif __name__ == '__main__':print(power_num(30000000))

# 定义装饰函数--能统计目标函数执行的耗时
def elapsed_time(func):"""能统计目标函数执行的耗时"""def decorated(*args, **kwargs):start_time = time()# 由于目标函数有可能传入一个参数，两个参数或者是关键字参数，所以这里用通用参数占位result = func(*args, **kwargs)end_time = time()print("函数执行的耗时：{}".format(end_time - start_time))return resultreturn decorated# 目标函数
def power_num(n):"""计算并返回1-n之间每个数的平方和:param n::return:"""total = 0for i in range(1, n+1):total += i ** 2return totalif __name__ == '__main__':# 调用装饰函数去装饰目标函数，返回被装饰后的函数  相当于在目标函数上添加@elapsed_time# 对目标函数进行elapsed装饰器函数的调用，返回一个装饰后的名字还是叫power_num的函数power_num = elapsed_time(power_num)print(power_num(30000000))

# 装饰函数需要传入参数
def elapsed_pre(pre):"""装饰函数需要传入参数"""def elapsed_time(func):"""能统计目标函数执行的耗时"""def decorated(*args, **kwargs):start_time = time()# 由于目标函数有可能传入一个参数，两个参数或者是关键字参数，所以这里用通用参数占位result = func(*args, **kwargs)end_time = time()print("函数执行的耗时：{}".format(round((end_time - start_time), pre)))return resultreturn decoratedreturn elapsed_time# 目标函数
@elapsed_pre(3)
def power_num(n):"""计算并返回1-n之间每个数的平方和:param n::return:"""total = 0for i in range(1, n+1):total += i ** 2return totalif __name__ == '__main__':print(power_num(30000000))