1. 首页
  2. 博客
  3. Python——traceback的优雅处理

Python——traceback的优雅处理

刚接触Python的时候，简单的异常处理已经可以帮助我们解决大多数问题；

但是随着逐渐地深入，我们会发现有很多情况下简单的异常处理已经无法解决问题了，如下代码，单纯的打印异常所能提供的信息会非常有限。

#!/usr/bin/env python3

def func():
        raise Exception("-- func exception --")

def main():
        try:
                func()
        except Exception as e:
                print(e)

if __name__ == "__main__":
        main()

执行后输出如下：

通过示例，我们发现普通的打印异常只有很少量的信息（通常是异常的value值），这种情况下我们很难定位在哪块代码出的问题，以及如何出现这种异常。

那么到底要如何打印更加详细的信息呢？下面我们就来一一介绍。

sys.exc_info和traceback object

Python程序的traceback信息均来源于一个叫做 traceback object 的对象，而这个traceback object通常是通过函数sys.exc_info()来获取的，先来看一个例子：

#!/usr/bin/env python3

import traceback
import sys

def func():
    raise Exception("-- func exception --")

def main():
    try:
            func()
    except Exception as e:
            exc_type, exc_value, exc_obj = sys.exc_info()
            print("exception_type: \t%s,\nexception_value: \t%s,\nexception_object: \t%s,\n" \
                    %(exc_type,exc_value,exc_obj))

if __name__ == "__main__":
    main()

执行后输出如下：

通过以上示例我们可以看出，sys.exc_info()获取了当前处理的exception的相关信息，并返回一个元组，元组的第一个数据是异常的类型(示例是NameError类型)，第二个返回值是异常的value值，第三个就是我们要的traceback object.

有了traceback object我们就可以通过traceback module来打印和格式化traceback的相关信息，下面我们就来看下traceback module的相关函数。

traceback module

Python的traceback module提供一整套接口用于提取，格式化和打印Python程序的stack traces信息，下面我们通过例子来详细了解下这些接口：

print_tb

#!/usr/bin/env python3

import traceback
import sys

def func():
    raise Exception("-- func exception --")

def main():
    try:
        func()  
    except Exception as e:
        exc_type, exc_value, exc_obj = sys.exc_info()
        traceback.print_tb(exc_obj)

if __name__ == "__main__":
    main()

输出:

这里我们可以发现打印的异常信息更加详细了，下面我们了解下print_tb的详细信息：

traceback.print_tb(tb[, limit[, file]])
  • tb: 这个就是traceback object, 是我们通过sys.exc_info获取到的
  • limit: 这个是限制stack trace层级的，如果不设或者为None，就会打印所有层级的stack trace
  • file: 这个是设置打印的输出流的，可以为文件，也可以是stdout之类的file-like object。如果不设或为None，则输出到sys.stderr。

#

print_exception

#!/usr/bin/env python3

import traceback
import sys

def func():
    raise Exception("-- func exception --")

def main():
    try:
        func()  
    except Exception as e:
        exc_type, exc_value, exc_obj = sys.exc_info()
        traceback.print_exception(exc_type,exc_value,exc_obj,limit=2,file=sys.stdout)

if __name__ == "__main__":
    main()

输出：

![](/media/images/blog/5fef9cb8f623dcd126ae837454cc734e.png)

看下定义：

traceback.print_exception(etype, value, tb[, limit[, file]])
  • 跟print_tb相比多了两个参数etype和value，分别是exception type和exception value，加上tb(traceback object)，正好是sys.exc_info()返回的三个值
  • 另外，与print_tb相比，打印信息多了开头的"Traceback (most...)"信息以及最后一行的异常类型和value信息
  • 还有一个不同是当异常为SyntaxError时，会有"^"来指示语法错误的位置

#

print_exc

print_exc是简化版的print_exception, 由于exception type, value和traceback object都可以通过sys.exc_info()获取，因此print_exc()就自动执行exc_info()来帮助获取这三个参数了，也因此这个函数是我们的程序中最常用的，因为它足够简单

#!/usr/bin/env python3

import traceback
import sys

def func():
    raise Exception("-- func exception --")

def main():
    try:
        func()  
    except Exception as e:
        exc_type, exc_value, exc_obj = sys.exc_info()
        traceback.print_exc(limit=2,file=sys.stdout)

if __name__ == "__main__":
    main()

输出（由于limit=1，因此只有一个层级被打印出来）：

![](/media/images/blog/74ed622f8c2f165044d87d7772520e4d.png)

定义如下：

traceback.print_exc([limit[, file]])
  • 只有两个参数，够简单

#

format_exc

#!/usr/bin/env python3

import traceback
import sys
import logging

logger = logging.getLogger("test_traceback")

def func():
    raise Exception("-- func exception --")

def main():
    try:
        func()  
    except Exception as e:
        exc_type, exc_value, exc_obj = sys.exc_info()
        logger.error(traceback.format_exc(limit=1))

if __name__ == "__main__":
    main()

输出：

![](/media/images/blog/9fb66b82e632449383692589143a944c.png)

从这个例子可以看出有时候我们想得到的是一个字符串，比如我们想通过logger将异常记录在log里，这个时候就需要format_exc了，这个也是最常用的一个函数，它跟print_exc用法相同，只是不直接打印而是返回了字符串。

traceback module中还有一些其它的函数，但因为并不常用，就不在展开来讲，感兴趣的同学可以看下参考链接中的文档。

#

获取线程中的异常信息

通常情况下我们无法将多线程中的异常带回主线程，所以也就无法打印线程中的异常，而通过上边学到这些知识，我们可以对线程做如下修改，从而实现捕获线程异常的目的。
以下示例来自weidong的博客文章，稍有修改（见参考链接）

#!/usr/bin/env python3

import threading
import traceback

def func():
    raise Exception("thread exception")


class ExceptionThread(threading.Thread):

    def __init__(self, group=None, target=None, name=None, args=(), kwargs=None, verbose=None):
        """
        Redirect exceptions of thread to an exception handler.
        """
        threading.Thread.__init__(self, group, target, name, args, kwargs, verbose)
        if kwargs is None:
            kwargs = {}
        self._target = target
        self._args = args
        self._kwargs = kwargs
        self._exc = None

    def run(self):
        try:
            if self._target:
                self._target()
        except BaseException as e:
            import sys
            self._exc = sys.exc_info()
        finally:
            #Avoid a refcycle if the thread is running a function with
            #an argument that has a member that points to the thread.
            del self._target, self._args, self._kwargs

    def join(self):
        threading.Thread.join(self)
        if self._exc:
            msg = "Thread '%s' threw an exception: %s" % (self.getName(), self._exc[1])
            new_exc = Exception(msg)
            raise new_exc.__class__, new_exc, self._exc[2]


t = ExceptionThread(target=func, name='my_thread')
t.start()
try:
    t.join()
except:
    traceback.print_exc()

输出如下：

![](/media/images/blog/a8a39720571720dc96f730984084991c.png)

这样我们就得到了线程中的异常信息。

原文链接：https://www.cnblogs.com/oddcat/articles/11362961.html

热门标签