metro endpoint as env var - issue noi-techpark#31

StefanoTavonatti · Sep 30, 2024 · a2a3c1a · a2a3c1a
1 parent 2fb42f4
commit a2a3c1a
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/pollution_v2/src/common/settings.py b/pollution_v2/src/common/settings.py
@@ -78,3 +78,5 @@
 TMP_DIR = f'{MAIN_DIR}/tmp'
 if not os.path.exists(TMP_DIR):
     os.makedirs(TMP_DIR)
+
+METRO_WS_PREDICTION_ENDPOINT = Variable.get("METRO_WS_PREDICTION_ENDPOINT", "http://metro:80/predict/?station_code=")
diff --git a/pollution_v2/src/road_weather/model/road_weather_model.py b/pollution_v2/src/road_weather/model/road_weather_model.py
@@ -16,7 +16,7 @@
 import urllib.request
 import mimetypes
 
-from common.settings import TMP_DIR
+from common.settings import TMP_DIR, METRO_WS_PREDICTION_ENDPOINT
 
 logger = logging.getLogger("pollution_v2.road_weather.model.road_weather_model")
 
@@ -73,8 +73,7 @@ def compute_data(self, observation: RoadWeatherObservationMeasureCollection,
                     f"{observation_filename} and forecast from {forecast_filename}")
         logger.info(f"Forecast start: {forecast_start}")
 
-        # TODO cablato
-        url = f"http://metro:80/predict/?station_code={station.wrf_code}"
+        url = f"{METRO_WS_PREDICTION_ENDPOINT}{station.wrf_code}"
 
         # List of files to upload
         files_to_upload = [forecast_filename, observation_filename]