Trap sigint maupun sigterm

Adhidarma Hadiwinoto
1 parent 961bfbde0c
Showing 1 changed file with 5 additions and 2 deletions Side-by-side Diff
main.py
@@ -72,8 +72,8 @@ redis_client = redis.StrictRedis(host=REDIS_HOST, port=REDIS_PORT, db=0)
 def signalHandler(signum, frame):
     global TERMINATING
  
-    if (signum == signal.SIGINT):
-        logger.info('SIGINT received, waiting for next loop to terminate. Terminating...')
+    if signum == signal.SIGINT or signum == signal.SIGTERM:
+        logger.info('Signal ({0}) received, waiting for next loop to terminate. Terminating...'.format(signum))
         TERMINATING = True
  
 def handleSms(sms):
@@ -728,6 +728,9 @@ if __name__ == &#39;__main__&#39;:
     pidfile.write(str(getpid()))
     pidfile.close()
  
+    # trap CTRL-C
     signal.signal(signal.SIGINT, signalHandler)
+    # trap supervisor stop
+    signal.signal(signal.SIGTERM, signalHandler)
  
     main()
...	...	@@ -72,8 +72,8 @@ redis_client = redis.StrictRedis(host=REDIS_HOST, port=REDIS_PORT, db=0)
72	72	def signalHandler(signum, frame):
73	73	global TERMINATING
74	74
75		- if (signum == signal.SIGINT):
76		- logger.info('SIGINT received, waiting for next loop to terminate. Terminating...')
	75	+ if signum == signal.SIGINT or signum == signal.SIGTERM:
	76	+ logger.info('Signal ({0}) received, waiting for next loop to terminate. Terminating...'.format(signum))
77	77	TERMINATING = True
78	78
79	79	def handleSms(sms):
...	...	@@ -728,6 +728,9 @@ if __name__ == '__main__':
728	728	pidfile.write(str(getpid()))
729	729	pidfile.close()
730	730
	731	+ # trap CTRL-C
731	732	signal.signal(signal.SIGINT, signalHandler)
	733	+ # trap supervisor stop
	734	+ signal.signal(signal.SIGTERM, signalHandler)
732	735
733	736	main()