diff options
author | Zac Medico <zmedico@gentoo.org> | 2012-02-07 21:16:51 -0800 |
---|---|---|
committer | Zac Medico <zmedico@gentoo.org> | 2012-02-07 22:08:45 -0800 |
commit | 7e38a0253bdc1a25b54f0783c0bb2a72239fdb05 (patch) | |
tree | 93a32e368b5c779b5e8dbdf84ca3ff89cda853d2 | |
parent | c02b0e535a4cfe4dc0a3c3d07124b8dfa5d4f9a5 (diff) | |
download | portage-7e38a0253bdc1a25b54f0783c0bb2a72239fdb05.tar.gz portage-7e38a0253bdc1a25b54f0783c0bb2a72239fdb05.tar.bz2 portage-7e38a0253bdc1a25b54f0783c0bb2a72239fdb05.zip |
AbstractPollTask: use timeout_add
-rw-r--r-- | 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 b4728e7a2..922905291 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) |