两个很实用的Python装饰器详解
程序员文章站
2022-03-15 19:24:21
目录1.超时函数2.trace函数总结1.超时函数这个函数的作用在于可以给任意可能会hang住的函数添加超时功能,这个功能在编写外部api调用 、网络爬虫、数据库查询的时候特别有用timeout装饰器...
1.超时函数
这个函数的作用在于可以给任意可能会hang住的函数添加超时功能,这个功能在编写外部api调用 、网络爬虫、数据库查询的时候特别有用
timeout装饰器的代码如下:
import signal,functools #下面会用到的两个库 class timeouterror(exception): pass #定义一个exception,后面超时抛出 def timeout(seconds, error_message = 'function call timed out'): def decorated(func): def _handle_timeout(signum, frame): raise timeouterror(error_message) def wrapper(*args, **kwargs): signal.signal(signal.sigalrm, _handle_timeout) signal.alarm(seconds) try: result = func(*args, **kwargs) finally: signal.alarm(0) return result return functools.wraps(func)(wrapper) return decorated
使用:
@timeout(5) #限定下面的slowfunc函数如果在5s内不返回就强制抛timeouterror exception结束 def slowfunc(sleep_time): import time time.sleep(sleep_time) #这个函数就是休眠sleep_time秒 slowfunc(3) #sleep 3秒,正常返回 没有异常 slowfunc(10) #被终止
## 输出
---------------------------------------------------------------------------
timeouterror traceback (most recent call last)
2.trace函数
有时候出于演示目的或者调试目的,我们需要程序运行的时候打印出每一步的运行顺序 和调用逻辑。类似写bash的时候的bash -x调试功能,然后python解释器并没有 内置这个时分有用的功能,那么我们就“自己动手,丰衣足食”。
trace装饰器的代码如下:
''' 学习中遇到问题没人解答?小编创建了一个python学习交流群:725638078 寻找有志同道合的小伙伴,互帮互助,群里还有不错的视频学习教程和pdf电子书! ''' import sys,os,linecache def trace(f): def globaltrace(frame, why, arg): if why == "call": return localtrace return none def localtrace(frame, why, arg): if why == "line": # record the file name and line number of every trace filename = frame.f_code.co_filename lineno = frame.f_lineno bname = os.path.basename(filename) print("{}({}): {}".format( bname, lineno, linecache.getline(filename, lineno).strip('\r\n')),) return localtrace def _f(*args, **kwds): sys.settrace(globaltrace) result = f(*args, **kwds) sys.settrace(none) return result return _f
使用:
@trace def xxx(): print (1) print (22) print (333) xxx() #调用
## 输出
<ipython-input-4-da50741ac84e>(3): print 1 # @trace 的输出
1
<ipython-input-4-da50741ac84e>(4): print 22 # @trace 的输出
22
<ipython-input-4-da50741ac84e>(5): print 333 # @trace 的输出
333
总结
本篇文章就到这里了,希望能够给你带来帮助,也希望您能够多多关注的更多内容!