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

Python之装饰器-带参装饰器

news 2025/8/2 17:34:07

Python之装饰器-带参装饰器

带参装饰器

@之后不是一个单独的标识符，是一个函数调用
函数调用的返回值又是一个函数，此函数是一个无参装饰器
带参装饰器，可以有任意个参数
- @func()
- @func(1)
- @func(1, 2)

def add(x, y):"""函数说明：参数说明返回值说明"""pass# 先写一段伪代码

add.__name__, add.__doc__# add的名字是什么，add的文档是什么
# 返回结果：('add', '\n    函数说明：\n    \n    参数说明\n    返回值说明\n    ')

help(add)	# 查看帮助# add的名字是什么，add的文档是什么
# 返回结果：Help on function add in module __main__:# 返回结果：add(x, y)
# 返回结果：    函数说明：# 返回结果：    参数说明
# 返回结果：    返回值说明

import datetime # 导入datetime模块start = datetime.datetime.now() # 开始时间
end = datetime.datetime.now()	# 结束时间

def logger(wrapped):def wrapper(*args, **kwargs):"wrapper +++"start = datetime.datetime.now()ret = wrapped(*args, **kwargs)delta = (datetime.datetime.now() - start).total_seconds()print("{} tooks {}s.".format(wrapped.__name__, delta))return retwrapper.__name__ = wrapped.__name__ # 通过这行，使装饰器装的更像wrapper.__doc__ = wrapped.__doc__return wrapper@logger # 等价式， add = logger(add) # logger应该等效为单参函数
def add(x, y): # add = wrapper"add description~~"#time.sleep(2)return x + yprint(add(4, 5)) # 非侵入代码，完成了功能，并且好像从来没有装饰过一样
print(add.__name__, add.__doc__)# 此为无参装饰器，已次代码为例演示带参装饰器。
# 返回结果：add tooks 4e-06s.
# 返回结果：9
# 返回结果：add add description~~

# def update(wrapper, wrapped):
#     wrapper.__name__ = wrapped.__name__ # 通过这行，使装饰器装的更像
#     wrapper.__doc__ = wrapped.__doc__
# 可以通过函数来调用，因为重复使用，不需要每次都创建，写到函数外方便调用# def update(src, dest): # 见名知意，这样写，src=源，dest=目标
#     dest.__name__ = src.__name__
#     dest.__doc__ = src.__doc__# from functools import update_wrapper # python内置函数调用from functools import update_wrapper, wraps # 装饰器版本def logger(wrapped):@wraps(wrapped) # 装饰器版本def wrapper(*args, **kwargs):"wrapper +++"start = datetime.datetime.now()ret = wrapped(*args, **kwargs)delta = (datetime.datetime.now() - start).total_seconds()print("{} tooks {}s.".format(wrapped.__name__, delta))return ret#wrapper.__name__ = wrapped.__name__ # 通过这行，使装饰器装的更像#wrapper.__doc__ = wrapped.__doc__#update(wrapper, wrapped)#update(wrapped, wrapper)#update_wrapper(wrapper, wrapped) # 调用return wrapper@logger # 等价式， add = logger(add) # logger应该等效为单参函数
def add(x, y): # add = wrapper"add description~~"#time.sleep(2)return x + yprint(add(4, 5)) # 非侵入代码，完成了功能，并且好像从来没有装饰过一样
print(add.__name__, add.__doc__)# 带参装饰器的对比解释版本
# 返回结果：add tooks 6e-06s.
# 返回结果：9
# 返回结果：add add description~~

from functools import update_wrapper, wrapsdef logger(wrapped):@wraps(wrapped) # 等价式 wrapper = wraps(wrapped)(wrapper) # partial function(偏函数)def wrapper(*args, **kwargs):"wrapper +++"start = datetime.datetime.now()ret = wrapped(*args, **kwargs)delta = (datetime.datetime.now() - start).total_seconds()print("{} tooks {}s.".format(wrapped.__name__, delta))return retreturn wrapper@logger # 等价式， add = logger(add) # logger应该等效为单参函数
def add(x, y): # add = wrapperreturn x + y@logger
def sub(x, y):return x - y#print(add(5, 4))
#print(sub(5, 4))
print(add.__name__, sub.__name__)# 函数，函数执行过程，函数作用域，形参，实参，解构，嵌套函数，LEGB、高阶，柯里化，闭包
# 返回结果：add sub