是否可以使用color -traceback.py以彩色打印`traceback.format_exc()`

Question

我正在记录traceback.format_exc()的输出。是否可以在使用print或alternative时使用colored-traceback.py对其进行着色？

示例代码：

import traceback
import colored_traceback
colored_traceback.add_hook(always=True)

a = 1 / 0 # prints colored traceback results 

try:
    a = 1 / 0
except:
    traceback.print_exc()  # prints in color white
    print(traceback.format_exc())  # prints in color white

Answer 1

colored_traceback看起来很有用，但我认为它对你的目标来说太过分了。您可以使用pygments库和几行代码来实现所需的效果：

import traceback

from pygments import formatters, highlight, lexers


try:
    a = 1 / 0
except:
    tb_text = "".join(traceback.format_exc())

    lexer = lexers.get_lexer_by_name("pytb", stripall=True)
    formatter = formatters.get_formatter_by_name("terminal256")
    tb_colored = highlight(tb_text, lexer, formatter)

    print(tb_colored)