两个很实用的Python装饰器详解两个很实用的Python装饰器详解

1.超时函数
2.Trace函数
总结

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 resultreturn functools.wraps(func)(wrapper)return decorated

使用：

@timeout(5) #限定下面的slowfunc函数如果在5s内不返回就强制抛TimeoutError Exception结束 def slowfunc(sleep_time):import timetime.sleep(sleep_time) #这个函数就是休眠sleep_time秒 slowfunc(3) #sleep 3秒，正常返回没有异常slowfunc(10) #被终止

【两个很实用的Python装饰器详解】## 输出
---------------------------------------------------------------------------
TimeoutError Traceback (most recent call last)

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

'''学习中遇到问题没人解答？小编创建了一个Python学习交流群：725638078寻找有志同道合的小伙伴，互帮互助,群里还有不错的视频学习教程和PDF电子书！'''import sys,os,linecachedef trace(f):def globaltrace(frame, why, arg):if why == "call": return localtracereturn Nonedef localtrace(frame, why, arg):if why == "line":# record the file name and line number of every tracefilename = frame.f_code.co_filenamelineno = frame.f_linenobname = os.path.basename(filename)print("{}({}): {}".format(bname,lineno,linecache.getline(filename, lineno).strip('\r\n')),)return localtracedef _f(*args, **kwds):sys.settrace(globaltrace)result = f(*args, **kwds)sys.settrace(None)return resultreturn _f

使用：