我建议使用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