1

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

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

一个例子：

作家.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()

主文件

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__(). 它直接从进入到退出，并且在 with 语句之后不执行任何职责。__init__我是否将 open 放在或 中似乎并不重要__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__......或者是否有第三种选择来实现这一点？

4

1 回答 1

8

该问题与垃圾收集无关。如文档所述，该as子句绑定了方法的返回值__enter__。你没有返回任何东西，所以你得到了 None。

如果要Writer返回对象，则需要return self在方法结束时执行__enter__。

于 2014-12-21T08:17:15.197 回答