• 处理异常

    处理异常

    我们还没有谈到__exit__方法的这三个参数：type, value和traceback。
    在第4步和第6步之间，如果发生异常，Python会将异常的type,value和traceback传递给__exit__方法。
    它让__exit__方法来决定如何关闭文件以及是否需要其他步骤。在我们的案例中，我们并没有注意它们。

    那如果我们的文件对象抛出一个异常呢？万一我们尝试访问文件对象的一个不支持的方法。举个例子：

    1. with File('demo.txt', 'w') as opened_file:
    2.     opened_file.undefined_function('Hola!')

    我们来列一下，当异常发生时，with语句会采取哪些步骤。

    1. 它把异常的type,value和traceback传递给__exit__方法
    2. 它让__exit__方法来处理异常
    3. 如果__exit__返回的是True，那么这个异常就被优雅地处理了。
    4. 如果__exit__返回的是True以外的任何东西，那么这个异常将被with语句抛出。

    在我们的案例中，__exit__方法返回的是None(如果没有return语句那么方法会返回None)。因此，with语句抛出了那个异常。

    1. Traceback (most recent call last):
    2.   File "<stdin>", line 2, in <module>
    3. AttributeError: 'file' object has no attribute 'undefined_function'

    我们尝试下在__exit__方法中处理异常：

    1. class File(object):
    2.     def __init__(self, file_name, method):
    3.         self.file_obj = open(file_name, method)
    4.     def __enter__(self):
    5.         return self.file_obj
    6.     def __exit__(self, type, value, traceback):
    7.         print("Exception has been handled")
    8.         self.file_obj.close()
    9.         return True
    11. with File('demo.txt', 'w') as opened_file:
    12.     opened_file.undefined_function()
    14. # Output: Exception has been handled

    我们的__exit__方法返回了True,因此没有异常会被with语句抛出。

    这还不是实现上下文管理器的唯一方式。还有一种方式，我们会在下一节中一起看看。