12345678910111213141516171819 |
- import contextlib
- from typing import Tuple
- @contextlib.contextmanager
- def rewrite_exception(old_name: str, new_name: str):
- """
- Rewrite the message of an exception.
- """
- try:
- yield
- except Exception as err:
- msg = err.args[0]
- msg = msg.replace(old_name, new_name)
- args: Tuple[str, ...] = (msg,)
- if len(err.args) > 1:
- args = args + err.args[1:]
- err.args = args
- raise
|