两个很实用的Python装饰器详解（python装饰器--原来如此简单）

​

这篇文章主要为大家介绍了Python的装饰器，具有一定的参考价值，感兴趣的小伙伴们可以参考一下，希望能够给你带来帮助，希望能够给你带来帮助

目录

1.超时函数

2.Trace函数

总结

1.超时函数

这个函数的作用在于可以给任意可能会hang住的函数添加超时功能，这个功能在编写外部API调用 、网络爬虫、数据库查询的时候特别有用

timeout装饰器的代码如下：

1

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

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

使用：

1

2

3

4

5

6

@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装饰器的代码如下：

1

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

17

18

19

20

21

22

23

24

25

'''

学习中遇到问题没人解答？小编创建了一个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

使用：

1

2

3

4

5

6

@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

总结

本篇文章就到这里了，希望能够给你带来帮助，也希望您能够多多关注脚本之家的更多内容!

版权声明：本文内容由网络用户投稿，版权归原作者所有，本站不拥有其著作权，亦不承担相应法律责任。如果您发现本站中有涉嫌抄袭或描述失实的内容，请联系我们jiasou666@gmail.com 处理，核实后本网站将在24小时内删除侵权内容。