AbstractPollTask: add _wait_loop method

1 files changed, 22 insertions, 1 deletions

diff --git a/pym/_emerge/AbstractPollTask.py b/pym/_emerge/AbstractPollTask.py
index 13e3c23f9..b4728e7a2 100644
--- a/pym/_emerge/AbstractPollTask.py
+++ b/pym/_emerge/AbstractPollTask.py
@@ -1,10 +1,11 @@
-# Copyright 1999-2011 Gentoo Foundation
+# Copyright 1999-2012 Gentoo Foundation
 # Distributed under the terms of the GNU General Public License v2
 
 import array
 import errno
 import logging
 import os
+import time
 
 from portage.util import writemsg_level
 from _emerge.AsynchronousTask import AsynchronousTask
@@ -121,3 +122,23 @@ class AbstractPollTask(AsynchronousTask):
 				self._unregister()
 				self.wait()
 
+	def _wait_loop(self, timeout=None):
+
+		if timeout is None:
+			while self._registered:
+				self.scheduler.iteration()
+			return
+
+		remaining_timeout = timeout
+		start_time = time.time()
+		while self._registered:
+			self.scheduler.iteration()
+			elapsed_time = time.time() - start_time
+			if elapsed_time < 0:
+				# The system clock has changed such that start_time
+				# is now in the future, so just assume that the
+				# timeout has already elapsed.
+				break
+			remaining_timeout = timeout - 1000 * elapsed_time
+			if remaining_timeout <= 0:
+				break