上下文管理器出口在体内代码之前调用

Question

我一直在玩Python自己的上下文管理器.我看到一些奇怪的行为很可能是由于我的实施.

我看到__exit__在'with'上下文中的语句之前调用的代码.例如,以下是代码段:

with ProgressBar(10) as p:
  p.update(1)

这是例外:

Traceback (most recent call last):
  File "<stdin>", line 3, in <module>
AttributeError: 'NoneType' object has no attribute 'update'

我把调试中的所有__enter__,__exit__和更新我的上下文管理器的方法.它看起来像是__exit__在update()之前调用的.这没有任何意义所以我必须遗漏一些简单的东西.

这是我的简单上下文管理器类:

class ProgressBar(object):
    """Progress bar that normalizes progress to [0 - 100] scale"""

    def __init__(self, max_value):
        """Create progress bar with max_value"""

        self._current_value = 0.0
        self._completed_value = 100.0
        self._max_value = float(max_value)
        print 'init', max_value

    def __enter__(self):
        """Start of context manager, 'with' statement"""

        print 'enter'
        self._current_value = 0.0

    def __exit__(self, exc_type, exc_value, traceback):
        """Start of context manager, 'with' statement"""

        print 'exit'
        self._current_value = self._completed_value

        # Not handling any exceptions, so they'll be raised automatically
        # To ignore exceptions return True or inspect arguments to handle

        return False

    def update(self, value):
        """Update progress value"""

        print 'update'
        if value >= self._max_value:
            self._current_value = 100
        else:
            self._current_value = (value / self._max_value) * self._completed_value

        print '\r%s' % (self._current_value),

Answer 1

从文档:

object.__enter__(self)

输入与此对象相关的运行时上下文.该with语句将此方法的返回值绑定到as语句子句中指定的目标(如果有).

你没有从中返回任何东西__enter__(因此你一如既往地返回None).如果你return self,你会得到

init 10
enter
update
10.0 exit