【原】python基础篇：如何把异常处理做的更加优雅

攻城狮成长日志 2023-10-13 发布于广东

展开全文

为什么需要异常处理

在 Python 中，异常处理是一种处理程序错误的方法。当程序出现错误时，它会引发异常并停止执行。异常处理允许我们在程序出现错误时采取措施，而不是让程序崩溃。

异常类型

Python 中有许多内置的异常类型，例如 TypeError、ValueError、ZeroDivisionError 等。每个异常类型都有其特定的原因和解决方法。

try 和 except 语句

要处理异常，我们可以使用 try 和 except 语句。try 语句包含可能引发异常的代码块，而 except 语句包含处理异常的代码块。

以下是一个简单的例子：

try:
    # 可能引发异常的代码块
    x = 1 / 0
except ZeroDivisionError:
    # 处理异常的代码块
    print("除数不能为零")

在上面的代码中，我们尝试将 1 除以 0，这会引发 ZeroDivisionError 异常。然后，我们使用 except 语句来处理这个异常，并打印一条消息。

else 语句

除了 try 和 except 语句外，Python 还提供了 else 语句。else 语句包含在 try 语句中，但在没有引发异常的情况下执行。

以下是一个例子：

try:
    # 可能引发异常的代码块
    x = int(input("请输入一个整数："))
except ValueError:
    # 处理 ValueError 异常的代码块
    print("输入无效")
else:
    # 在没有异常的情况下执行的代码块
    print("你输入的是：", x)

在上面的代码中，我们尝试将用户输入的字符串转换为整数。如果用户输入的不是整数，将引发 ValueError 异常。否则，我们将打印用户输入的整数。

finally 语句

最后，Python 还提供了finally 语句。finally 语句包含在 try 语句中，并在 try 语句结束后执行，无论是否引发异常。

以下是一个例子：

try:
    # 可能引发异常的代码块
    f = open("file.txt", "r")
    # 读取文件内容
finally:
    # 关闭文件
    f.close()

在上面的代码中，我们尝试打开一个文件并读取其内容。无论是否引发异常，我们都需要关闭文件。因此，我们使用 finally 语句来确保文件被关闭。

自定义异常

除了内置的异常类型外，Python 还允许我们创建自定义异常。要创建自定义异常，我们可以定义一个新的类，并从 Exception 类继承。

以下是一个例子：

class MyError(Exception):
    def __init__(self, message):
        self.message = message

try:
    # 可能引发自定义异常的代码块
    raise MyError("这是一个自定义异常")
except MyError as e:
    # 处理自定义异常的代码块
    print(e.message)

在上面的代码中，我们定义了一个名为 MyError 的自定义异常类，并在 「init」 方法中定义了一个 message 属性。然后，我们尝试引发这个自定义异常，并使用 except 语句来处理它。

装饰器封装异常处理

当业务逻辑变得非常复杂时，代码中这种try语句就显得不是很美观而且影响阅读代码，我们可以把它做成装饰器。下面是一个简单的例子：

def handle_exceptions(func):
    def wrapper(*args, **kwargs):
        try:
            return func(*args, **kwargs)
        except Exception as e:
            print(f"An error occurred: {e}")
    return wrapper

@handle_exceptions
def divide(x, y):
    return x / y

result = divide(1, 0)

在上面的代码中，我们定义了一个名为 handle_exceptions 的装饰器函数。该函数接受一个函数作为参数，并返回一个新的函数 wrapper。wrapper 函数包含了原始函数的逻辑，并在 try 语句中执行该逻辑。如果发生异常，except 语句将捕获该异常并打印一条错误消息。

然后，我们使用@handle_exceptions装饰器将divide函数进行装饰。当我们调用divide(1, 0)时，由于除数为零，将引发 ZeroDivisionError 异常。但是，由于我们已经将 divide 函数装饰成了handle_exceptions(divide)，因此异常将被捕获并打印一条错误消息。

这样，我们就可以将异常处理逻辑封装成一个装饰器函数，并在需要时将其应用到其他函数上。