python3基础：with上下文管理机制

最新推荐文章于 2021-10-03 12:04:48 发布

小小小小人水豚

最新推荐文章于 2021-10-03 12:04:48 发布

阅读量1.8k

点赞数 2

分类专栏： python

python 专栏收录该内容

75 篇文章 31 订阅

订阅专栏

with是一种上下文管理协议，目的在于从流程图中把 try,except 和finally 关键字和资源分配释放相关代码统统去掉，简化try….except….finlally的处理流程。with通过enter方法初始化，然后在exit中做善后以及处理异常。所以使用with处理的对象必须有__enter__()和__exit__()这两个方法。其中__enter__()方法在语句体（with语句包裹起来的代码块）执行之前进入运行，exit()方法在语句体执行完毕退出后运行。
with 语句适用于对资源进行访问的场合，确保不管使用过程中是否发生异常都会执行必要的“清理”操作，释放资源，比如文件使用后自动关闭、线程中锁的自动获取和释放等。

with语法

语法格式：

with expression [as target]:
    with_body

参数说明：
expression：是一个需要执行的表达式；
target：是一个变量或者元组，存储的是expression表达式执行返回的结果，可选参数。

with的使用场景

如果某项工作完成后需要有释放资源或者其他清理工作，比如说文件操作时，就可以使用with优雅的处理，不用自己手动关闭文件句柄，而且with还能很好的管理上下文异常。
代码示例：

with open('a.py') as fp:
      for line in fp:
            print(line,end=" ")

在这里插入图片描述

with工作原理

with后面的语句被求值后，该语句返回的对象的__enter__()方法被调用，这个方法将返回的值赋给as后面的变量，当with包围的语句块全部执行完毕后，自动调用对象的__exit__()方法。
with处理异常会更方便，省去try…else…finally复杂的流程，这个用到的就是对象的__exit__()方法：
exit( exc_type, exc_value, exc_trackback)
后面三个参数是固定的，用来接收with执行过程中异常类型，异常名称和异常的详细信息

代码示例：

class sample(object):
      def __enter__(self):
            return self
      def __exit__(self, exc_type, exc_value, exc_traceback):
            print("type:",exc_type)
            print("value:",exc_value)
            print("traceback:",exc_traceback)
      def fun(self):
            return 1/0
#调用自定义的异常类
with sample() as s:
      s.fun()

在这里插入图片描述

自定义with异常

从前面我们知道，with语句最关键的地方在于被求值对象必须有__enter__()和__exit__()这两个方法，那我们就可以通过实现这两方法来自定义with语句处理异常。

代码示例：

class opened(object):
    def __init__(self, filename):
        self.handle = open(filename,encoding='utf-8')
        print ('Resource: %s'% filename)
    def __enter__(self):
        print ('[Enter %s]: Allocate resource.' % self.handle)
        return self.handle  # 可以返回不同的对象
    def __exit__(self, exc_type, exc_value, exc_trackback):#后面三个参数是固定的
        print ('[Exit %s]: Free resource.' % self.handle)
        if exc_trackback is None:#有异常信息会存在exc_trackback中
            print ('[Exit %s]: Exited without exception.,' % self.handle)
            self.handle.close()
        else:
            print ("error occur!")  #句柄泄漏
        #return True  #异常不抛出
        return False  #会抛出异常，中断程序的执行

with opened(r'f:\a.py') as fp:
    for line in fp.readlines():
        print(line)
    raise TypeError

print ("Dnoe")

示例代码说明：

opened中的__enter__() 返回的是自身的引用，这个引用可以赋值给 as 子句中的fp变量；返回值的类型可以根据实际需要设置为不同的类型，不必是上下文管理器对象本身。
exit() 方法中对变量exc_trackback进行检测，缺省值为None，如果不为 None，表示发生了异常；没有异常发生，exit() 的三个参数都为None，上下文管理代码可以检测这种情况，做正常处理。exit()方法的3个参数，分别代表异常的类型、值、以及堆栈信息。
不管执行过程中是否发生了异常，都会执行上下文管理器的 exit() 方法。exit() 方法负责执行“清理”工作，如释放资源等。如果执行过程中没有出现异常，或者语句体中执行了语句 break/continue/return，则以 None 作为参数调用 exit(None, None, None) ；如果执行过程中出现异常，则使用 sys.exc_info 得到的异常信息为参数调用__exit__(exc_type, exc_value, exc_traceback)
出现异常时，如果 exit(type, value, traceback) 返回 False，则会重新抛出异常，让with 之外的语句逻辑来处理异常，这也是通用做法；如果返回 True，则忽略异常，不再对异常进行处理。

exit()方法返回True:
在这里插入图片描述