add debug toolbar to app middleware

michael7nightingale · Oct 19, 2023 · 176de5f · 176de5f
1 parent f7cda6e
commit 176de5f
Show file tree

Hide file tree

Showing 7 changed files with 15 additions and 2 deletions.
diff --git a/backend/requirements.txt b/backend/requirements.txt
@@ -24,6 +24,7 @@ ecdsa==0.18.0
 email-validator==2.0.0.post2
 fastapi==0.100.0
 fastapi-authtools==0.6
+fastapi-debug-toolbar==0.5.0
 flake8==6.1.0
 fonttools==4.42.0
 frozenlist==1.4.0
@@ -61,6 +62,7 @@ pydantic-extra-types==2.0.0
 pydantic-settings==2.0.2
 pydantic_core==2.1.2
 pyflakes==3.1.0
+pyinstrument==4.6.0
 pymongo==4.5.0
 pyparsing==3.0.9
 pypika-tortoise==0.1.6
@@ -77,6 +79,7 @@ requests==2.31.0
 rsa==4.9
 six==1.16.0
 sniffio==1.3.0
+sqlparse==0.4.4
 starlette==0.27.0
 sympy==1.12
 tomlkit==0.12.1

diff --git a/backend/src/core/middleware.py b/backend/src/core/middleware.py
@@ -1,7 +1,8 @@
 from fastapi import Request, Response, FastAPI
+from starlette.middleware.cors import CORSMiddleware
+from debug_toolbar.middleware import DebugToolbarMiddleware
 import time
 
-from starlette.middleware.cors import CORSMiddleware
 
 middleware_stack = {}
 
@@ -31,6 +32,10 @@ def register_middleware(app: FastAPI) -> None:
             app.middleware(type_)(func)
 
     origins = ["*"]
+    app.add_middleware(
+        DebugToolbarMiddleware,
+        panels=["debug_toolbar.panels.tortoise.TortoisePanel"],
+    )
     app.add_middleware(
         CORSMiddleware,
         allow_origins=origins,

diff --git a/backend/src/core/server.py b/backend/src/core/server.py
@@ -15,8 +15,8 @@
 class Server:
 
     def __init__(self):
-        self._app = FastAPI()
         self._settings = get_app_settings()
+        self._app = FastAPI(debug=self.settings.DEBUG)
 
         self._configurate_db()
         self._configurate_app()

diff --git a/backend/src/core/settings/base.py b/backend/src/core/settings/base.py
@@ -9,6 +9,7 @@ class AppEnvTypes(StrEnum):
 
 
 class BaseAppSettings(BaseSettings):
+    DEBUG: bool
 
     SUPERUSER_USERNAME: str
     SUPERUSER_PASSWORD: str

diff --git a/backend/src/core/settings/development.py b/backend/src/core/settings/development.py
@@ -8,6 +8,7 @@ class DevAppSettings(BaseAppSettings):
     DB_HOST: str
     DB_PORT: str
     DB_NAME: str
+    DEBUG: bool = True
 
     @property
     def db_uri(self) -> str:

diff --git a/backend/src/core/settings/production.py b/backend/src/core/settings/production.py
@@ -9,6 +9,8 @@ class ProdAppSettings(BaseAppSettings):
     DB_PORT: str
     DB_NAME: str
 
+    DEBUG: bool = False
+
     @property
     def db_uri(self) -> str:
         host_address = self.DB_HOST

diff --git a/backend/src/core/settings/test.py b/backend/src/core/settings/test.py
@@ -4,6 +4,7 @@
 class TestAppSettings(BaseAppSettings):
     DB_DRIVER: str
     DB_NAME: str
+    DEBUG: bool = True
 
     @property
     def db_uri(self) -> str: