Python进阶
搜索文档…
⌃K

处理异常

我们还没有谈到 __exit__ 方法的这三个参数:typevaluetraceback。 在第4步和第6步之间,如果发生异常,Python 会将异常的 typevaluetraceback 传递给 __exit__ 方法。 它让 __exit__ 方法来决定如何关闭文件以及是否需要其他步骤。在我们的案例中,我们并没有注意它们。
那如果我们的文件对象抛出一个异常呢?万一我们尝试访问文件对象的一个不支持的方法。举个例子:
with File('demo.txt', 'w') as opened_file:
opened_file.undefined_function('Hola!')
我们来列一下,当异常发生时,with 语句会采取哪些步骤。
  1. 1.
    它把异常的 typevaluetraceback 传递给 __exit__方法。
  2. 2.
    它让 __exit__ 方法来处理异常。
  3. 3.
    如果 __exit__ 返回的是 True,那么这个异常就被优雅地处理了。
  4. 4.
    如果 __exit__ 返回的是 True 以外的任何东西,那么这个异常将被 with 语句抛出。
在我们的案例中,__exit__ 方法返回的是 None (如果没有 return 语句那么方法会返回 None)。因此,with 语句抛出了那个异常。
Traceback (most recent call last):
File "<stdin>", line 2, in <module>
AttributeError: 'file' object has no attribute 'undefined_function'
我们尝试下在 __exit__ 方法中处理异常:
class File(object):
def __init__(self, file_name, method):
self.file_obj = open(file_name, method)
def __enter__(self):
return self.file_obj
def __exit__(self, type, value, traceback):
print("Exception has been handled")
self.file_obj.close()
return True
with File('demo.txt', 'w') as opened_file:
opened_file.undefined_function()
# Output: Exception has been handled
我们的 __exit__ 方法返回了 True,因此没有异常会被 with 语句抛出。
这还不是实现上下文管理器的唯一方式。还有一种方式,我们会在下一节中一起看看。