graceful_int_handler.py

class InterruptableRegion(object):
    
    def __init__(self, signum_list=None):
        if signum_list is None:
            signum_list = [signal.SIGINT]
        self.signum_list = signum_list
        self.reset()
        
    def __enter__(self):
        self.reset()
        def handler(signum, frame):
            self.release()
            self.interrupted = True
        for signum in self.signum_list:
            self.original_handler_map[signum] = signal.signal(signum, handler)
        return self
        
    def __exit__(self, type, value, tb):
        self.release()
    
    def reset(self):
        self.interrupted = False
        self.released = False
        self.original_handler_map = {}
        
    def release(self):
        if self.released:
            return
        for signum, handler in self.original_handler_map.items():
            signal.signal(signum, handler)
        self.released = True