trfSignal.py

Go to the documentation of this file.

# Copyright (C) 2002-2020 CERN for the benefit of the ATLAS collaboration
 
 
 
import signal
 
import logging
msg = logging.getLogger(__name__)
 
_savedSignalHandlerDict = {}
 
# N.B. Do not catch SIGINT (^C). This is delt with more cleanly via
# the stdTrfExceptionHandler wrapper which handles the KeyboardInterrupt exception.
# Also, don't intercept SIGTERM as this is used by the multiprocessing module to
# terminate child proceses
_defaultSignalList = ['SIGABRT', 'SIGFPE', 'SIGBUS', 'SIGHUP', 'SIGILL', 'SIGIO', 'SIGPIPE', 'SIGQUIT', 'SIGSEGV', 'SIGSYS', 'SIGXCPU', 'SIGXFSZ']
 
 
def setTrfSignalHandlers(handler):
    for s in _defaultSignalList:
        try:
            msg.debug("Setting signalhandler for %s to %s", s, handler)
            _savedSignalHandlerDict[s] =  signal.signal(getattr(signal, s), handler)
        except Exception as e:
            msg.error("Unable to attach custom signal handler to %s: %s", s, e)
            continue
 
 
def resetTrfSignalHandlers():
    for s in _defaultSignalList:
        try:
            signal.signal(getattr(signal, s), _savedSignalHandlerDict.get(s, signal.SIG_DFL))
        except Exception as e:
            msg.error("Unable to attach custom signal handler to %s: %s", s, e)
            continue