SimBenchConverter

Signed-off-by: Bram Stoeller <bram.stoeller@alliander.com>

Author: bramstoeller

src/power_grid_model_io/config/csv/sim_bench.yaml

@@ -0,0 +1,25 @@
+# SPDX-FileCopyrightText: 2022 Contributors to the Power Grid Model project <dynamic.grid.calculation@alliander.com>
+#
+# SPDX-License-Identifier: MPL-2.0
+---
+grid:
+  Node:
+    node:
+      id:
+        auto_id:
+          key: id
+  Line:
+    line:
+      id:
+        auto_id:
+          key: id
+      from_node:
+        auto_id:
+          table: Node
+          key:
+            id: nodeA
+      to_node:
+        auto_id:
+          table: Node
+          key:
+            id: nodeB

src/power_grid_model_io/converters/sim_bench_converter.py

@@ -0,0 +1,51 @@
+# SPDX-FileCopyrightText: 2022 Contributors to the Power Grid Model project <dynamic.grid.calculation@alliander.com>
+#
+# SPDX-License-Identifier: MPL-2.0
+"""
+Sim Bench Converter: Download, extract and convert a sim bench dataset to PGM
+"""
+
+from pathlib import Path
+from tempfile import gettempdir
+from typing import Optional
+
+from power_grid_model_io.converters.tabular_converter import TabularConverter
+from power_grid_model_io.data_stores.csv_dir_store import CsvDirStore
+from power_grid_model_io.data_types import TabularData
+from power_grid_model_io.utils.download import download_and_extract
+
+DEFAULT_MAPPING_FILE = Path(__file__).parent.parent / "config" / "csv" / "sim_bench.yaml"
+DEFAULT_DOWNLOAD_URL = "http://141.51.193.167/simbench/gui/usecase/download/?simbench_code={simbench_code:s}&format=csv"
+
+
+class SimBenchConverter(TabularConverter):
+    """
+    Sim Bench Converter: Download, extract and convert a sim bench dataset to PGM
+    """
+
+    __slots__ = ("_simbench_code", "_download_url", "_download_dir")
+
+    def __init__(self, simbench_code: Optional[str] = None, download_dir: Optional[Path] = None):
+        super().__init__(mapping_file=DEFAULT_MAPPING_FILE)
+        self.simbench_code: Optional[str] = simbench_code
+        self._download_url: Optional[str] = None
+        self._download_dir: Path = download_dir or Path(gettempdir())
+        if simbench_code is not None:
+            self.set_download_url(DEFAULT_DOWNLOAD_URL.format(simbench_code=simbench_code))
+
+    def set_download_url(self, url: str):
+        """
+        Use a custom sim bench URL
+        """
+        self._download_url = url
+
+    def _load_data(self, data: Optional[TabularData]) -> TabularData:
+        if data is None and self._source is None and self._download_url is not None:
+            try:
+                csv_dir = download_and_extract(self._download_url, dir_path=self._download_dir)
+            except ValueError as ex:
+                if str(ex).endswith(".download"):
+                    raise ValueError(f"Invalid SimBench dataset URL: {self._download_url}") from None
+                raise ex
+            self._source = CsvDirStore(csv_dir, delimiter=";")
+        return super()._load_data(data=data)