mirror of
https://github.com/python/cpython.git
synced 2025-07-21 02:05:20 +00:00
65b69a1093
svn+ssh://pythondev@svn.python.org/python/trunk ........ r79437 | michael.foord | 2010-03-26 03:18:31 +0000 (Fri, 26 Mar 2010) | 1 line Addition of -c command line option to unittest, to handle ctrl-c during a test run more elegantly ........
38 lines
1.1 KiB
Python
38 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)
|