Python 如何使用 raise 关键字抛出异常|极客教程

Python 如何使用 raise 关键字抛出异常

在 Python 中，我们可以使用 raise 关键字来主动抛出异常。抛出异常可以帮助我们在代码执行过程中发现问题并进行相应的处理，从而使我们的程序更加健壮。本文将详细介绍 Python 中如何使用 raise 关键字抛出异常，以及如何自定义异常类和处理异常。

使用 raise 抛出异常

在 Python 中，我们可以使用 raise 关键字来抛出各种类型的异常。下面是一个简单的示例，演示了如何使用 raise 抛出一个通用的 Exception 异常：

def divide(x, y):
    if y == 0:
        raise Exception("Division by zero")
    return x / y

try:
    result = divide(10, 0)
except Exception as e:
    print("An error occurred:", e)

# Output:
# An error occurred: Division by zero

在上面的示例中，我们定义了一个 divide 函数，用于计算两个数的除法。但是如果除数为 0，则会抛出一个 Exception 异常，并在 try-except 块中捕获并处理这个异常。

除了抛出通用的 Exception 异常外，我们还可以根据具体情况自定义异常类。下面是一个示例，演示了如何定义一个自定义的 MyCustomError 异常类，并使用 raise 抛出该异常：

class MyCustomError(Exception):
    pass

def random_function():
    raise MyCustomError("An error occurred in random_function")

try:
    random_function()
except MyCustomError as e:
    print("An error occurred:", e)

# Output:
# An error occurred: An error occurred in random_function

在上面的示例中，我们定义了一个名为 MyCustomError 的自定义异常类，并在 random_function 函数中使用 raise 抛出该异常。然后在 try-except 块中捕获并处理该异常。

自定义异常信息

除了在 raise 关键字后面指定异常类外，我们还可以为异常指定具体的错误信息。示例代码如下：

def check_input(input_str):
    if "geek-docs.com" not in input_str:
        raise ValueError("The input must contain 'geek-docs.com'")
    else:
        print("Input is valid")

try:
    check_input("example.com")
except ValueError as e:
    print("An error occurred:", e)

# Output:
# An error occurred: The input must contain 'geek-docs.com'

在上面的示例中，我们定义了一个 check_input 函数，用于检查输入字符串中是否包含 “geek-docs.com”。如果不包含，则会抛出一个 ValueError 异常，并指定错误信息为 “The input must contain ‘geek-docs.com'”。在 try-except 块中捕获并处理该异常。

继承内置异常类

除了自定义异常类外，我们还可以继承 Python 内置的异常类来创建自定义异常。下面是一个示例，演示了如何定义一个 CustomError 类，并继承自 ValueError 异常类：

class CustomError(ValueError):
    def __init__(self, message):
        self.message = message
        super().__init__(self.message)

def process_data(data):
    if len(data) < 5:
        raise CustomError("Data is too short")

try:
    process_data("Python")
except CustomError as e:
    print("An error occurred:", e)

# Output:
# An error occurred: Data is too short

在上面的示例中，我们定义了一个 CustomError 异常类，继承自 ValueError 异常类，并在 process_data 函数中根据特定条件抛出该异常。然后在 try-except 块中捕获并处理该异常。