Python中如何自定义异常

在Python中，我们可以自定义自己的异常类，来处理一些特定的错误或异常情况。自定义异常类可以继承自Python内置的Exception类或其子类，也可以继承自其他自定义的异常类。

自定义异常类的好处是能够更好地组织和管理代码，将特定的错误或异常情况进行分类，从而使得代码更加模块化和可读性更高。下面我们将通过一个例子来演示如何自定义异常类。

假设我们正在开发一个图书管理系统，在进行图书借阅时，如果用户借阅的图书数量超过了系统中该图书的库存数量，我们就希望抛出一个自定义的异常来提示用户。

首先，我们可以定义一个图书库存不足的异常类 BookStockException，继承自Exception类。代码如下：

class BookStockException(Exception):
    def __init__(self, book_id, current_stock):
        self.book_id = book_id
        self.current_stock = current_stock

    def __str__(self):
        return f"Book stock is not enough. Book ID: {self.book_id}, Current stock: {self.current_stock}"

在该自定义异常类中，我们定义了两个属性 book_id 和 current_stock，分别表示图书的ID和当前库存数量，方便后续代码的处理。我们还重写了 __str__ 方法，使得在打印异常时能够直观地给出异常的相关信息。

接下来，我们可以在图书借阅的函数中使用该自定义异常类。假设我们有一个函数 borrow_book，接收两个参数：book_id 表示图书的ID，num 表示用户借阅的数量。代码如下：

def borrow_book(book_id, num):
    # 假设库存数量存储在一个字典中，键是图书ID，值是库存数量
    stock_dict = {1: 5, 2: 3, 3: 10}

    if book_id not in stock_dict:
        raise BookStockException(book_id, 0)

    if num > stock_dict[book_id]:
        raise BookStockException(book_id, stock_dict[book_id])

    print("Borrow book successful!")

在 borrow_book 函数中，首先我们模拟了一个库存数量的字典 stock_dict，表示图书的库存情况。然后，我们判断用户借阅的图书是否存在，如果不存在，就抛出 BookStockException 异常，并传入 book_id 和当前库存数0。如果图书存在，再判断用户借阅的数量是否大于库存数量，如果大于，则同样抛出 BookStockException 异常，并传入 book_id 和当前库存数。如果两个判断都通过，表示借阅成功，打印"Borrow book successful!"。

接下来，我们可以在主程序中调用 borrow_book 函数，并进行异常处理。代码如下：

try:
    borrow_book(1, 10)
except BookStockException as e:
    print(e)

在主程序中，我们调用 borrow_book 函数并传入参数1和10，即尝试借阅ID为1的图书10本。由于库存只有5本，所以会触发第二个判断语句，抛出 BookStockException 异常，在异常处理部分将异常对象赋值给变量 e，并打印异常信息。最终结果为"Book stock is not enough. Book ID: 1, Current stock: 5"。

以上就是在Python中自定义异常的过程和使用例子，通过自定义异常类，我们可以更好地管理和组织代码，提高代码的可读性和可维护性。