Merge pull request #105 from jgranstrom/sigterm-fix

Gracefully shutdown server on SIGTERM in addition to SIGINT
RasaHQ · Jan 16, 2017 · 63452b7 · 63452b7
2 parents e930f51 + b12259b
commit 63452b7
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/src/server.py b/src/server.py
@@ -6,6 +6,7 @@
 import glob
 import warnings
 import logging
+import signal
 from BaseHTTPServer import BaseHTTPRequestHandler, HTTPServer
 from rasa_nlu.train import do_train
 from rasa_nlu.config import RasaNLUConfig
@@ -244,7 +245,12 @@ def create_argparser():
     logging.captureWarnings(True)
     logging.debug(config.view())
     try:
+        def stop(signal_number, frame):
+            raise KeyboardInterrupt()
+
+        signal.signal(signal.SIGTERM, stop)
         server = RasaNLUServer(config)
         server.start()
+
     except KeyboardInterrupt:
         server.stop()