diff --git a/ddj_cloud/scrapers/talsperren/common.py b/ddj_cloud/scrapers/talsperren/common.py
index 3a0ca5d..e2ea45d 100644
--- 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-
talsperre",
- "Genkeltalsperre": "Genkel-
talsperre",
- "Wiehltalsperre": "Wiehl-
talsperre",
- "Talsperren Haltern und Hullern": "Talsperren
Haltern und
Hullern",
- "Biggetalsperre": "Bigge-
talsperre",
- "Ennepetalsperre": "Ennepe-
talsperre",
- "Fürwiggetalsperre": "Fürwigge-
talsperre",
- "Hennetalsperre": "Henne-
talsperre",
- "Listertalsperre": "Lister-
talsperre",
- "Möhnetalsperre": "Möhne-
talsperre",
- "Sorpetalsperre": "Sorpe-
talsperre",
- "Stausee Ahausen": "Stausee
Ahausen",
- "Versetalsperre": "Verse-
talsperre",
- "Wahnbachtalsperre": "Wahnbach-
talsperre",
- "Oleftalsperre": "Olef-
talsperre",
+ "Oleftalsperre": "Oleftalsperre",
"Rurtalsperre Hauptsee": "Rurtalsperre
Hauptsee",
"Rurtalsperre Obersee": "Rurtalsperre
Obersee",
- "Stauanlage Heimbach": "Stauanlage
Heimbach",
- "Stauanlage Obermaubach": "Stauanlage
Obermaubach",
- "Urfttalsperre": "Urft-
talsperre",
+ "Stauanlage Heimbach": "Stausee
Heimbach",
+ "Stauanlage Obermaubach": "Stausee
Obermaubach",
+ "Urfttalsperre": "Urfttalsperre",
"Wehebachtalsperre": "Wehebach-
talsperre",
- "Bevertalsperre": "Bever-
talsperre",
- "Bruchertalsperre": "Brucher-
talsperre",
- "Eschbachtalsperre": "Eschbach-
talsperre",
- "Große Dhünntalsperre": "Große
Dhünn-
talsperre",
- "Herbringhauser Talsperre": "Herbringhauser
Talsperre",
- "Kerspetalsperre": "Kerspe-
talsperre",
- "Lingesetalsperre": "Lingese-
talsperre",
- "Neyetalsperre": "Neye-
talsperre",
- "Panzertalsperre": "Panzer-
talsperre",
- "Ronsdorfer Talsperre": "Ronsdorfer
Talsperre",
- "Schevelinger Talsperre": "Schevelinger
Talsperre",
- "Stausee Beyenburg": "Stausee
Beyenburg",
- "Vorsperre Große Dhünn": "Vorsperre
Große Dhünn",
- "Wuppertalsperre": "Wupper-
talsperre",
}
diff --git a/ddj_cloud/scrapers/talsperren/exporters/map.py b/ddj_cloud/scrapers/talsperren/exporters/map.py
index dbc86b3..9381110 100644
--- 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