两个实用的Python的装饰器-白红宇

两个实用的Python的装饰器

阅读量：4311 次

发布时间：2019-06-06

本文共 1907 字，大约阅读时间需要 6 分钟。

超时函数

这个函数的作用在于可以给任意可能会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)

Trace函数

有时候出于演示目的或者调试目的，我们需要程序运行的时候打印出每一步的运行顺序和调用逻辑。类似写bash的时候的bash -x调试功能，然后Python解释器并没有内置这个时分有用的功能，那么我们就“自己动手，丰衣足食”。

Trace装饰器的代码如下：

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

使用：

@tracedef xxx():  print 1 print 22 print 333 xxx() #调用 ## 输出 
      
       (3): print 1 # @trace 的输出 1 
       
        (4): print 22 # @trace 的输出 22 
        
         (5): print 333 # @trace 的输出 333 作者：auxten 链接：https://zhuanlan.zhihu.com/p/20175869

转载于:https://www.cnblogs.com/WayneZeng/p/9290729.html

你可能感兴趣的文章