处理异常

我们还没有谈到__exit__方法的这三个参数：type, value和traceback。

在第4步和第6步之间，如果发生异常，Python会将异常的type,value和traceback传递给__exit__方法。

它让__exit__方法来决定如何关闭文件以及是否需要其他步骤。在我们的案例中，我们并没有注意它们。

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

  

   

    

     

    
    

     

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

    

     

    
    

     
 
     
    
   

    

     

    
    

     

          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语句抛出了那个异常。

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语句抛出。

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

基于生成器的实现

我们还可以用装饰器(decorators)和生成器(generators)来实现上下文管理器。

Python有个contextlib模块专门用于这个目的。我们可以使用一个生成器函数来实现一个上下文管理器，而不是使用一个类。

让我们看看一个基本的，没用的例子：

  

   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

      from contextlib import contextmanager
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

      @contextmanager
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

      def open_file(name):
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

          f = open(name, 'w')
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

          yield f
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

          f.close()
     
    
  
 

OK啦！这个实现方式看起来更加直观和简单。然而，这个方法需要关于生成器、yield和装饰器的一些知识。在这个例子中我们还没有捕捉可能产生的任何异常。它的工作方式和之前的方法大致相同。

让我们小小地剖析下这个方法。

1. Python解释器遇到了yield关键字。因为这个缘故它创建了一个生成器而不是一个普通的函数。

2. 因为这个装饰器，contextmanager会被调用并传入函数名(open_file)作为参数。

3. contextmanager函数返回一个以GeneratorContextManager对象封装过的生成器。

4. 这个GeneratorContextManager被赋值给open_file函数，我们实际上是在调用GeneratorContextManager对象。

那现在我们既然知道了所有这些，我们可以用这个新生成的上下文管理器了，像这样：

  

   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

      with open_file('some_file') as f:
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

          f.write('hola!')
     
    
  
 

至此，Python进阶已结束，感谢支持。

文章来源: zhulin1028.blog.csdn.net，作者：zhulin1028，版权归原作者所有，如需转载，请联系作者。

原文链接：zhulin1028.blog.csdn.net/article/details/125112227