对于在python中使用`with`关键字的类,我可以自己使用__repr __(print)吗？

我正在尝试创建一个与Python中的with关键字很好地匹配的对象.我知道你必须创建__enter__和__exit__方法,但我不太确定如何操纵对象.作为一个具体的例子,我写了一个创建本地空间的类,并在退出时清理：

import tempfile, os, shutil
class temp_workspace(object):

    def __enter__(self):
        self.local_dir = os.getcwd()
        self.temp_dir  = tempfile.mkdtemp()
        os.chdir(self.temp_dir)

    def __exit__(self, exc_type, exc_value, traceback):
        os.chdir(self.local_dir)
        shutil.rmtree(self.temp_dir)

    def __repr__(self):
        return self.temp_dir

这工作正常,但当我尝试打印本地目录名称时：

with temp_workspace() as T:
    print "Temp directory name is ", T

它显示为None,甚至没有调用__repr__！这真是令人困惑,因为T也是NoneType.我究竟做错了什么？

解决方法:

您没有按照context manager protocol的指定从__enter__返回对象.将return self添加到__enter__方法的末尾.

标签：with-statement,python
来源： https://codeday.me/bug/20190901/1782432.html