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

目录

• [1.超时函数]
• [2.Trace函数]
• [总结]

1.超时函数

这个函数的作用在于可以给任意可能会hang住的函数添加超时功能，这个功能在编写外部API调用 、网络爬虫、数据库查询的时候特别有用。最后，如果你的时间不是很紧张，并且又想快速的python提高，最重要的是不怕吃苦，建议你可以架群：【832357663】 ，那个真的很不错，很多人进步都很快，需要你不怕吃苦哦！大家可以去添加上看一下~

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学习交流群：832357663

寻找有志同道合的小伙伴，互帮互助,群里还有不错的视频学习教程和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() #调用

|

输出

(3): print 1 # @trace 的输出
1
(4): print 22 # @trace 的输出
22
(5): print 333 # @trace 的输出
333

总结

本篇文章就到这里了，希望能够给你带来帮助，也希望您能够多多关注【python教程入门学习】的更多内容!