feat: Add setting for minimal degrees search distance for weather sta…

…tion (#21) * define default closest weather stations * refactoring * update settings for default information * refactoring --------- Signed-off-by: Nicolas Höning <[email protected]> Co-authored-by: Nicolas Höning <[email protected]>
SeitaBV · Jul 26, 2023 · e7edf6e · e7edf6e
1 parent e83466e
commit e7edf6e
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 2 deletions.
diff --git a/flexmeasures_openweathermap/__init__.py b/flexmeasures_openweathermap/__init__.py
@@ -29,6 +29,7 @@
 DEFAULT_DATA_SOURCE_NAME = "OpenWeatherMap"
 DEFAULT_WEATHER_STATION_NAME = "weather station (created by FM-OWM)"
 WEATHER_STATION_TYPE_NAME = "weather station"
+DEFAULT_MAXIMAL_DEGREE_LOCATION_DISTANCE = 1
 
 __version__ = "0.1"
 __settings__ = {
@@ -48,6 +49,10 @@
         description=f"Name of the weather station asset, defaults to '{DEFAULT_WEATHER_STATION_NAME}'",
         level="debug",
     ),
+    "OPENWEATHERMAP_MAXIMAL_DEGREE_LOCATION_DISTANCE": dict(
+        descripion=f"Maximum distance (in degrees latitude & longitude) for weather stations from forecast location, defaults to {DEFAULT_MAXIMAL_DEGREE_LOCATION_DISTANCE}",
+        level="debug",
+    ),
 }
 
 # CLI

diff --git a/flexmeasures_openweathermap/utils/owm.py b/flexmeasures_openweathermap/utils/owm.py
@@ -14,6 +14,7 @@
 from flexmeasures.data.models.time_series import Sensor, TimedBelief
 from flexmeasures.data.utils import save_to_db
 
+from flexmeasures_openweathermap import DEFAULT_MAXIMAL_DEGREE_LOCATION_DISTANCE
 from .locating import find_weather_sensor_by_location
 from ..sensor_specs import mapping
 from .modeling import (
@@ -72,15 +73,17 @@ def call_openweatherapi(
 def save_forecasts_in_db(
     api_key: str,
     locations: List[Tuple[float, float]],
-    max_degree_difference_for_nearest_weather_sensor: int = 2,
 ):
     """Process the response from OpenWeatherMap API into timed beliefs.
     Collects all forecasts for all locations and all sensors at all locations, then bulk-saves them.
     """
     click.echo("[FLEXMEASURES-OWM] Getting weather forecasts:")
     click.echo("[FLEXMEASURES-OWM] Latitude, Longitude")
     click.echo("[FLEXMEASURES-OWM] -----------------------")
-
+    max_degree_difference_for_nearest_weather_sensor = current_app.config.get(
+        "OPENWEATHERMAP_MAXIMAL_DEGREE_LOCATION_DISTANCE",
+        DEFAULT_MAXIMAL_DEGREE_LOCATION_DISTANCE,
+    )
     for location in locations:
         click.echo("[FLEXMEASURES] %s, %s" % location)
         weather_sensors: Dict[