Python深入上下文管理器

傷城~ 2021-09-19 12:42 353阅读 0赞

上下文管理器(context manager)是Python2.5开始支持的一种语法，用于规定某个对象的使用范围。一旦进入或者离开该使用范围，会有特殊操作被调用 (比如为对象分配或者释放内存)。它的语法形式是with...as...

### 关闭文件 ###

我们会进行这样的操作：打开文件，读写，关闭文件。程序员经常会忘记关闭文件。上下文管理器可以在不需要文件的时候，自动关闭文件。

下面我们看一下两段程序：

![复制代码][copycode.gif]

# without context manager
    f = open("new.txt", "w")
    print(f.closed)               # whether the file is open
    f.write("Hello World!")
    f.close()
    print(f.closed)

![复制代码][copycode.gif]

以及：

# with context manager
    with open("new.txt", "w") as f:
        print(f.closed)
        f.write("Hello World!")
    print(f.closed)

两段程序实际上执行的是相同的操作。我们的第二段程序就使用了上下文管理器 (with...as...)。上下文管理器有隶属于它的程序块。当隶属的程序块执行结束的时候(也就是不再缩进)，上下文管理器自动关闭了文件 (我们通过f.closed来查询文件是否关闭)。我们相当于使用缩进规定了文件对象f的使用范围。

上面的上下文管理器基于f对象的\_\_exit\_\_()特殊方法(还记得我们如何利用特殊方法来实现各种语法？参看[特殊方法与多范式][Link 1])。当我们使用上下文管理器的语法时，我们实际上要求Python在进入程序块之前调用对象的\_\_enter\_\_()方法，在结束程序块的时候调用\_\_exit\_\_()方法。对于文件对象f来说，它定义了\_\_enter\_\_()和\_\_exit\_\_()方法(可以通过dir(f)看到)。在f的\_\_exit\_\_()方法中，有self.close()语句。所以在使用上下文管理器时，我们就不用明文关闭f文件了。

### 自定义 ###

任何定义了\_\_enter\_\_()和\_\_exit\_\_()方法的对象都可以用于上下文管理器。文件对象f是内置对象，所以f自动带有这两个特殊方法，不需要自定义。

下面，我们自定义用于上下文管理器的对象，就是下面的myvow：

![复制代码][copycode.gif]

# customized object
    
    class VOW(object):
        def __init__(self, text):
            self.text = text
        def __enter__(self):
            self.text = "I say: " + self.text    # add prefix
            return self                          # note: return an object
        def __exit__(self,exc_type,exc_value,traceback):
            self.text = self.text + "!"          # add suffix
    
    
    with VOW("I'm fine") as myvow:
        print(myvow.text)
    
    print(myvow.text)

![复制代码][copycode.gif]

我们的运行结果如下:

I say: I'm fine
    I say: I'm fine!

我们可以看到，在进入上下文和离开上下文时，对象的text属性发生了改变(最初的text属性是"I'm fine")。

\_\_enter\_\_()返回一个对象。上下文管理器会使用这一对象作为as所指的变量，也就是myvow。在\_\_enter\_\_()中，我们为myvow.text增加了前缀 ("I say: ")。在\_\_exit\_\_()中，我们为myvow.text增加了后缀("!")。

注意: \_\_exit\_\_()中有四个参数。当程序块中出现异常(exception)，\_\_exit\_\_()的参数中exc\_type, exc\_value, traceback用于描述异常。我们可以根据这三个参数进行相应的处理。如果正常运行结束，这三个参数都是None。在我们的程序中，我们并没有用到这一特性。

### 总结： ###

通过上下文管理器，我们控制对象在程序不同区间的特性。上下文管理器(with EXPR as VAR)大致相当于如下流程:

![复制代码][copycode.gif]

# with EXPR as VAR:
    
    VAR = EXPR
    VAR = VAR.__enter__()
    try:
        BLOCK
    finally:
        VAR.__exit__()

![复制代码][copycode.gif]

由于上下文管理器带来的便利，它是一个值得使用的工具。

**如果你对编程感兴趣或者想往编程方向发展，可以关注微信公众号【筑梦编程】，大家一起交流讨论！小编也会每天定时更新既有趣又有用的编程知识！**

[copycode.gif]: /images/20210812/6b6c3a3c6d64418b97bf995a817c2ea6.png
[Link 1]: http://www.cnblogs.com/vamei/archive/2012/11/19/2772441.html