Python中with语句用法详解：文件处理与自动资源管理

python中with的使用说明

Python 的用法

如何在Python中使用with

1 什么是With语句？

有些任务可能需要提前设置并在事后清理。

对于这种场景，Python的with语句提供了一种非常方便的处理方式。一个很好的例子是文件处理，您需要获取文件句柄，从文件中读取数据，然后关闭文件句柄。

我们都知道打开文件有两种方法：

f = open()
with open() as f:

这两种方法的区别在于，第一种方法需要我们自己关闭文件，f.close();

第二种方法不需要我们自己关闭文件。不管有没有异常，with都会自动帮我们关闭文件。

2 f = open() 与 open() as f 之间的比较

如果不使用with语句，代码如下：

file = open("/tmp/foo.txt")
data = file.read()
file.close()

这里有两个问题：

首先，你可能忘记关闭文件句柄；

其次，从文件中读取数据时出现异常，没有进行任何处理。

这是处理异常的增强版本：

file = open("/tmp/foo.txt")
try:
    data = file.read()
finally:
    file.close()

这段代码工作正常，但过于冗长。这时候，with就显示出了它的优势。 with 除了具有更优雅的语法之外，还可以很好地处理上下文产生的异常。这是 with 版本的代码：

with open("/tmp/foo.txt") as file:
    data = file.read()

是不是很简单呢？

但如果你不熟悉with的工作原理，你可能会像我刚才一样，不明白原理。

那么我们先简单看一下with的工作原理

3 怎样工作？

基本思想是：用with求值的对象必须有一个enter()方法和一个exit()方法。

在对紧随 with 的语句求值后，调用返回对象的 Enter() 方法，并将该方法的返回值赋给 as 后面的变量。当 with 后面的所有代码块都被执行时，之前返回的对象的 exit() 方法将被调用。

以下示例说明了 with 的工作原理：

#!/usr/bin/env python
# with_example01.py
class Sample:
    def __enter__(self):
        print "In __enter__()"
        return "Foo"
    def __exit__(self, type, value, trace):
        print "In __exit__()"
def get_sample():
    return Sample()
with get_sample() as sample:
    print "sample:", sample

运行代码，输出如下

bash-3.2$ ./with_example01.py
In __enter__()
sample: Foo
In __exit__()