__del__是不可靠的，但如果我尝试使用上下文管理器，则该对象不会持久存在

Question

我有一个实例化多个子对象的脚本。当脚本（和这些对象）结束时...对象需要做一些清理（特别是临时文件关闭和删除）。

我一直在阅读__del__如何不可靠 - 但是上下文管理似乎不起作用，子对象将不会持久存在。他们需要四处闲逛（做文件读写等事情）

一个例子：

WRITER.PY

import os

class Writer(object):
    def __init__(self, filename):
        self.filename = filename
        self.open()

    def open(self):
        self.fh = open(self.filename, "w+", 0)

    def write(self, msg):
        print >>self.fh, str(msg)

    def close(self):
        self.fh.close()
        os.remove(self.filename)

    def __enter__(self):
        print "entered"
#         self.open()


    def __exit__(self, type, value, traceback):
        print "__exit__"
        self.close()

MAIN.PY

def myfunc(filename):
    with WRITER.Writer(filename) as writeit:
        # do some stuff
        writeit.write("hallo")
        # do some more stuff
        writeit.write("more results")
        # even more stuff
        writeit.write("goodbye")

但是当我运行myfunc()时，一旦完成__init__()，对象就会被垃圾收集。它直接从enter到exit并且在with语句之后不执行任何任务。如果我将开放放在__init__或__enter__中，那似乎并不重要。

输出：

Traceback (most recent call last):
  File "/shared/GitHub/Tesera/MRAT_Refactor/bin/MAIN.py", line 13, in <module>
    myfunc("./tempfile")
  File "/shared/GitHub/Tesera/MRAT_Refactor/bin/MAIN.py", line 6, in myfunc
    writeit.write("hallo")
AttributeError: 'NoneType' object has no attribute 'write'
entered
__exit__            

有没有办法以这种方式使用上下文管理器，更好的方法来使用__del__ ......还是有第三种选择来实现这个目标？

Answer 1

问题与垃圾收集无关。作为documented，as子句绑定__enter__方法的返回值。你没有返回任何东西，所以你得到无。

如果您希望返回Writer对象，则需要在return self方法的末尾执行__enter__。