Python 异常处理机制：从基础语法到自定义异常的实战指南

二月龙
13 阅读5分钟

Python 异常处理机制：从基础语法到自定义异常的实战指南

在 Python 编程中，异常处理是构建健壮、可靠程序的基石。当程序遇到错误（如除零、文件不存在）时，如果未加处理，程序会直接崩溃。异常处理机制允许我们捕获这些错误，进行优雅的处理或记录，从而保证程序的持续运行。

本文将带你深入理解 Python 的异常处理机制，从基础的 try-except-else-finally 结构，到如何创建符合业务需求的自定义异常类。

核心结构：try-except-else-finally

Python 的异常处理结构由四个关键字组成，它们各司其职，共同协作。

try 块：风险代码的“隔离区”

try 块用于包裹可能引发异常的代码。Python 解释器会监控 try 块内的执行。一旦发生异常，try 块中剩余的代码会立即停止执行，程序流程会跳转到相应的 except 块。

except 块：异常的“处理器”

except 块紧跟在 try 块之后，用于定义当特定异常发生时应执行的操作。你可以捕获一个或多个具体的异常类型。

  • 捕获单个异常：针对特定错误进行处理。
  • 捕获多个异常：使用元组 (Exception1, Exception2) 在一个 except 块中处理多种异常。
  • 获取异常信息：使用 as e 语法可以捕获异常对象，从而获取详细的错误信息，便于调试和日志记录。
try:
    result = 10 / 0
except ZeroDivisionError as e:
    print(f"捕获到异常：{e}")  # 输出: 捕获到异常：division by zero

else 块：成功执行的“奖励区”

else 块是可选的，它只有在 try 块没有引发任何异常时才会执行。这是一个非常重要的设计，它将“可能出错的代码”和“依赖于成功执行的代码”清晰地分离开来。

finally 块：无论如何都要执行的“清理区”

finally 块也是可选的，但它在实际开发中极其重要。无论 try 块中是否发生异常，也无论异常是否被 except 捕获，finally 块中的代码总是会被执行。这使得它成为释放资源的理想场所，例如关闭文件、断开数据库连接等。

file = None
try:
    file = open("data.txt", "r")
    content = file.read()
except FileNotFoundError:
    print("文件未找到")
else:
    print("文件读取成功")
finally:
    if file:
        file.close()  # 确保文件总是被关闭
        print("文件已关闭")

主动抛出异常：raise

除了被动捕获异常，我们还可以在代码中主动抛出异常。这在参数校验、业务规则检查等场景中非常有用。使用 raise 语句可以手动引发一个异常。

def check_age(age):
    if age < 0:
        raise ValueError("年龄不能为负数")
    return age

try:
    check_age(-5)
except ValueError as e:
    print(f"参数校验失败：{e}")

自定义异常类：满足个性化需求

Python 内置的异常类型（如 ValueError, KeyError）虽然强大，但在复杂的业务场景中，我们往往需要更具语义化的异常来清晰地表达错误。这时，自定义异常类就派上用场了。

创建自定义异常非常简单，只需继承内置的 Exception 类或其子类即可。通过自定义异常，我们可以：

  • 增强代码可读性：LoginError 比通用的 Exception 更能说明问题。
  • 携带更多上下文信息：可以在异常类中定义额外的属性。
  • 构建异常体系：可以创建基类异常和更具体的子类异常，形成层次结构。

场景一：简单的自定义异常

class LoginError(Exception):
    """登录失败的自定义异常"""
    pass

def login(username, password):
    if username != "admin" or password != "123456":
        raise LoginError("用户名或密码错误")
    print("登录成功")

try:
    login("admin", "wrong_password")
except LoginError as e:
    print(f"登录失败：{e}")

场景二：携带额外信息的自定义异常

在更复杂的场景中，异常可能需要携带额外的数据，比如错误的值、错误码等。

class InsufficientFundsError(Exception):
    """余额不足异常"""
    def __init__(self, balance, amount):
        self.balance = balance
        self.amount = amount
        self.message = f"余额不足：当前余额为 {balance}，尝试取款 {amount}"
        super().__init__(self.message)

class BankAccount:
    def __init__(self, balance):
        self.balance = balance

    def withdraw(self, amount):
        if amount > self.balance:
            raise InsufficientFundsError(self.balance, amount)
        self.balance -= amount

# 使用示例
account = BankAccount(100)
try:
    account.withdraw(150)
except InsufficientFundsError as e:
    print(e)  # 输出: 余额不足：当前余额为 100，尝试取款 150
    # 我们还可以访问异常中的额外属性
    print(f"差额: {e.amount - e.balance}")

最佳实践与避坑指南

为了写出更健壮的代码，请遵循以下最佳实践：

  • 精准捕获：尽量捕获具体的异常类型（如 FileNotFoundError），避免使用过于宽泛的 except Exception 或空的 except:，因为它们会掩盖真正的编程错误，使问题难以排查。
  • 使用 finally 释放资源：所有涉及资源操作（文件、数据库、网络连接）的代码，都必须使用 finally 块或 with 语句来确保资源被正确释放。
  • 记录日志：在生产环境中，不要仅仅 print 异常信息。应使用 logging 模块将异常堆栈记录下来，便于后续分析和排查。
  • 提供友好的错误提示：给用户或调用方返回清晰、有用的错误信息，而不是原始的异常堆栈。
avatar
文章
阅读
粉丝