From 6858832c23227b59324ea82ff47660dcdca76d42 Mon Sep 17 00:00:00 2001
From: Brian Broll <brian.broll@gmail.com>
Date: Sat, 9 Mar 2019 14:32:15 -0800
Subject: [PATCH] Log signals received by job process. Closes #1207

---
 src/plugins/GenerateJob/templates/main.ejs | 10 ++++++++++
 1 file changed, 10 insertions(+)

diff --git a/src/plugins/GenerateJob/templates/main.ejs b/src/plugins/GenerateJob/templates/main.ejs
index 19bedb605..1ed28d284 100644
--- a/src/plugins/GenerateJob/templates/main.ejs
+++ b/src/plugins/GenerateJob/templates/main.ejs
@@ -1,5 +1,15 @@
 import json
 import deepforge
+import signal
+import sys
+
+def signal_handler(signum, frame):
+    sig = signal.Signals(signum)
+    print('Received ' + sig.name + '. Exiting.')
+    sys.exit(1)
+
+signal.signal(signal.SIGINT, signal_handler)
+signal.signal(signal.SIGTERM, signal_handler)
 
 <%= initCode %>
 # input data<% inputs.forEach(function(pair) { var input = pair[0], type = pair[1], isNil = pair[2];%>