Added bootstrap command for observation series configurations

geobeyond · Jan 9, 2025 · 150b4ea · 150b4ea
1 parent 5797509
commit 150b4ea
Show file tree

Hide file tree

Showing 19 changed files with 959 additions and 800 deletions.
diff --git a/README.md b/README.md
@@ -236,22 +236,28 @@ dev environment is located at individual devs machine(s). In order to get a work
 Build the docker image by running this command:
 
 ```shell
-docker build --tag ghcr.io/geobeyond/arpav-ppcv-backend/arpav-ppcv-backend
+docker build \
+    --tag ghcr.io/geobeyond/arpav-ppcv-backend/arpav-ppcv-backend \
+    --file docker/Dockerfile \
+    .
 ```
 
 If you want to build an image for the current branch, such as when you added a new third-party dependency as part of
 an ongoing task, add the branch name to the build image:
 
 ```shell
-docker build --tag ghcr.io/geobeyond/arpav-ppcv-backend/arpav-ppcv-backend:$(git branch --show-current)
+docker build
+    --tag ghcr.io/geobeyond/arpav-ppcv-backend/arpav-ppcv-backend:$(git branch --show-current) \
+    --file docker/Dockerfile \
+    .
 ```
 
 In order to use this custom named image on your local development, set the `CURRENT_GIT_BRANCH` env variable before
 launching the docker compose stack, _i.e._:
 
 ```shell
 export CURRENT_GIT_BRANCH=$(git branch --show-current)
-docker compose -f docker/compose.yaml -f docker/compose.dev.yaml up -d
+docker compose -f docker/compose.yaml -f docker/compose.dev.yaml up -d --force-recreate
 ```
 
 

diff --git a/arpav_ppcv/bootstrapper/cliapp.py b/arpav_ppcv/bootstrapper/cliapp.py
@@ -60,6 +60,9 @@
     tasmin as tasmin_climatic_indicators,
     tr as tr_climatic_indicators,
 )
+from .observation_series_configurations import (
+    generate_observation_series_configurations,
+)
 
 from .configurationparameters import generate_configuration_parameters
 
@@ -250,6 +253,28 @@ def bootstrap_climatic_indicators(
                     session.rollback()
 
 
+@app.command("observation-series-configurations")
+def bootstrap_observation_series_configurations(ctx: typer.Context):
+    """Create initial observation series configurations."""
+    with sqlmodel.Session(ctx.obj["engine"]) as session:
+        all_climatic_indicators = database.collect_all_climatic_indicators(session)
+        to_generate = generate_observation_series_configurations(
+            {ind.identifier: ind for ind in all_climatic_indicators}
+        )
+        for obs_series_create in to_generate:
+            try:
+                db_series_conf = database.create_observation_series_configuration(
+                    session, obs_series_create
+                )
+                print(
+                    f"Created observation series "
+                    f"configuration {db_series_conf.identifier!r}"
+                )
+            except IntegrityError as err:
+                print(f"Could not create observation series {to_generate!r}: {err}")
+                session.rollback()
+
+
 @app.command("coverage-configuration-parameters")
 def bootstrap_coverage_configuration_parameters(
     ctx: typer.Context,

diff --git a/arpav_ppcv/bootstrapper/climaticindicators/cdds.py b/arpav_ppcv/bootstrapper/climaticindicators/cdds.py
@@ -1,8 +1,12 @@
 from ...schemas.static import (
     AggregationPeriod,
     MeasureType,
+    ObservationStationManager,
+)
+from ...schemas.climaticindicators import (
+    ClimaticIndicatorCreate,
+    ClimaticIndicatorObservationNameCreate,
 )
-from ...schemas.climaticindicators import ClimaticIndicatorCreate
 
 
 def generate_climatic_indicators() -> list[ClimaticIndicatorCreate]:
@@ -28,6 +32,12 @@ def generate_climatic_indicators() -> list[ClimaticIndicatorCreate]:
             color_scale_max=1000,
             data_precision=0,
             sort_order=8,
+            observation_names=[
+                ClimaticIndicatorObservationNameCreate(
+                    observation_station_manager=ObservationStationManager.ARPAV,
+                    indicator_observation_name="CDD_jrc",
+                )
+            ],
         ),
         ClimaticIndicatorCreate(
             name="cdds",

diff --git a/arpav_ppcv/bootstrapper/climaticindicators/fd.py b/arpav_ppcv/bootstrapper/climaticindicators/fd.py
@@ -1,8 +1,12 @@
 from ...schemas.static import (
     AggregationPeriod,
     MeasureType,
+    ObservationStationManager,
+)
+from ...schemas.climaticindicators import (
+    ClimaticIndicatorCreate,
+    ClimaticIndicatorObservationNameCreate,
 )
-from ...schemas.climaticindicators import ClimaticIndicatorCreate
 
 
 def generate_climatic_indicators() -> list[ClimaticIndicatorCreate]:
@@ -26,6 +30,12 @@ def generate_climatic_indicators() -> list[ClimaticIndicatorCreate]:
             color_scale_max=200,
             data_precision=0,
             sort_order=5,
+            observation_names=[
+                ClimaticIndicatorObservationNameCreate(
+                    observation_station_manager=ObservationStationManager.ARPAV,
+                    indicator_observation_name="FD",
+                )
+            ],
         ),
         ClimaticIndicatorCreate(
             name="fd",

diff --git a/arpav_ppcv/bootstrapper/climaticindicators/hdds.py b/arpav_ppcv/bootstrapper/climaticindicators/hdds.py
@@ -1,8 +1,12 @@
 from ...schemas.static import (
     AggregationPeriod,
     MeasureType,
+    ObservationStationManager,
+)
+from ...schemas.climaticindicators import (
+    ClimaticIndicatorCreate,
+    ClimaticIndicatorObservationNameCreate,
 )
-from ...schemas.climaticindicators import ClimaticIndicatorCreate
 
 
 def generate_climatic_indicators() -> list[ClimaticIndicatorCreate]:
@@ -28,6 +32,12 @@ def generate_climatic_indicators() -> list[ClimaticIndicatorCreate]:
             color_scale_max=7000,
             data_precision=0,
             sort_order=7,
+            observation_names=[
+                ClimaticIndicatorObservationNameCreate(
+                    observation_station_manager=ObservationStationManager.ARPAV,
+                    indicator_observation_name="HDD_it",
+                ),
+            ],
         ),
         ClimaticIndicatorCreate(
             name="hdds",

diff --git a/arpav_ppcv/bootstrapper/climaticindicators/pr.py b/arpav_ppcv/bootstrapper/climaticindicators/pr.py
@@ -1,8 +1,12 @@
 from ...schemas.static import (
     AggregationPeriod,
     MeasureType,
+    ObservationStationManager,
+)
+from ...schemas.climaticindicators import (
+    ClimaticIndicatorCreate,
+    ClimaticIndicatorObservationNameCreate,
 )
-from ...schemas.climaticindicators import ClimaticIndicatorCreate
 
 
 def generate_climatic_indicators() -> list[ClimaticIndicatorCreate]:
@@ -54,6 +58,12 @@ def generate_climatic_indicators() -> list[ClimaticIndicatorCreate]:
             color_scale_max=800,
             data_precision=0,
             sort_order=9,
+            observation_names=[
+                ClimaticIndicatorObservationNameCreate(
+                    observation_station_manager=ObservationStationManager.ARPAV,
+                    indicator_observation_name="PRCPTOT",
+                )
+            ],
         ),
         ClimaticIndicatorCreate(
             name="pr",

diff --git a/arpav_ppcv/bootstrapper/climaticindicators/su30.py b/arpav_ppcv/bootstrapper/climaticindicators/su30.py
@@ -1,8 +1,12 @@
 from ...schemas.static import (
     AggregationPeriod,
     MeasureType,
+    ObservationStationManager,
+)
+from ...schemas.climaticindicators import (
+    ClimaticIndicatorCreate,
+    ClimaticIndicatorObservationNameCreate,
 )
-from ...schemas.climaticindicators import ClimaticIndicatorCreate
 
 
 def generate_climatic_indicators() -> list[ClimaticIndicatorCreate]:
@@ -26,6 +30,12 @@ def generate_climatic_indicators() -> list[ClimaticIndicatorCreate]:
             color_scale_max=100,
             data_precision=0,
             sort_order=4,
+            observation_names=[
+                ClimaticIndicatorObservationNameCreate(
+                    observation_station_manager=ObservationStationManager.ARPAV,
+                    indicator_observation_name="SU30",
+                )
+            ],
         ),
         ClimaticIndicatorCreate(
             name="su30",

diff --git a/arpav_ppcv/bootstrapper/climaticindicators/tas.py b/arpav_ppcv/bootstrapper/climaticindicators/tas.py
@@ -1,8 +1,12 @@
 from ...schemas.static import (
     AggregationPeriod,
     MeasureType,
+    ObservationStationManager,
+)
+from ...schemas.climaticindicators import (
+    ClimaticIndicatorCreate,
+    ClimaticIndicatorObservationNameCreate,
 )
-from ...schemas.climaticindicators import ClimaticIndicatorCreate
 
 
 def generate_climatic_indicators() -> list[ClimaticIndicatorCreate]:
@@ -22,6 +26,12 @@ def generate_climatic_indicators() -> list[ClimaticIndicatorCreate]:
             color_scale_max=32,
             data_precision=1,
             sort_order=0,
+            observation_names=[
+                ClimaticIndicatorObservationNameCreate(
+                    observation_station_manager=ObservationStationManager.ARPAV,
+                    indicator_observation_name="TDd",
+                )
+            ],
         ),
         ClimaticIndicatorCreate(
             name="tas",

diff --git a/arpav_ppcv/bootstrapper/climaticindicators/tasmax.py b/arpav_ppcv/bootstrapper/climaticindicators/tasmax.py
@@ -1,8 +1,12 @@
 from ...schemas.static import (
     AggregationPeriod,
     MeasureType,
+    ObservationStationManager,
+)
+from ...schemas.climaticindicators import (
+    ClimaticIndicatorCreate,
+    ClimaticIndicatorObservationNameCreate,
 )
-from ...schemas.climaticindicators import ClimaticIndicatorCreate
 
 
 def generate_climatic_indicators() -> list[ClimaticIndicatorCreate]:
@@ -24,6 +28,12 @@ def generate_climatic_indicators() -> list[ClimaticIndicatorCreate]:
             color_scale_max=32,
             data_precision=1,
             sort_order=2,
+            observation_names=[
+                ClimaticIndicatorObservationNameCreate(
+                    observation_station_manager=ObservationStationManager.ARPAV,
+                    indicator_observation_name="TXd",
+                )
+            ],
         ),
         ClimaticIndicatorCreate(
             name="tasmax",

diff --git a/arpav_ppcv/bootstrapper/climaticindicators/tasmin.py b/arpav_ppcv/bootstrapper/climaticindicators/tasmin.py
@@ -1,8 +1,12 @@
 from ...schemas.static import (
     AggregationPeriod,
     MeasureType,
+    ObservationStationManager,
+)
+from ...schemas.climaticindicators import (
+    ClimaticIndicatorCreate,
+    ClimaticIndicatorObservationNameCreate,
 )
-from ...schemas.climaticindicators import ClimaticIndicatorCreate
 
 
 def generate_climatic_indicators() -> list[ClimaticIndicatorCreate]:
@@ -24,6 +28,12 @@ def generate_climatic_indicators() -> list[ClimaticIndicatorCreate]:
             color_scale_max=32,
             data_precision=1,
             sort_order=1,
+            observation_names=[
+                ClimaticIndicatorObservationNameCreate(
+                    observation_station_manager=ObservationStationManager.ARPAV,
+                    indicator_observation_name="TNd",
+                )
+            ],
         ),
         ClimaticIndicatorCreate(
             name="tasmin",

diff --git a/arpav_ppcv/bootstrapper/climaticindicators/tr.py b/arpav_ppcv/bootstrapper/climaticindicators/tr.py
@@ -1,8 +1,12 @@
 from ...schemas.static import (
     AggregationPeriod,
     MeasureType,
+    ObservationStationManager,
+)
+from ...schemas.climaticindicators import (
+    ClimaticIndicatorCreate,
+    ClimaticIndicatorObservationNameCreate,
 )
-from ...schemas.climaticindicators import ClimaticIndicatorCreate
 
 
 def generate_climatic_indicators() -> list[ClimaticIndicatorCreate]:
@@ -26,6 +30,12 @@ def generate_climatic_indicators() -> list[ClimaticIndicatorCreate]:
             color_scale_max=120,
             data_precision=1,
             sort_order=3,
+            observation_names=[
+                ClimaticIndicatorObservationNameCreate(
+                    observation_station_manager=ObservationStationManager.ARPAV,
+                    indicator_observation_name="TR",
+                )
+            ],
         ),
         ClimaticIndicatorCreate(
             name="tr",

diff --git a/arpav_ppcv/bootstrapper/observation_series_configurations.py b/arpav_ppcv/bootstrapper/observation_series_configurations.py
@@ -0,0 +1,78 @@
+from ..schemas.climaticindicators import ClimaticIndicator
+from ..schemas.observations import ObservationSeriesConfigurationCreate
+from ..schemas.static import (
+    MeasurementAggregationType,
+    ObservationStationManager,
+)
+
+
+def generate_observation_series_configurations(
+    indicators: dict[str, ClimaticIndicator],
+) -> list[ObservationSeriesConfigurationCreate]:
+    return [
+        ObservationSeriesConfigurationCreate(
+            climatic_indicator_id=indicators["hdds-absolute-annual"].id,
+            measurement_aggregation_type=MeasurementAggregationType.YEARLY,
+            station_managers=[ObservationStationManager.ARPAV],
+        ),
+        ObservationSeriesConfigurationCreate(
+            climatic_indicator_id=indicators["cdds-absolute-annual"].id,
+            measurement_aggregation_type=MeasurementAggregationType.YEARLY,
+            station_managers=[ObservationStationManager.ARPAV],
+        ),
+        ObservationSeriesConfigurationCreate(
+            climatic_indicator_id=indicators["tas-absolute-annual"].id,
+            measurement_aggregation_type=MeasurementAggregationType.SEASONAL,
+            station_managers=[ObservationStationManager.ARPAV],
+        ),
+        ObservationSeriesConfigurationCreate(
+            climatic_indicator_id=indicators["tas-absolute-annual"].id,
+            measurement_aggregation_type=MeasurementAggregationType.YEARLY,
+            station_managers=[ObservationStationManager.ARPAV],
+        ),
+        ObservationSeriesConfigurationCreate(
+            climatic_indicator_id=indicators["tasmax-absolute-annual"].id,
+            measurement_aggregation_type=MeasurementAggregationType.SEASONAL,
+            station_managers=[ObservationStationManager.ARPAV],
+        ),
+        ObservationSeriesConfigurationCreate(
+            climatic_indicator_id=indicators["tasmax-absolute-annual"].id,
+            measurement_aggregation_type=MeasurementAggregationType.YEARLY,
+            station_managers=[ObservationStationManager.ARPAV],
+        ),
+        ObservationSeriesConfigurationCreate(
+            climatic_indicator_id=indicators["tasmin-absolute-annual"].id,
+            measurement_aggregation_type=MeasurementAggregationType.SEASONAL,
+            station_managers=[ObservationStationManager.ARPAV],
+        ),
+        ObservationSeriesConfigurationCreate(
+            climatic_indicator_id=indicators["tasmin-absolute-annual"].id,
+            measurement_aggregation_type=MeasurementAggregationType.YEARLY,
+            station_managers=[ObservationStationManager.ARPAV],
+        ),
+        ObservationSeriesConfigurationCreate(
+            climatic_indicator_id=indicators["pr-absolute-annual"].id,
+            measurement_aggregation_type=MeasurementAggregationType.SEASONAL,
+            station_managers=[ObservationStationManager.ARPAV],
+        ),
+        ObservationSeriesConfigurationCreate(
+            climatic_indicator_id=indicators["pr-absolute-annual"].id,
+            measurement_aggregation_type=MeasurementAggregationType.YEARLY,
+            station_managers=[ObservationStationManager.ARPAV],
+        ),
+        ObservationSeriesConfigurationCreate(
+            climatic_indicator_id=indicators["tr-absolute-annual"].id,
+            measurement_aggregation_type=MeasurementAggregationType.YEARLY,
+            station_managers=[ObservationStationManager.ARPAV],
+        ),
+        ObservationSeriesConfigurationCreate(
+            climatic_indicator_id=indicators["su30-absolute-annual"].id,
+            measurement_aggregation_type=MeasurementAggregationType.YEARLY,
+            station_managers=[ObservationStationManager.ARPAV],
+        ),
+        ObservationSeriesConfigurationCreate(
+            climatic_indicator_id=indicators["fd-absolute-annual"].id,
+            measurement_aggregation_type=MeasurementAggregationType.YEARLY,
+            station_managers=[ObservationStationManager.ARPAV],
+        ),
+    ]