如何使用 except in Python 打印 function 中的错误行号？

Question

我想以一种很好的显示方式打印错误的行号和错误消息。 以下是我的代码，它使用linecache ：

import linecache

def func():
    if xx == 1:
        print('ok')

try:
    func()
except:
    exc_type, exc_obj, tb = sys.exc_info()
    f = tb.tb_frame
    lineno = tb.tb_lineno
    filename = f.f_code.co_filename
    linecache.checkcache(filename)
    line = linecache.getline(filename, lineno, f.f_globals)
    print_('ERROR - (LINE {} "{}"): {}'.format(lineno, line.strip(), exc_obj))

但是，这仅给出了func()的位置：

ERROR - (LINE 8 ""): name 'xx' is not defined

有没有办法打印实际发生错误的行号，应该是第4行？ 或者更好的是，我可以打印第 8 行然后追溯到第 4 行吗？ 例如，如果我不使用try - except ，代码：

def func():
    if xx == 1:
        print('ok')

func()

会给我如下错误信息，这样定位错误就好多了：

  File "<input>", line 5, in <module>
  File "<input>", line 2, in func
NameError: name 'xx' is not defined. Did you mean: 'xxx'?

Answer 1

您可以使用traceback和sys modules 来获得高级 traceback output，就像您希望的那样。

这是一个例子：

import traceback
import sys

def func():
    zeroDivide = 1 / 0

try:
    func()

except Exception:
    print(traceback.format_exc()) # This line is for getting traceback.
    print(sys.exc_info()[2]) # This line is getting for the error type.

Output 将是：

Traceback (most recent call last):
  File "b:\abc\1234\pppp\main.py", line 10, in <module>
    func()
  File "b:\abc\1234\pppp\main.py", line 7, in func     
    zeroDivide = 1 / 0

ZeroDivisionError: division by zero

Answer 2

您可以使用 traceback 模块来获取错误的行号，

import traceback

def function():
    try:
        # code 
    except:
        tb_list = traceback.extract_tb(sys.exc_info()[2])
        line_number = tb_list[-1][1]
        print("An error occurred on line:", line_number)

您可以使用traceback.extract_tb() function。此 function 返回一个回溯对象列表，每个对象都包含有关堆栈跟踪的信息。 此列表的最后一个元素tb_list[-1]包含有关发生异常的行的信息。 要访问行号，您可以使用此元组的第二个元素tb_list[-1][1] 。 然后可以使用print() function 打印此值。

Answer 3

要将行号作为 int 获取，您可以从traceback.extract_tb()获取作为列表的回溯。 查看最后一项会为您提供引发异常的行：

#soPrintLineOfError2
import sys
import traceback

def func():
    if xx == 1:
        print('ok')

try:
    func()
except Exception as e:
    tb = sys.exc_info()[2]
    ss = traceback.extract_tb(tb)
    ss1 = ss[-1]
    print(ss1.line)
    print(ss1.lineno)

Output：

if xx == 1:
6