Gracefully shutdown on SIGTERM to support docker environment and dock…

…er stop
RasaHQ · Jan 13, 2017 · b12259b · b12259b
1 parent 7e2b33f
commit b12259b
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()