fixup

rai_core_flask/rai_core_flask/flask_helper.py

@@ -10,6 +10,7 @@
 
 from flask import Flask
 from gevent.pywsgi import WSGIServer
+import signal
 
 from .environment_detector import build_environment
 from .environments.credentialed_vm_environment import CREDENTIALED_VM
@@ -62,6 +63,12 @@ def __init__(self, ip=None, port=None, with_credentials=False):
         self._thread = threading.Thread(target=self.run, daemon=True)
         self._thread.start()
 
+        # Closes server on program exit, including freeing all sockets
+        def closeserver():
+            self.stop()
+
+        atexit.register(closeserver)
+
     @staticmethod
     def _is_local_port_available(ip, port, raise_error=True):
         """Check whether the specified local port is available.
@@ -100,15 +107,8 @@ def run(self):
         logger.setLevel(logging.ERROR)
         self.server = WSGIServer((ip, self.port), self.app, log=logger)
         self.app.config["server"] = self.server
-        # self.app.config["CACHE_TYPE"] = "null"
         self.server.serve_forever()
 
-        # Closes server on program exit, including freeing all sockets
-        def closeserver():
-            self.stop()
-
-        atexit.register(closeserver)
-
     def stop(self):
-        if(self.server.started):
+        if (self.server.started):
             self.server.stop()