我們還沒有談到__exit__方法的這三個(gè)參數(shù)：type, value和traceback。
在第4步和第6步之間，如果發(fā)生異常，Python會(huì)將異常的type,value和traceback傳遞給__exit__方法。
它讓__exit__方法來決定如何關(guān)閉文件以及是否需要其他步驟。在我們的案例中，我們并沒有注意它們。

那如果我們的文件對(duì)象拋出一個(gè)異常呢？萬一我們嘗試訪問文件對(duì)象的一個(gè)不支持的方法。舉個(gè)例子：

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

我們來列一下，當(dāng)異常發(fā)生時(shí)，with語句會(huì)采取哪些步驟。

它把異常的type,value和traceback傳遞給__exit__方法
它讓__exit__方法來處理異常
如果__exit__返回的是True，那么這個(gè)異常就被優(yōu)雅地處理了。
如果__exit__返回的是True以外的任何東西，那么這個(gè)異常將被with語句拋出。

在我們的案例中，__exit__方法返回的是None(如果沒有return語句那么方法會(huì)返回None)。因此，with語句拋出了那個(gè)異常。

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,因此沒有異常會(huì)被with語句拋出。

這還不是實(shí)現(xiàn)上下文管理器的唯一方式。還有一種方式，我們會(huì)在下一節(jié)中一起看看。

上一篇：列表推導(dǎo)式（<code>list</code> comprehensions）下一篇：字典推導(dǎo)式（<code>dict</code> comprehensions）