mirror of
https://github.com/akuker/RASCSI.git
synced 2024-06-22 05:29:39 +00:00
089dc302e5
* python client library clean branch for PR. #455 * removed superfluous file. #455 * removed one more superfluous file. #455 * README.md, .pylintrc and pylint based fixes. #455 * updated wrt. to the review comments. #455 * removed pylint documentation duplication. #455
40 lines
940 B
Python
40 lines
940 B
Python
"""
|
|
Linux interrupt handling module
|
|
"""
|
|
import signal
|
|
|
|
|
|
class GracefulInterruptHandler:
|
|
"""
|
|
Class for handling Linux signal interrupts
|
|
"""
|
|
def __init__(self, signals=(signal.SIGINT, signal.SIGTERM)):
|
|
self.signals = signals
|
|
self.original_handlers = {}
|
|
self.interrupted = False
|
|
self.released = False
|
|
|
|
def __enter__(self):
|
|
for sig in self.signals:
|
|
self.original_handlers[sig] = signal.getsignal(sig)
|
|
signal.signal(sig, self.handler)
|
|
|
|
return self
|
|
|
|
def handler(self, signum, frame):
|
|
self.release()
|
|
self.interrupted = True
|
|
|
|
def __exit__(self, exception_type, exception_value, traceback):
|
|
self.release()
|
|
|
|
def release(self):
|
|
if self.released:
|
|
return False
|
|
|
|
for sig in self.signals:
|
|
signal.signal(sig, self.original_handlers[sig])
|
|
|
|
self.released = True
|
|
return True
|