我建议使用Python的
with语句来管理需要清理的资源。使用显式
close()语句的问题在于,你必须担心人们会忘记完全调用它,或者忘记将其放在
finally块中以防止发生异常时发生资源泄漏。
要使用该
with语句,请使用以下方法创建一个类:
def __enter__(self) def __exit__(self, exc_type, exc_value, traceback)
在上面的示例中,你将使用
class Package: def __init__(self): self.files = [] def __enter__(self): return self # ... def __exit__(self, exc_type, exc_value, traceback): for file in self.files: os.unlink(file)
然后,当有人想使用你的课程时,他们将执行以下操作:
with Package() as package_obj: # use package_obj
变量
package_obj将是
Package类型的实例(它是
__enter__方法返回的值)。
__exit__无论是否发生异常,都会自动调用其方法。
你甚至可以进一步采用这种方法。在上面的示例中,仍然可以使用其构造函数实例化Package而无需使用该
with子句。你不希望这种情况发生。你可以通过创建定义
__enter__和
__exit__方法的
PackageResource类来解决此问题。然后,将严格在
__enter__方法内部定义Package类并返回。这样,调用者永远无法在不使用with语句的情况下实例化Package类:
class PackageResource: def __enter__(self): class Package: ... self.package_obj = Package() return self.package_obj def __exit__(self, exc_type, exc_value, traceback): self.package_obj.cleanup()
你将按以下方式使用它:
with PackageResource() as package_obj: # use package_obj



