内容简介:Python上下文管理器
目录
with <context_manager_object> [as <variable>]:
with-block
上下文管理器,本质上是使用with语句来简化try/finally模式。它可以保证,即使在出错的情况下,也可以执行 某些诸如释放数据库连接等 清理语句。
实现了__enter__(self)和__exit__(self, exc_type, exc_value, traceback)这两个方法的对象,就支持上下文管理。当进入with语句块时,会执行上下文管理器对象的__enter__方法,并且把__enter__方法的返回值作为as子句后面的变量的值;当离开with语句块的时候,会执行上下文管理器对象的__exit__方法,并把with语句块中抛出的异常信息作为__exit__方法的参数。
[root@iZj6chejzrsqpclb7miryaZ ~]# cat test.py
import time
class TimeUsed:
def __init__(self):
self._start_time = None
self._end_time = None
def __enter__(self):
self._start_time = time.time()
def __exit__(self, exc_type, exc_value, traceback):
self._end_time = time.time()
print "time used: %ss" % (self._end_time - self._start_time)
if __name__ == "__main__":
def func(n):
time.sleep(n)
with TimeUsed():
func(2)
with TimeUsed():
func(1)
[root@iZj6chejzrsqpclb7miryaZ ~]# python test.py
time used: 2.00210905075s
time used: 1.00110411644s
contextmanager装饰器[]
为了简化上下文管理器的编写,Python标准库的contextlib模块提供了contextmanager装饰器,它可以把生成器函数包装成支持上下文管理的对象。
当进入with语句块的时候,会调用生成器对象的next方法,并把yield传递出的值,作为as子句后面的变量的值;当离开with语句块的时候,会调用生成器对象的next或throw方法(当with语句块出异常时,会调用生成器对象的throw方法,此时,可以 在生成器中对异常进行处理 ,不做任何异常处理的话,会重新抛出异常)。
[root@iZj6chejzrsqpclb7miryaZ ~]# cat test.py
from contextlib import contextmanager
import time
@contextmanager
def time_used():
start = time.time()
try:
yield
finally:
end = time.time()
print "time used: %s" % (end - start)
if __name__ == "__main__":
def func(n, will_raise=False):
time.sleep(n)
if will_raise:
raise RuntimeError("in func")
with time_used():
func(2)
with time_used():
func(1, True)
[root@iZj6chejzrsqpclb7miryaZ ~]# python test.py
time used: 2.00210595131
time used: 1.00112104416
Traceback (most recent call last):
File "test.py", line 25, in
<module>
func(1, True)
File "test.py", line 19, in func
raise RuntimeError("in func")
RuntimeError: in func
</module>
以上就是本文的全部内容,希望本文的内容对大家的学习或者工作能带来一定的帮助,也希望大家多多支持 码农网
猜你喜欢:本站部分资源来源于网络,本站转载出于传递更多信息之目的,版权归原作者或者来源机构所有,如转载稿涉及版权问题,请联系我们。
深入理解计算机系统
Randal E.Bryant、David O'Hallaron / 龚奕利、雷迎春 / 中国电力出版社 / 2004-5-1 / 85.00元
从程序员的视角,看计算机系统! 本书适用于那些想要写出更快、更可靠程序的程序员。通过掌握程序是如何映射到系统上,以及程序是如何执行的,读者能够更好的理解程序的行为为什么是这样的,以及效率低下是如何造成的。粗略来看,计算机系统包括处理器和存储器硬件、编译器、操作系统和网络互连环境。而通过程序员的视角,读者可以清晰地明白学习计算机系统的内部工作原理会对他们今后作为计算机科学研究者和工程师的工作有......一起来看看 《深入理解计算机系统》 这本书的介绍吧!