🔧 Add env var to enable presentation definition OpenAPI extra

fundacionctic · Apr 16, 2024 · 5ce8aba · 5ce8aba
1 parent fbf1aa1
commit 5ce8aba
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 2 deletions.
diff --git a/mock-backend/docker-compose.yml b/mock-backend/docker-compose.yml
@@ -5,5 +5,7 @@ services:
     pull_policy: never
     command: ["uvicorn", "http-api:app", "--host", "0.0.0.0", "--port", "9090"]
     restart: on-failure
+    environment:
+      ENABLE_PRESENTATION_DEFINITION: ${ENABLE_PRESENTATION_DEFINITION:-}
     ports:
       - "9090:9090"
diff --git a/mock-backend/http-api.py b/mock-backend/http-api.py
@@ -1,5 +1,6 @@
 import asyncio
 import logging
+import os
 import pprint
 import random
 import uuid
@@ -82,10 +83,17 @@ def _build_presentation_definition() -> Dict[str, Any]:
     }
 
 
+def _get_openapi_extra() -> Dict[str, Any]:
+    if not bool(os.getenv("ENABLE_PRESENTATION_DEFINITION", False)):
+        return {}
+
+    return {_PRESENTATION_DEFINITION_EXT: _build_presentation_definition()}
+
+
 @app.post(
     "/consumption/prediction",
     tags=["Electricity consumption"],
-    # openapi_extra={_PRESENTATION_DEFINITION_EXT: _build_presentation_definition()},
+    openapi_extra=_get_openapi_extra(),
 )
 async def run_consumption_prediction(
     body: ElectricityConsumptionPredictionRequest,
@@ -112,7 +120,7 @@ async def run_consumption_prediction(
 @app.get(
     "/consumption",
     tags=["Electricity consumption"],
-    # openapi_extra={_PRESENTATION_DEFINITION_EXT: _build_presentation_definition()},
+    openapi_extra=_get_openapi_extra(),
 )
 async def get_consumption_data(
     location: str = "Asturias", day: date = None