python - 如何从 Python 中用户创建的异常类的开头删除 "__main__."

Question

有没有办法获得一个“更漂亮”的异常而不是一个以 __main__MyExceptionTitle 开头的异常？

例子:

>>> class BadThings(Exception):
...     def __init__(self, msg):
...         self.msg = msg
...         return
... 
>>> class BadThings(Exception):
...     def __init__(self, msg):
...         self.msg = msg
...         return
...     def __str__(self):
...         return self.msg
... 
>>> raise BadThings("This is really, really bad")
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
__main__.BadThings: This is really, really bad

我只想说:

BadThings: This is really, really bad

就像一个类型:

>>> raise Exception("This, too, is really, really bad")
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
Exception: This, too, is really, really bad

我想要__main__。走了!

谢谢，纳尼

Answer 1

class MyException(Exception):
    __module__ = Exception.__module__

这种方式看起来/比 sys.excepthook 效果更好

---

源代码引用

Python2

参见PyErr_Display

有两种方法可以绕过模块名称部分:

class A(Exception):
    __module__ = None

class B(Exception):
    __module__ = 'exceptions'

但是如果你需要使用threading.Thread，第一个会引入一个TypeError。

python 3

参见PyErr_WriteUnraisable

class C(Exception):
    __module__ = 'builtins'