Subclass app to fix typing error

rwb27 · rwb27 · commit 82d2232339a7 · 2024-08-05T18:02:11.000+01:00
I've subclassed FastAPI to make it easier to attch LabThings error info to the fallback server.
diff --git a/src/labthings_fastapi/server/fallback.py b/src/labthings_fastapi/server/fallback.py
@@ -3,7 +3,16 @@
 from fastapi.responses import HTMLResponse
 from starlette.responses import RedirectResponse
 
-app = FastAPI()
+
+class FallbackApp(FastAPI):
+    def __init__(self, *args, **kwargs):
+        super().__init__(*args, **kwargs)
+        self.labthings_config = None
+        self.labthings_server = None
+        self.labthings_error = None
+
+
+app = FallbackApp()
 
 ERROR_PAGE = """
 <!DOCTYPE html>
@@ -27,10 +36,6 @@
 </html>
 """
 
-app.labthings_config = None
-app.labthings_server = None
-app.labthings_error = None
-
 
 @app.get("/")
 async def root():
