python - 保证文件关闭

Question

我有一个类，我在构造函数中创建了一个文件对象。此类还实现了一个 finish() 方法作为其接口(interface)的一部分，在该方法中我关闭了文件对象。问题是如果我在这之前得到异常，文件将不会被关闭。有问题的类有许多使用文件对象的其他方法。我是否需要将所有这些都包装在 try finally 子句中，还是有更好的方法？

谢谢，

巴里

Answer 1

您可以使您的类成为上下文管理器，然后将对象创建和该类的使用包装在 with 语句中。参见 PEP 343了解详情。

要使您的类成为上下文管理器，它必须实现方法 __enter__() 和 __exit__()。 __enter__() 在您进入 with 语句时被调用，而 __exit__() 保证在您离开时被调用，无论怎么样。

然后你可以像这样使用你的类:

with MyClass() as foo:
    # use foo here

如果您在构造函数中获取资源，您可以使 __enter__() 简单地返回 self 而无需执行任何操作。 __exit__() 应该只调用您的 finish() 方法。