📊 Revise renames for reservoirs

wdr-data · May 28, 2024 · 0c9cf88 · 0c9cf88
1 parent 2378e1f
commit 0c9cf88
Show file tree

Hide file tree

Showing 2 changed files with 20 additions and 36 deletions.
diff --git a/ddj_cloud/scrapers/talsperren/common.py b/ddj_cloud/scrapers/talsperren/common.py
@@ -30,43 +30,19 @@
     "Gelsenwasser",
 ]
 
+RESERVOIR_RENAMES = {
+    "Stauanlage Heimbach": "Stausee Heimbach",
+    "Stauanlage Obermaubach": "Stausee Obermaubach",
+}
 
 RESERVOIR_RENAMES_BREAKS = {
-    "Aggertalsperre": "Agger-<br>talsperre",
-    "Genkeltalsperre": "Genkel-<br>talsperre",
-    "Wiehltalsperre": "Wiehl-<br>talsperre",
-    "Talsperren Haltern und Hullern": "Talsperren<br>Haltern und<br>Hullern",
-    "Biggetalsperre": "Bigge-<br>talsperre",
-    "Ennepetalsperre": "Ennepe-<br>talsperre",
-    "Fürwiggetalsperre": "Fürwigge-<br>talsperre",
-    "Hennetalsperre": "Henne-<br>talsperre",
-    "Listertalsperre": "Lister-<br>talsperre",
-    "Möhnetalsperre": "Möhne-<br>talsperre",
-    "Sorpetalsperre": "Sorpe-<br>talsperre",
-    "Stausee Ahausen": "Stausee<br>Ahausen",
-    "Versetalsperre": "Verse-<br>talsperre",
-    "Wahnbachtalsperre": "Wahnbach-<br>talsperre",
-    "Oleftalsperre": "Olef-<br>talsperre",
+    "Oleftalsperre": "Oleftalsperre",
     "Rurtalsperre Hauptsee": "Rurtalsperre<br>Hauptsee",
     "Rurtalsperre Obersee": "Rurtalsperre<br>Obersee",
-    "Stauanlage Heimbach": "Stauanlage<br>Heimbach",
-    "Stauanlage Obermaubach": "Stauanlage<br>Obermaubach",
-    "Urfttalsperre": "Urft-<br>talsperre",
+    "Stauanlage Heimbach": "Stausee<br>Heimbach",
+    "Stauanlage Obermaubach": "Stausee<br>Obermaubach",
+    "Urfttalsperre": "Urfttalsperre",
     "Wehebachtalsperre": "Wehebach-<br>talsperre",
-    "Bevertalsperre": "Bever-<br>talsperre",
-    "Bruchertalsperre": "Brucher-<br>talsperre",
-    "Eschbachtalsperre": "Eschbach-<br>talsperre",
-    "Große Dhünntalsperre": "Große<br>Dhünn-<br>talsperre",
-    "Herbringhauser Talsperre": "Herbringhauser<br>Talsperre",
-    "Kerspetalsperre": "Kerspe-<br>talsperre",
-    "Lingesetalsperre": "Lingese-<br>talsperre",
-    "Neyetalsperre": "Neye-<br>talsperre",
-    "Panzertalsperre": "Panzer-<br>talsperre",
-    "Ronsdorfer Talsperre": "Ronsdorfer<br>Talsperre",
-    "Schevelinger Talsperre": "Schevelinger<br>Talsperre",
-    "Stausee Beyenburg": "Stausee<br>Beyenburg",
-    "Vorsperre Große Dhünn": "Vorsperre<br>Große Dhünn",
-    "Wuppertalsperre": "Wupper-<br>talsperre",
 }
 
 

diff --git a/ddj_cloud/scrapers/talsperren/exporters/map.py b/ddj_cloud/scrapers/talsperren/exporters/map.py
@@ -7,6 +7,7 @@
 from ddj_cloud.scrapers.talsperren.common import (
     Exporter,
     FEDERATION_RENAMES,
+    RESERVOIR_RENAMES,
     RESERVOIR_RENAMES_BREAKS,
 )
 from ddj_cloud.scrapers.talsperren.federations.agger import AggerFederation
@@ -181,7 +182,7 @@ def _add_marker_size(self, df_map: pd.DataFrame) -> pd.DataFrame:
         )
         return df_map
 
-    def run(self, df_base: pd.DataFrame) -> pd.DataFrame:
+    def run(self, df_base: pd.DataFrame, do_reservoir_rename: bool = True) -> pd.DataFrame:
         df_base.insert(0, "id", df_base["federation_name"] + "_" + df_base["name"])
 
         # Gernerate map with latest data
@@ -207,10 +208,17 @@ def run(self, df_base: pd.DataFrame) -> pd.DataFrame:
         df_map = df_map.round(5)
 
         # Rename federation names
-        df_map["federation_name"] = df_map["federation_name"].apply(
-            lambda x: FEDERATION_RENAMES.get(x, x)
+        df_map["federation_name"].replace(
+            FEDERATION_RENAMES,
+            inplace=True,
         )
 
+        if do_reservoir_rename:
+            df_map["name"].replace(
+                RESERVOIR_RENAMES,
+                inplace=True,
+            )
+
         return df_map
 
 
@@ -234,7 +242,7 @@ class FilteredMapExporter(MapExporter):
         filename = f"filtered_map_{slugify('_'.join(federation_names))}"
 
         def run(self, df_base: pd.DataFrame) -> pd.DataFrame:
-            df_map = super().run(df_base)
+            df_map = super().run(df_base, do_reservoir_rename=False)
 
             translated_names = [
                 FEDERATION_RENAMES.get(fed_name, fed_name) for fed_name in federation_names