我建议使用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子句。你不会希望发生这种事的。你可以通过创建一个PackageResource类来修复这个问题,这个类定义了__enter__和__exit__方法。然后,Package类将在__enter__方法中严格定义并返回。这样,调用者不使用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