Basic python server working including hot-reload

yacv_server/server.py

@@ -0,0 +1,69 @@
+import asyncio
+import atexit
+import os
+import signal
+from threading import Thread
+from typing import Optional
+
+from OCP.TopoDS import TopoDS_Shape
+from aiohttp import web
+
+FRONTEND_BASE_PATH = os.getenv('FRONTEND_BASE_PATH', '../dist')
+
+
+# noinspection PyUnusedLocal
+async def _index_handler(request: web.Request) -> web.Response:
+    return web.HTTPTemporaryRedirect(location='index.html')
+
+
+class Server:
+    app = web.Application()
+    runner: web.AppRunner
+    thread: Optional[Thread] = None
+
+    def __init__(self, *args, **kwargs):
+        # --- Routes ---
+        # - API
+        # self.app.router.add_route({'POST','GET'}, '/api/{collection}', api_handler)
+        # - Static files from the frontend
+        self.app.router.add_get('/{path:(.*/|)}', _index_handler)  # Any folder -> index.html
+        self.app.router.add_static('/', path=FRONTEND_BASE_PATH, name='static_frontend')
+        # --- Misc ---
+        self.runner = web.AppRunner(self.app)
+
+    def start(self):
+        """Starts the web server in the background"""
+        assert self.thread is None, "Server already started"
+        self.thread = Thread(target=self.run_server, name='yacv_server', daemon=True)
+        signal.signal(signal.SIGINT | signal.SIGTERM, self.stop)
+        atexit.register(self.stop)
+        self.thread.start()
+
+    # noinspection PyUnusedLocal
+    def stop(self, *args):
+        """Stops the web server"""
+        print('Stopping server...')
+        if self.thread is None:
+            print('Cannot stop server because it is not running')
+            return
+        asyncio.run(self.runner.shutdown())
+        asyncio.run(self.app.cleanup())
+        self.thread = None  # FIXME: Not properly cleaned up (join blocks forever)
+        print('Cleanup done')
+
+    def run_server(self):
+        """Runs the web server"""
+        loop = asyncio.new_event_loop()
+        asyncio.set_event_loop(loop)
+        loop.run_until_complete(self.runner.setup())
+        site = web.TCPSite(self.runner, os.getenv('YACV_HOST', 'localhost'), int(os.getenv('YACV_PORT', 32323)))
+        loop.run_until_complete(site.start())
+        loop.run_forever()
+
+    def show_object(self, obj: TopoDS_Shape):
+        pass
+
+
+def get_app() -> web.Application:
+    """Required by aiohttp-devtools"""
+    return Server().app