Question

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

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

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

Answer 2

with语句将上下文管理器__enter__ method的返回值绑定到变量。在您的情况下，这意味着p绑定到None（您的__enter__根本没有返回语句，因此None是默认值），{{1引发并调用AttributeError方法。

解决方案是从__exit__方法返回self：

__enter__

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

2 个答案: