浅谈Python中with(上下文管理器)的用法,,例子一首先来看一段代


例子一

首先来看一段代码:

class Foo(object):    def __init__(self):        print(‘实例化一个对象‘)    def __enter__(self):        print(‘进入‘)    def __exit__(self, exc_type, exc_val, exc_tb):        print(‘退出‘)obj = Foo()with obj:    print(‘正在执行‘)

上面代码执行结果为:

实例化一个对象进入正在执行退出

结论1

我们知道,实例化Foo,得到obj对象,会执行Foo的__init__方法,也就是打印了第一句;

按照,程序从上至下执行,应该会打印“正在执行”才对,为什么会在它之前先打印了进入,在它之后打印了退出呢?

因为我们在定义Foo时,定义了__enter__和__exit__方法,那么我们实例化的对象obj就是一个上下文管理器,

含有__enter__和__exit__方法的对象就是上下文管理器。

with 上下文管理器:     语句体

当with遇到上下文管理器,就会在执行语句体之前,先执行上下文管理器的__enter__方法,然后再执行语句体,执行完语句体后,最后执行__exit__方法

这也就是为什么会出现文章开头的情况的原因。

例子二

再看看这段代码:

class Foo(object):    def __init__(self):        print(‘实例化一个对象‘)    def __enter__(self):        print(‘进入‘)    def __exit__(self, exc_type, exc_val, exc_tb):        print(‘退出‘)     # return Trueobj = Foo()with obj:    raise ImportError    print(‘正在执行‘)

结果如下:

技术图片

把上面代码中我们注释掉的那一行代码取消注释,结果如下

技术图片

我们会发现,虽然我们故意在语句体中抛出一个错误,按照正常情况,执行到报错地方就不会执行了,而__exit__是在语句体执行完之后执行的,但还是执行了__exit__方法;当我们在__exit__中给一个返回值为Ture时,就会忽略错误。

结论2

所有我们可以发现

with语句类似

  try :

  except:

  finally:

的功能:但是with语句更简洁。而且更安全。代码量更少。

  出现异常时,如果 __exit__ 返回 False(默认不写返回值时,即为False),则会重新抛出异常,让with 之外的语句逻辑来处理异常,这也是通用做法;如果返回 True,则忽略异常,不再对异常进行处理

例子三

class Foo(object):    def __init__(self):        print(‘实例化一个对象‘)    def __enter__(self):        print(‘进入‘)        # return self    def __exit__(self, exc_type, exc_val, exc_tb):        print(‘退出‘)with Foo() as obj:    print(obj,type(obj))    print(‘正在执行‘)

技术图片

把上面代码中我们注释掉的那一行代码取消注释,结果如下

技术图片

结论

调用上下文管理器的 __enter__ 方法时;如果使用了 as 子句,则将 __enter__() 方法的返回值赋值给 as 子句中的目标

with 上下文管理器  as  target:    代码语句体

with后面必须跟一个上下文管理器,如果使用了as,则是把上下文管理器的 __enter__() 方法的返回值赋值给 target,target 可以是单个变量,或者由“()”括起来的元组(不能是仅仅由“,”分隔的变量列表,必须加“()”)

例子四

我们经常会看到这样的代码:

with open("/tmp/foo.txt") as file:    data = file.read()

结论

这里使用了 with 语句,不管在处理文件过程中是否发生异常,都能保证 with 语句执行完毕后已经关闭了打开的文件句柄。如果使用传统的 try/finally 范式,则要使用类似如下代码:

somefile = open(r‘somefileName‘)try:    for line in somefile:        print line        # ...more codefinally:    somefile.close()

比较起来,使用 with 语句可以减少编码量。已经加入对上下文管理协议支持的还有模块 threading、decimal 等。

补充

with只能配合上下文管理器使用,常见的上下文管理器有

filedecimal.Contextthread.LockTypethreading.Lockthreading.RLockthreading.Conditionthreading.Semaphorethreading.BoundedSemaphore

浅谈Python中with(上下文管理器)的用法

评论关闭