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