python中@contextmanager是什么?

美女程序员鼓励师

本文教程操作环境:windows7系统、Python 3.9.1,DELL G3电脑。

1、概念

contextlib模块提供的一个装饰器,用于将一个函数声明上下文管理,无需创建一个类或者单独的__enter__()方法和__exit__()方法,就可以实现上下文管理。

2、注意

装饰器@contextmanager真正的用法并不是简化上下文管理器,而是去定义上下文管理器的方法,是对原来不是上下文管理器的类变成了一个上下文管理器。

3、实例

from contextlib import contextmanager

 

import time

 

 

 

@contextmanager

 

def demo(label):

 

    start = time.time()

 

    try:

 

        yield

 

    finally:

 

        end = time.time()

 

        print('{}: {}'.format(label, end - start))

 

 

 

with demo('counting'):

 

    n = 10000000

 

    while n > 0:

 

        n -= 1

 

 

 

# counting: 1.32399988174

看上面这个例子,函数中yield之前的所有代码都类似于上下文管理器中__enter__方法的内容。而yield之后的所有代码都如__exit__方法的内容。如果执行过程中发生了异常,则会在yield语句触发。

以上就是python中@contextmanager的基本介绍,相信大家对于这种直接实现上下文管理器的方法,还是非常感兴趣的。看完文章后,可以运行上面的代码进行体验。

以上是 python中@contextmanager是什么? 的全部内容, 来源链接: utcz.com/z/543185.html

回到顶部