Enable faulthandler if config.debugger is set

steersbob · Jan 24, 2024 · f3db3a7 · f3db3a7
1 parent c83a433
commit f3db3a7
Show file tree

Hide file tree

Showing 4 changed files with 38 additions and 1 deletion.
diff --git a/brewblox_history/app_factory.py b/brewblox_history/app_factory.py
@@ -60,6 +60,14 @@ def create_app() -> FastAPI:
     config = utils.get_config()
     setup_logging(config.debug)
 
+    if config.debugger:  # pragma: no cover
+        import faulthandler
+        faulthandler.enable()
+
+        import debugpy
+        debugpy.listen(('0.0.0.0', 5678))
+        LOGGER.info('Debugger is enabled and listening on 5678')
+
     # Call setup functions for modules
     mqtt.setup()
     redis.setup()

diff --git a/brewblox_history/models.py b/brewblox_history/models.py
@@ -40,6 +40,7 @@ class ServiceConfig(BaseSettings):
 
     name: str = 'history'
     debug: bool = False
+    debugger: bool = False
 
     mqtt_protocol: Literal['mqtt', 'mqtts'] = 'mqtt'
     mqtt_host: str = 'eventbus'

diff --git a/poetry.lock b/poetry.lock
diff --git a/pyproject.toml b/pyproject.toml
@@ -19,6 +19,7 @@ pydantic-settings = "^2.1.0"
 fastapi-mqtt = "^2.0.0"
 httpx = "^0.25.2"
 websockets = "^12.0"
+debugpy = "^1.8.0"
 
 [tool.poetry.group.dev.dependencies]
 pytest-cov = "*"