Python上下文管理器

内容简介：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>

以上就是本文的全部内容，希望本文的内容对大家的学习或者工作能带来一定的帮助，也希望大家多多支持 码农网

查看所有标签

猜你喜欢:

• DDD：识别限界上下文以及理解上下文映射
• DDD：识别限界上下文以及理解上下文映射
• 如何划分限界上下文
• 详解Flask上下文
• Spring 上下文的加载
• 层叠等级和层叠上下文

本站部分资源来源于网络，本站转载出于传递更多信息之目的，版权归原作者或者来源机构所有，如转载稿涉及版权问题，请联系我们。