diff options
author | Zac Medico <zmedico@gentoo.org> | 2011-01-15 19:43:35 -0800 |
---|---|---|
committer | Zac Medico <zmedico@gentoo.org> | 2011-01-15 19:43:35 -0800 |
commit | f707221b69ed913b7a1cb66c7ab45be2d1586f7c (patch) | |
tree | d45571c3576a17df3bbf50073544a0573599974c /pym/_emerge/Scheduler.py | |
parent | 3119eb0c7ae911e14f78624763cc79edfe953ad8 (diff) | |
download | portage-f707221b69ed913b7a1cb66c7ab45be2d1586f7c.tar.gz portage-f707221b69ed913b7a1cb66c7ab45be2d1586f7c.tar.bz2 portage-f707221b69ed913b7a1cb66c7ab45be2d1586f7c.zip |
Tweak PollScheduler signal handling.
Diffstat (limited to 'pym/_emerge/Scheduler.py')
-rw-r--r-- | pym/_emerge/Scheduler.py | 32 |
1 files changed, 29 insertions, 3 deletions
diff --git a/pym/_emerge/Scheduler.py b/pym/_emerge/Scheduler.py index ac48f853e..463d0d85e 100644 --- a/pym/_emerge/Scheduler.py +++ b/pym/_emerge/Scheduler.py @@ -7,6 +7,7 @@ import gc import gzip import logging import shutil +import signal import sys import tempfile import textwrap @@ -1166,10 +1167,35 @@ class Scheduler(PollScheduler): return rval while True: - rval = self._merge() - if self._terminated.is_set(): - return 1 + received_signal = [] + + def sighandler(signum, frame): + signal.signal(signal.SIGINT, signal.SIG_IGN) + signal.signal(signal.SIGTERM, signal.SIG_IGN) + portage.util.writemsg("\n\nExiting on signal %(signal)s\n" % \ + {"signal":signum}) + self.terminate() + received_signal.append(128 + signum) + + earlier_sigint_handler = signal.signal(signal.SIGINT, sighandler) + earlier_sigterm_handler = signal.signal(signal.SIGTERM, sighandler) + + try: + rval = self._merge() + finally: + # Restore previous handlers + if earlier_sigint_handler is not None: + signal.signal(signal.SIGINT, earlier_sigint_handler) + else: + signal.signal(signal.SIGINT, signal.SIG_DFL) + if earlier_sigterm_handler is not None: + signal.signal(signal.SIGTERM, earlier_sigterm_handler) + else: + signal.signal(signal.SIGTERM, signal.SIG_DFL) + + if received_signal: + sys.exit(received_signal[0]) if rval == os.EX_OK or fetchonly or not keep_going: break |