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 True

obj = 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 code
finally:
somefile.close()
比较起来,使用 with 语句可以减少编码量。已经加入对上下文管理协议支持的还有模块 threading、decimal 等。

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

file
decimal.Context
thread.LockType
threading.Lock
threading.RLock
threading.Condition
threading.Semaphore
threading.BoundedSemaphore

上一篇:tensorboardx


下一篇:使用动态路由实现三层架构网络互通