[gentoo-commits] proj/portage:master commit in: pym/_emerge/

[Zac Medico]

commit:     6b985978952fce6f1bb0741d8657d08a20c9937f
Author:     Zac Medico <zmedico <AT> gentoo <DOT> org>
AuthorDate: Wed Feb  8 05:16:51 2012 +0000
Commit:     Zac Medico <zmedico <AT> gentoo <DOT> org>
CommitDate: Wed Feb  8 05:16:51 2012 +0000
URL:        http://git.overlays.gentoo.org/gitweb/?p=proj/portage.git;a=commit;h=6b985978

AbstractPollTask: use timeout_add

---
 pym/_emerge/AbstractPollTask.py |   25 +++++++++++--------------
 1 files changed, 11 insertions(+), 14 deletions(-)

diff --git a/pym/_emerge/AbstractPollTask.py b/pym/_emerge/AbstractPollTask.py
index b4728e7..9229052 100644
--- a/pym/_emerge/AbstractPollTask.py
+++ b/pym/_emerge/AbstractPollTask.py
@@ -5,7 +5,6 @@ import array
 import errno
 import logging
 import os
-import time
 
 from portage.util import writemsg_level
 from _emerge.AsynchronousTask import AsynchronousTask
@@ -129,16 +128,14 @@ class AbstractPollTask(AsynchronousTask):
 				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
+		def timeout_cb():
+			timeout_cb.timed_out = True
+			return False
+		timeout_cb.timed_out = False
+		timeout_cb.timeout_id = self.scheduler.timeout_add(timeout, timeout_cb)
+
+		try:
+			while self._registered and not timeout_cb.timed_out:
+				self.scheduler.iteration()
+		finally:
+			self.scheduler.unregister(timeout_cb.timeout_id)