Use PollScheduler iteration method.

[portage.git] / pym / _emerge / AsynchronousLock.py

diff --git a/pym/_emerge/AsynchronousLock.py b/pym/_emerge/AsynchronousLock.py
index e166df344202ca7b98269a38cc8a57d54c22aa37..2c4fe108c4fd186a0f4e35a115fada95f675d767 100644 (file)
--- a/pym/_emerge/AsynchronousLock.py
+++ b/pym/_emerge/AsynchronousLock.py
@@ -1,4 +1,4 @@
-# Copyright 2010-2011 Gentoo Foundation
+# Copyright 2010-2012 Gentoo Foundation
 # Distributed under the terms of the GNU General Public License v2
 
 import dummy_threading
@@ -152,8 +152,8 @@ class _LockThread(AbstractPollTask):
        def _wait(self):
                if self.returncode is not None:
                        return self.returncode
-               if self._registered:
-                       self.scheduler.schedule(self._reg_id)
+               while self._registered:
+                       self.scheduler.iteration()
                return self.returncode
 
        def unlock(self):
@@ -264,8 +264,8 @@ class _LockProcess(AbstractPollTask):
        def _wait(self):
                if self.returncode is not None:
                        return self.returncode
-               if self._registered:
-                       self.scheduler.schedule(self._reg_id)
+               while self._registered:
+                       self.scheduler.iteration()
                return self.returncode
 
        def _output_handler(self, f, event):