Please PyPy

celery · Apr 22, 2015 · 6efba62 · 6efba62
1 parent d62629f
commit 6efba62
Showing 1 changed file with 8 additions and 8 deletions.
diff --git a/celery/utils/functional.py b/celery/utils/functional.py
@@ -84,17 +84,17 @@ def __setitem__(self, key, value):
     def __iter__(self):
         return iter(self.data)
 
-    def _iterate_items(self):
-        for k in self:
-            try:
-                yield (k, self.data[k])
-            except KeyError:  # pragma: no cover
-                pass
+    def _iterate_items(self, _need_lock=IS_PYPY):
+        with self.mutex if _need_lock else DummyContext():
+            for k in self:
+                try:
+                    yield (k, self.data[k])
+                except KeyError:  # pragma: no cover
+                    pass
     iteritems = _iterate_items
 
     def _iterate_values(self, _need_lock=IS_PYPY):
-        ctx = self.mutex if _need_lock else DummyContext()
-        with ctx:
+        with self.mutex if _need_lock else DummyContext():
             for k in self:
                 try:
                     yield self.data[k]