如何实现一个python函数装饰器(Decorator)
装饰器本质上是一个 Python 函数或类，它可以让其他函数或类在不需要做任何代码修改的前提下增加额外功能，装饰器的返回值也是一个函数/类对象。它经常用于为已有函数/类添加记录日志、计时统计、性能测试等。
首先定义一个倒计时函数，这个函数的功能非常简单，就是把n从当前值减少到0。
def countdown(n):
while n > 0:
print('time' + str(n))
n -= 1
print(countdown.__name__)
程序输出:
countdown
1.为函数增加一个日志装饰器
假设现在要增强countdown的功能，在函数调用前后自动打印日志，又不想修改函数自身的功能。这种在代码运行期间动态增加功能的方式，称之为装饰器(Decorator)。
能打印日志的decorator，可以定义如下：
def log(func):
def wrapper(*args, **kw):
print('call %s().' % func.__name__)
return func(*args, **kw)
return wrapper
然后我们借助Python的@语法，把decorator置于函数的定义处：
@log
def countdown(n):
while n > 0:
print('time:' + str(n))
n -= 1
countdown(10)
程序输出：
call countdown().
time:10
time:9
time:8
time:7
time:6
time:5
time:4
time:3
time:2
time:1
但此时我们再打印函数的name:
print(countdown.__name__)
程序输出:
wrapper
我们发现函数的元数据信息变了，这显然不是我们想要的结果。
2. 在装饰器中拷贝元数据
为了把函数的元数据信息都保留下来，我们可以直接使用Python提供的functools库中的@wraps装饰器。
from functools import wraps
def log(func):
@wraps(func)
def wrapper(*args, **kw):
print('call %s().' % func.__name__)
return func(*args, **kw)
return wrapper
@log
def countdown(n):
while n > 0:
print('time:' + str(n))
n -= 1
print(countdown.__name__)
程序输出：
countdown
3.为函数增加一个计时装饰器
添加函数装饰器的方法已经讲清楚了，现在再实现一个完整的函数计时耗时装饰器。
import time
from functools import wraps
def TimeCost(func):
@wraps(func)
def wrapper(*arg, **kwargs):
start = time.time()
result = func(*args, **kwargs)
end = time.time()
print(func.__name__, end - start)
return result
return wrapper
@TimeCost
def countdown(n):
while n > 0:
print('time:' + str(n))
n -= 1
countdown(10000)
函数输出：
('countdown', 0.0004801750183105469)
参考资料：
https://www.liaoxuefeng.com/wiki/1016959663602400/1017451662295584
以上就是如何实现一个python函数装饰器(Decorator)的详细内容，更多关于python函数装饰器的资料请关注脚本之家其它相关文章！
相关文章
浅谈python中的@以及@在tensorflow中的作用说明
这篇文章主要介绍了浅谈python中的@以及@在tensorflow中的作用说明，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧2021-03-03
这篇文章主要介绍 Python 编程技巧,我们知道，字典的本质是哈希表，本身是无法排序的，但 Python 3.6 之后，字典是可以按照插入的顺序进行遍历的，这就是有序字典，其中的原理，可以阅读为什么 Python3.6 之后字典是有序的。本文也会介绍该内容,需要的朋友可以参考一下2021-10-10
这篇文章主要介绍了编程语言Python的发展史,本文详细介绍了Python语言的发展历史,需要的朋友可以参考下2014-09-09
Python 装饰器@，对函数进行功能扩展操作示例【开闭原则】
这篇文章主要介绍了Python 装饰器@，对函数进行功能扩展操作,结合实例形式分析了装饰器的相关使用技巧,以及开闭原则下的函数功能扩展,需要的朋友可以参考下2019-10-10
本文主要介绍了Python数据获取实现图片数据提取，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧2022-05-05
这篇文章主要介绍了Python中使用socks5设置全局代理的方法示例，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧2020-04-04
这篇文章主要介绍了在Python web中实现验证码图片代码分享，具有一定参考价值，需要的朋友可以了解下。2017-11-11
这篇文章主要介绍了Ubuntu16安装Python3.9的实现步骤，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧2020-12-12
这篇文章主要介绍了python装饰器原理与用法,结合实例形式深入分析了Python装饰器的概念、原理、使用方法及相关操作注意事项,需要的朋友可以参考下2019-12-12
这篇文章主要介绍了pytorch中的dataloader使用方法详解,构建自己的dataloader是模型训练的第一步,本篇文章介绍下pytorch与dataloader以及与其相关的类的用法,需要的朋友可以参考下2023-11-11
最新评论