Python 编程：从零基础到精通

第 19 课 - 自定义异常与异常链

学习目标

理解为什么需要自定义异常，以及它能带来哪些好处。
掌握通过继承内置 Exception 类来创建自定义异常类的方法。
学会使用 raise 关键字显式地抛出自定义或内置异常。
理解异常链的概念，并能够使用 raise ... from ... 语法来创建清晰、可追溯的异常链。
能够在适当的位置设计和使用自定义异常，使代码更清晰、更易维护。

核心概念

1. 为什么需要自定义异常？

Python 内置了丰富的异常类型（如 ValueError, TypeError, FileNotFoundError），但有时它们无法精确描述我们程序中发生的特定业务逻辑错误。例如，在一个用户年龄验证函数中，年龄超出合理范围（如小于0或大于150）的错误，用内置的 ValueError 来表示虽然可以，但不够精确。

自定义异常允许我们为特定的错误场景创建有意义的名称。这使得代码：

更清晰：看到 InvalidAgeError 就立刻明白是年龄无效。
更易维护：错误处理可以更精准，我们只需捕获 InvalidAgeError，而不是泛泛的 ValueError。
更健壮：为不同模块或层次定义不同的异常，有助于构建清晰的错误处理架构。

2. 如何创建自定义异常？

创建自定义异常非常简单，只需要定义一个类，并让它继承自 Python 内置的 Exception 类或其子类。通常，我们直接继承 Exception 即可。

# 定义一个名为 `InvalidAgeError` 的自定义异常
class InvalidAgeError(Exception):
    """当提供的年龄无效（如负数或过大）时引发。"""
    pass

这是一个最简单的自定义异常，它本身没有添加任何新的行为或属性，仅仅是一个新的类名。我们可以像使用内置异常一样使用它。

3. 显式抛出异常 (`raise`)

使用 raise 语句可以手动触发一个异常。这是我们"告诉"程序某个错误情况已经发生的方式。

def verify_age(age):
    if age < 0 or age > 150:
        # 使用 raise 抛出我们刚刚定义的自定义异常
        raise InvalidAgeError(f"年龄 {age} 无效。年龄必须在 0 到 150 之间。")
    return f"年龄 {age} 有效。"

4. 异常链 (`raise ... from ...`)

在实际开发中，我们经常在一个函数中捕获一个底层异常（如 IOError），然后抛出一个更符合上层业务逻辑的自定义异常。异常链的作用就是将这两个异常关联起来，这样当用户看到我们的自定义异常时，还能追溯到导致它的原始底层异常。这极大地提高了调试效率。

Python 3 使用 raise NewException(...) from original_exception 语法来显式地建立异常链。

代码示例

示例 1：定义和使用自定义异常

# 定义自定义异常：无效的年龄错误
class InvalidAgeError(Exception):
    """自定义异常：当年龄值无效时引发。"""
    def __init__(self, age, message="年龄无效。"):
        self.age = age
        self.message = message
        super().__init__(self.message) # 调用父类 Exception 的初始化方法
    
    def __str__(self):
        return f'{self.age} -> {self.message}'

def register_user(name, age):
    """
    用户注册函数，会验证年龄。
    """
    if not isinstance(age, int):
        raise TypeError("年龄必须是整数。") # 内置异常
    if age < 0 or age > 150:
        # 抛出自定义异常，并附带详细的错误信息
        raise InvalidAgeError(age, f"年龄 {age} 不在合理范围内 (0-150)。")
    
    print(f"用户 {name} (年龄: {age}) 注册成功！")

# 主程序
try:
    register_user("Alice", 25)  # 成功
    register_user("Bob", -5)    # 会引发 InvalidAgeError
except InvalidAgeError as e:
    print(f"注册失败（自定义异常）: {e}")
    print(f"无效的年龄值是: {e.age}") # 可以访问自定义属性
except TypeError as e:
    print(f"注册失败（类型错误）: {e}")

示例 2：使用异常链

class ConfigurationError(Exception):
    """自定义异常：当配置加载失败时引发。"""
    pass

def load_config_from_file(file_path):
    """
    从文件加载配置，模拟一个可能出错的操作。
    """
    try:
        # 模拟一个底层错误，例如文件权限问题或文件损坏
        with open(file_path, 'r') as f:
            # 假设文件内容是空的或格式错误，我们手动引发一个 KeyError
            data = {'key': 'value'}
            config = data['required_key'] # 这里会引发 KeyError
    except KeyError as e:
        # 捕获底层的 KeyError，然后抛出一个更友好的业务异常
        # 使用 `raise ... from e` 将原始的 KeyError 链接到新的异常上
        raise ConfigurationError(
            f"配置文件 '{file_path}' 缺少必需的键: {e}"
        ) from e
    except FileNotFoundError as e:
        # 同样，处理文件不存在的情况
        raise ConfigurationError(
            f"找不到配置文件: {file_path}"
        ) from e # 显式链接
    # 注意：这里故意没有捕获所有异常，让其他意外错误（如PermissionError）自然传播

# 主程序
try:
    config = load_config_from_file("nonexistent.cfg") # 假设文件不存在
except ConfigurationError as e:
    print(f"配置错误: {e}")
    # 通过 `__cause__` 属性可以查看引发当前异常的原始异常
    print(f"原始错误原因: {e.__cause__}")

实践练习

练习 1：基础自定义异常与 `raise`

要求：

创建一个名为 InsufficientFundsError 的自定义异常，用于表示银行账户余额不足。
编写一个 withdraw 函数，参数为 balance（当前余额）和 amount（取款金额）。如果取款金额大于余额，则抛出 InsufficientFundsError；否则，返回取款后的余额。
使用 try...except 块捕获此异常，并打印一条友好的错误信息。

预期输出：当调用 withdraw(100, 150) 时，程序应捕获异常并输出类似：

错误：余额不足。当前余额：100.0，尝试取款：150。

练习 2：使用异常链

要求：

创建一个自定义异常 DatabaseConnectionError。
编写一个 connect_to_db 函数，它尝试连接到一个不存在的数据库主机（例如 fakehost）。这个操作会引发一个底层的 OSError 或 ConnectionRefusedError。
在 connect_to_db 函数中捕获这个底层异常，并抛出 DatabaseConnectionError，同时使用 from 语法建立异常链。
在主程序中调用该函数，并打印出自定义异常和它链接的原始异常。

预期输出：

无法连接到数据库：[Errno 11001] getaddrinfo failed
原始连接错误：[Errno 11001] getaddrinfo failed

(注：具体的错误信息可能因系统而异，但结构应一致)

练习 3：设计更复杂的异常体系

要求：

定义一个基类 AppError 继承自 Exception。
从 AppError 派生出两个子类：ValidationError 和 AuthenticationError。
为 ValidationError 添加一个 field 属性，表示验证失败的字段名。
编写一个 login 函数，它接受用户名和密码。如果用户名为空，抛出 ValidationError；如果密码长度少于8位，也抛出 ValidationError；如果模拟的用户名或密码不匹配（可以简单实现），抛出 AuthenticationError。
编写一个 except 块来分别处理这些异常。

常见错误

忘记调用父类的 __init__：在自定义异常类中添加了自定义属性后，需要在 __init__ 方法中调用 super().__init__(message)，以确保异常的 args 和 str 等标准行为正确。否则，异常的打印可能不符合预期。

# 错误示例
class MyError(Exception):
    def __init__(self, code):
        self.code = code # 忘记了调用 super().__init__()！

# 正确示例
class MyError(Exception):
    def __init__(self, code, message=""):
        self.code = code
        super().__init__(message) # 或 super().__init__(f"Error {code}")

滥用或误用异常链：只在确实需要保留底层异常上下文时才使用 raise ... from ...。不要为了链而链。如果新异常与旧异常无关，直接 raise NewException() 即可。
异常捕获过于宽泛：不要总是使用 except Exception: 来捕获所有异常。这可能会掩盖真正的编程错误（如 TypeError、NameError）。应该尽量捕获特定的异常类型。使用自定义异常可以帮助实现更精确的捕获。
在自定义异常中处理过多逻辑：异常类应保持简单，主要用于携带错误信息和状态。复杂的错误处理逻辑应放在捕获异常的代码块中。

小结

本课我们学习了如何超越Python内置异常，为我们的程序量身定制错误信息：

自定义异常：通过继承 Exception 类，我们创建了有意义的、特定于业务的错误类型，使代码意图更清晰，错误处理更精准。
raise 语句：这是程序主动报告错误的"话筒"，允许我们在检测到问题时立即中断当前流程。
异常链：通过 raise ... from ...，我们可以将底层异常与高层业务异常关联起来，在保持业务语义清晰的同时，保留了宝贵的底层调试信息。

合理地使用这些技术，可以显著提高代码的可读性、可维护性和可调试性，是编写健壮、专业级 Python 程序的重要一步。在下一课中，我们将学习断言（assert）和日志记录（logging），它们是另外两种强大的调试和状态检查工具。

练习编辑器

python