如何为Python内置异常编写自定义错误提示信息

你最初的写法出现双异常栈，是因为Python 3默认会自动关联异常上下文：在except块内抛出新异常时，解释器会自动把捕获到的原始异常挂在新异常的__context__属性上，输出时就会打印「During handling of the above exception, another exception occurred」提示和两段栈信息。
你看到的魔改__import__.Exception的写法属于非规范的野路子，直接修改内置对象的属性没有任何兼容性保证，不同Python版本行为不可预期，绝对不要在生产代码里用。而且那个自定义异常的实现把路径、模块名这类变量硬编码在全局域拼__str__，完全没有复用性，设计上就是错的。

方案1：最小改动，截断异常链 #

不需要自定义异常，直接使用Python官方支持的raise ... from None语法显式截断异常链即可，这是PEP 3134明确定义的标准写法，改动量最小：

try:
    module = __import__(module_name)
    return module
except ModuleNotFoundError:
    err_msg = (
        "\nCould not load module script. \n"
        f'Module "{module_name}" was not found in the current directory. \n'
        "Try using --help for help.\n"
    )
    raise ModuleNotFoundError(err_msg) from None

加了from None之后，解释器就不会附加原始异常上下文，输出里不会再出现原始异常栈和冗余提示，只会展示你自定义的错误信息。

方案2：自定义业务异常（适合中大型脚本/工具） #

如果需要把用户友好的业务错误和系统底层错误做明确区分，可以自定义异常子类，但不要修改任何内置对象的属性，正确写法如下：

# 先定义异常类，异常信息通过构造函数传入，不要依赖全局变量
class ModuleLoadError(ModuleNotFoundError):
    """脚本自定义模块加载失败异常，与系统原生模块缺失错误区分"""
    pass

# 业务逻辑处的捕获逻辑
try:
    module = __import__(module_name)
    return module
except ModuleNotFoundError as e:
    # 可选判断：如果是导入的目标模块本身缺失才抛自定义提示，
    # 如果是目标模块内部依赖的其他模块缺失，直接抛原始异常方便排查
    if e.name != module_name:
        raise
    err_msg = (
        "\nCould not load module script. \n"
        f'Module "{module_name}" was not found in the current directory. \n'
        "Try using --help for help.\n"
    )
    raise ModuleLoadError(err_msg) from None

方案3：顶层统一异常处理（适合面向普通用户的命令行工具） #

如果希望面向终端用户完全隐藏异常栈，只输出干净的错误提示，同时不影响开发阶段排查问题，可以在程序最外层入口加统一异常捕获：

import sys

def main():
    # 把你所有的脚本逻辑放到main函数里
    pass

if __name__ == "__main__":
    try:
        main()
    except ModuleLoadError as e:
        # 自定义的业务异常，只打印友好提示到标准错误，退出码设为1标识执行失败
        print(f"Error: {e}", file=sys.stderr)
        sys.exit(1)
    # 其他非预期异常会走Python默认的异常输出逻辑，打印完整栈方便排查bug

• 不要直接在异常捕获块里print错误然后调用sys.exit()：这种写法会把错误处理逻辑散落在业务代码各处，后续要加错误日志、错误上报、统一错误格式的时候维护成本极高。
• 不要修改__import__这类内置对象的属性：属于CPython的未定义行为，版本升级就可能失效，还可能引发其他难以排查的诡异问题。
• 自定义异常不要依赖全局变量拼接错误信息：异常实例本身应该持有所有错误上下文，保证异常可以独立抛出、捕获，不受外层代码变量作用域影响。

内容的提问来源于stack exchange，提问作者Aman Ahmed Siddiqui