我建议使用 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 类来解决此问题。然后,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