pandaharvester/harvestermisc/signal_utils.py

0001 import os
0002 import signal
0003
0004
0005 # signal handler for suicide
0006 def suicide_handler(sig, frame):
0007     if os.getppid() == 1:
0008         os.killpg(os.getpgrp(), signal.SIGKILL)
0009     else:
0010         os.kill(os.getpid(), signal.SIGKILL)
0011
0012
0013 # set suicide handler
0014 def set_suicide_handler(signal_type=signal.SIGTERM):
0015     if signal_type is not None:
0016         signal.signal(signal_type, suicide_handler)
0017     else:
0018         signal.signal(signal.SIGINT, suicide_handler)
0019         signal.signal(signal.SIGHUP, suicide_handler)
0020         signal.signal(signal.SIGTERM, suicide_handler)
0021         signal.signal(signal.SIGALRM, suicide_handler)