在Python中定义清理动作

在Python中，我们经常需要在程序运行完毕后进行清理工作，比如关闭文件、网络连接等。为了方便管理这些清理动作，Python提供了with语句以及contextlib模块来定义清理动作。

使用with语句定义清理动作

使用with语句可以更方便、更安全地管理资源，可以保证资源在离开with语句块后一定会被释放。

with open('example.txt', 'r') as f:
    # do something with the file

在这个例子中，open('example.txt', 'r')返回的文件对象被赋值给变量f。一旦with语句块结束，Python会自动调用f.close()关闭文件。

同时，with语句还支持同时使用多个上下文管理器：

with open('example.txt', 'r') as f, open('output.txt', 'w') as output:
    # do something with the files

在这个例子中，我们打开了两个文件，分别用于读取和写入。在离开with语句块后，f和output都会被自动关闭。

使用contextlib模块定义清理动作

除了使用with语句，Python还提供了contextlib模块来定义清理动作。这个模块的主要作用是负责生成一个上下文对象，从而提供了类似于with语句的功能。

最常用的函数是closing，用于关闭实现了close方法的对象：

from contextlib import closing
import urllib

with closing(urllib.urlopen('http://www.example.com/')) as webpage:
    # do something with the page

在这个例子中，我们使用closing装饰器将urllib.urlopen返回的对象打包成上下文管理器，然后在with语句块内读取该网页。

除了closing，contextlib模块还提供了其他一些函数，比如redirect_stdout、redirect_stderr等，用于重定向输出。

总之，with语句和contextlib模块为我们提供了一种简洁、高效的方式来管理资源和执行清理操作，这对于编写高质量的Python程序至关重要。