修改异常消息而不丢失引发的堆栈

Question

我有一个try/except 子句，它将返回或捕获KeyError，如下所示：

try:
    return super().__new__(globals()[kls])
except KeyError:
    raise

这会在使用不当时生成堆栈跟踪：

>>> g = Grid(cell='Circle')
Traceback (most recent call last):
  File "<pyshell#16>", line 1, in <module>
    g = Grid(cell='Circle')
  File "<pyshell#1>", line 8, in __new__
    return super().__new__(globals()[kls])
KeyError: 'SHPCircleGrid'
>>> g
Traceback (most recent call last):
  File "<pyshell#17>", line 1, in <module>
    g
NameError: name 'g' is not defined

这很好，但是我想“扩展/修改”消息以向用户解释如何再次避免此错误；即从：

KeyError: 'SHPCircleGrid'

到

KeyError: 'SHPCircleGrid'. Use 'Hex', 'Rect' or 'Tri' for cell keyword.

同时为用户维护堆栈。捕获部分中的通用print() 将g 设置为NoneType，这是我不喜欢的，所以简单地打印不是处理这个问题的方法。添加另一个 raise KeyError('some message') 会打印两个堆栈（“同时处理另一个异常时...”消息），这也是不希望的。

处理此问题的适当方法是什么，以便可以将其扩展到类实例化可能抛出KeyError 的任何其他关键字？

Answer 1

您不能通过将您的消息提供给KeyError 来实现这一点吗：

try:
    return super().__new__(globals()[kls])
except KeyError as e:
    key = "'{}'".format(*e.args)
    base = "{}. Use 'Hex', 'Rect' or 'Tri' for cell keyword."
    raise KeyError(base.format(key)) from None