如何正确清理Python对象？

Question

class Package:
    def __init__(self):
        self.files = []

    # ...

    def __del__(self):
        for file in self.files:
            os.unlink(file)

__del__(self)以上失败，并出现AttributeError异常。我了解Python在__del__()调用时不保证 “全局变量”（在这种情况下是否存在成员数据）的存在。如果是这种情况，并且这是导致异常的原因，那么如何确保对象正确销毁？

Answer 1

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