39 lines
1.1 KiB
Python
39 lines
1.1 KiB
Python
|
import signal
|
||
|
|
import weakref
|
||
|
|
|
||
|
|
__unittest = True
|
||
|
|
|
||
|
|
|
||
|
|
class _InterruptHandler(object):
|
||
|
|
def __init__(self, default_handler):
|
||
|
|
self.called = False
|
||
|
|
self.default_handler = default_handler
|
||
|
|
|
||
|
|
def __call__(self, signum, frame):
|
||
|
|
installed_handler = signal.getsignal(signal.SIGINT)
|
||
|
|
if installed_handler is not self:
|
||
|
|
# if we aren't the installed handler, then delegate immediately
|
||
|
|
# to the default handler
|
||
|
|
self.default_handler(signum, frame)
|
||
|
|
|
||
|
|
if self.called:
|
||
|
|
self.default_handler(signum, frame)
|
||
|
|
self.called = True
|
||
|
|
for result in _results.keys():
|
||
|
|
result.stop()
|
||
|
|
|
||
|
|
_results = weakref.WeakKeyDictionary()
|
||
|
|
def registerResult(result):
|
||
|
|
_results[result] = 1
|
||
|
|
|
||
|
|
def removeResult(result):
|
||
|
|
return bool(_results.pop(result, None))
|
||
|
|
|
||
|
|
_interrupt_handler = None
|
||
|
|
def installHandler():
|
||
|
|
global _interrupt_handler
|
||
|
|
if _interrupt_handler is None:
|
||
|
|
default_handler = signal.getsignal(signal.SIGINT)
|
||
|
|
_interrupt_handler = _InterruptHandler(default_handler)
|
||
|
|
signal.signal(signal.SIGINT, _interrupt_handler)
|