pydra-core 0.0.1__py2.py3-none-any.whl → 0.0.3__py2.py3-none-any.whl

pydra_core/io/database_settings.py

@@ -1,183 +1,183 @@
-import os
-import pandas as pd
-import sqlite3
-
-from ..common.enum import WaterSystem
-
-
-class DatabaseSettings:
-    """
-    Settings database
-    """
-
-    def __init__(self) -> None:
-        self.con = None
-
-    def __enter__(self) -> "DatabaseSettings":
-        # Init the connection
-        self.con = sqlite3.connect(
-            os.path.join(
-                os.path.split(os.path.dirname(__file__))[0],
-                "data",
-                "settings",
-                "calculation_settings.sqlite",
-            )
-        )
-        return self
-
-    def __exit__(self, exc_type, exc_val, exc_tb) -> None:
-        # Close the connection
-        self.con.close()
-
-    def get_settings(self, watersystem: WaterSystem) -> pd.DataFrame:
-        """
-        Return all settings from the database
-
-        Parameters
-        ----------
-        watersystem : WaterSystem
-            WaterSystem to get the settings for
-
-        Returns
-        -------
-        pd.DataFrame
-            A DataFrame with settings
-        """
-        # Query
-        sql = f"SELECT SettingName, SettingValue FROM CalculationSettings WHERE WaterSystem = {watersystem.value}"
-        settings = pd.read_sql(sql, self.con)
-
-        # Parse to int and float if possible
-        for n, row in settings.iterrows():
-            settings.loc[n, "SettingValue"] = self.__parse_value(row["SettingValue"])
-
-        # Return the settings
-        return settings
-
-    def get_sea_level_statistic_points(self, watersystem: WaterSystem) -> pd.DataFrame:
-        """
-        Return the reference points for sea level statistics
-
-        Parameters
-        ----------
-        watersystem : WaterSystem
-            Watersystem for which to return the reference points
-
-        Returns
-        -------
-        pd.DataFrame
-            All reference points
-        """
-        # Obtain the right table name
-        if watersystem in [
-            WaterSystem.COAST_SOUTH,
-            WaterSystem.COAST_CENTRAL,
-            WaterSystem.COAST_NORTH,
-        ]:
-            tablename = "CoastReferencePoints"
-        elif watersystem in [WaterSystem.WADDEN_SEA_WEST, WaterSystem.WADDEN_SEA_EAST]:
-            tablename = "WaddenSeaReferencePoints"
-        else:
-            raise NotImplementedError(
-                f"[ERROR] No sea level reference points implemented for {watersystem}."
-            )
-
-        # SQL
-        data = pd.read_sql(
-            f"SELECT Name, X, Y FROM {tablename}", self.con, index_col="Name"
-        )
-
-        # Return data
-        return data
-
-    def get_sea_level_sub_systems(self, watersystem: WaterSystem) -> pd.DataFrame:
-        """
-        Return all subsystems of the coast or Wadden sea.
-
-        Parameters
-        ----------
-        watersystem : WaterSystem
-            Watersystem for which to return the reference points
-
-        Returns
-        -------
-        pd.DataFrame
-            All subsystems
-        """
-        # Obtain the right table name
-        if watersystem in [
-            WaterSystem.COAST_SOUTH,
-            WaterSystem.COAST_CENTRAL,
-            WaterSystem.COAST_NORTH,
-        ]:
-            tablename = "CoastSubSystems"
-        elif watersystem in [WaterSystem.WADDEN_SEA_WEST, WaterSystem.WADDEN_SEA_EAST]:
-            tablename = "WaddenSeaSubSystems"
-        else:
-            raise NotImplementedError(
-                f"[ERROR] No sea level reference points implemented for {watersystem}."
-            )
-
-        # SQL
-        data = pd.read_sql(
-            f"SELECT SubSystemId, Point1, Point2, Point3 FROM {tablename}",
-            self.con,
-            index_col="SubSystemId",
-        )
-
-        # Return data
-        return data
-
-    def get_input_variable_ids(self) -> dict:
-        """
-        Return a dictionary with the input variable ids (ivid) and symbols (isymbol)
-
-        Returns
-        -------
-        dict
-            A dictionary with {ivid : isymbol}
-        """
-        # Query
-        sql = "SELECT InputVariableId, InputVariableSymbol FROM InputVariables"
-        data = self.con.execute(sql).fetchall()
-
-        # Process data
-        results = {data[i][0]: data[i][1] for i in range(len(data))}
-
-        # Return the dictionary
-        return results
-
-    def get_result_variable_ids(self) -> dict:
-        """
-        Return a dictionary with the result variable ids (rvid) and symbols (rsymbol)
-
-        Returns
-        -------
-        dict
-            A dictionary with {rvid : rsymbol}
-        """
-        # Query
-        sql = "SELECT ResultVariableId, ResultVariableSymbol FROM ResultVariables"
-        data = self.con.execute(sql).fetchall()
-
-        # Process data
-        results = {data[i][0]: data[i][1] for i in range(len(data))}
-
-        # Return the dictionary
-        return results
-
-    def __parse_value(self, value):
-        # Parse value from string
-        try:
-            return float(value)
-        except ValueError:
-            pass
-        try:
-            return int(value)
-        except ValueError:
-            pass
-        if value.lower() == "true":
-            return True
-        elif value.lower() == "false":
-            return False
-        return value
+import os
+import pandas as pd
+import sqlite3
+
+from ..common.enum import WaterSystem
+
+
+class DatabaseSettings:
+    """
+    Settings database
+    """
+
+    def __init__(self) -> None:
+        self.con = None
+
+    def __enter__(self) -> "DatabaseSettings":
+        # Init the connection
+        self.con = sqlite3.connect(
+            os.path.join(
+                os.path.split(os.path.dirname(__file__))[0],
+                "data",
+                "settings",
+                "calculation_settings.sqlite",
+            )
+        )
+        return self
+
+    def __exit__(self, exc_type, exc_val, exc_tb) -> None:
+        # Close the connection
+        self.con.close()
+
+    def get_settings(self, watersystem: WaterSystem) -> pd.DataFrame:
+        """
+        Return all settings from the database
+
+        Parameters
+        ----------
+        watersystem : WaterSystem
+            WaterSystem to get the settings for
+
+        Returns
+        -------
+        pd.DataFrame
+            A DataFrame with settings
+        """
+        # Query
+        sql = f"SELECT SettingName, SettingValue FROM CalculationSettings WHERE WaterSystem = {watersystem.value}"
+        settings = pd.read_sql(sql, self.con)
+
+        # Parse to int and float if possible
+        for n, row in settings.iterrows():
+            settings.loc[n, "SettingValue"] = self.__parse_value(row["SettingValue"])
+
+        # Return the settings
+        return settings
+
+    def get_sea_level_statistic_points(self, watersystem: WaterSystem) -> pd.DataFrame:
+        """
+        Return the reference points for sea level statistics
+
+        Parameters
+        ----------
+        watersystem : WaterSystem
+            Watersystem for which to return the reference points
+
+        Returns
+        -------
+        pd.DataFrame
+            All reference points
+        """
+        # Obtain the right table name
+        if watersystem in [
+            WaterSystem.COAST_SOUTH,
+            WaterSystem.COAST_CENTRAL,
+            WaterSystem.COAST_NORTH,
+        ]:
+            tablename = "CoastReferencePoints"
+        elif watersystem in [WaterSystem.WADDEN_SEA_WEST, WaterSystem.WADDEN_SEA_EAST]:
+            tablename = "WaddenSeaReferencePoints"
+        else:
+            raise NotImplementedError(
+                f"[ERROR] No sea level reference points implemented for {watersystem}."
+            )
+
+        # SQL
+        data = pd.read_sql(
+            f"SELECT Name, X, Y FROM {tablename}", self.con, index_col="Name"
+        )
+
+        # Return data
+        return data
+
+    def get_sea_level_sub_systems(self, watersystem: WaterSystem) -> pd.DataFrame:
+        """
+        Return all subsystems of the coast or Wadden sea.
+
+        Parameters
+        ----------
+        watersystem : WaterSystem
+            Watersystem for which to return the reference points
+
+        Returns
+        -------
+        pd.DataFrame
+            All subsystems
+        """
+        # Obtain the right table name
+        if watersystem in [
+            WaterSystem.COAST_SOUTH,
+            WaterSystem.COAST_CENTRAL,
+            WaterSystem.COAST_NORTH,
+        ]:
+            tablename = "CoastSubSystems"
+        elif watersystem in [WaterSystem.WADDEN_SEA_WEST, WaterSystem.WADDEN_SEA_EAST]:
+            tablename = "WaddenSeaSubSystems"
+        else:
+            raise NotImplementedError(
+                f"[ERROR] No sea level reference points implemented for {watersystem}."
+            )
+
+        # SQL
+        data = pd.read_sql(
+            f"SELECT SubSystemId, Point1, Point2, Point3 FROM {tablename}",
+            self.con,
+            index_col="SubSystemId",
+        )
+
+        # Return data
+        return data
+
+    def get_input_variable_ids(self) -> dict:
+        """
+        Return a dictionary with the input variable ids (ivid) and symbols (isymbol)
+
+        Returns
+        -------
+        dict
+            A dictionary with {ivid : isymbol}
+        """
+        # Query
+        sql = "SELECT InputVariableId, InputVariableSymbol FROM InputVariables"
+        data = self.con.execute(sql).fetchall()
+
+        # Process data
+        results = {data[i][0]: data[i][1] for i in range(len(data))}
+
+        # Return the dictionary
+        return results
+
+    def get_result_variable_ids(self) -> dict:
+        """
+        Return a dictionary with the result variable ids (rvid) and symbols (rsymbol)
+
+        Returns
+        -------
+        dict
+            A dictionary with {rvid : rsymbol}
+        """
+        # Query
+        sql = "SELECT ResultVariableId, ResultVariableSymbol FROM ResultVariables"
+        data = self.con.execute(sql).fetchall()
+
+        # Process data
+        results = {data[i][0]: data[i][1] for i in range(len(data))}
+
+        # Return the dictionary
+        return results
+
+    def __parse_value(self, value):
+        # Parse value from string
+        try:
+            return float(value)
+        except ValueError:
+            pass
+        try:
+            return int(value)
+        except ValueError:
+            pass
+        if value.lower() == "true":
+            return True
+        elif value.lower() == "false":
+            return False
+        return value

pydra_core/io/file_hydranl.py

@@ -1,92 +1,92 @@
-import numpy as np
-import os
-
-
-class FileHydraNL:
-    """
-    Class with staticsmethods to read Hydra-NL input files
-    """
-
-    # Path to the package
-    PACKAGE_PATH = os.path.join(
-        os.path.split(os.path.dirname(__file__))[0], "data", "statistics"
-    )
-
-    @staticmethod
-    def read_file_2columns(path):
-        """
-        Read a Hydra-NL statistics file with two columns
-
-        Parameters
-        ----------
-        path : str
-            Path to the statistics file
-        """
-        # Open file
-        if "\\" in path:
-            path = path.replace("\\", "/")
-        with open(
-            os.path.join(FileHydraNL.PACKAGE_PATH, path), "r", encoding="cp1252"
-        ) as f:
-            lines = f.readlines()
-
-        # Read non commented values from file, and split per line
-        vals = [line.strip().split() for line in lines if not line.startswith("*")]
-
-        # Convert to float and split in columns
-        kol1, kol2 = np.vstack(vals).astype(float).T
-
-        return kol1, kol2
-
-    @staticmethod
-    def read_file_ncolumns(path):
-        """
-        Read a Hydra-NL statistics file with more than two columns
-
-        Parameters
-        ----------
-        path : str
-            Path to the statistics file
-        """
-        # Open file
-        if "\\" in path:
-            path = path.replace("\\", "/")
-        with open(
-            os.path.join(FileHydraNL.PACKAGE_PATH, path), "r", encoding="cp1252"
-        ) as f:
-            lines = f.readlines()
-
-        # Read non commented values from file, and split per line
-        vals = [line.strip().split() for line in lines if not line.startswith("*")]
-
-        # Convert to float and split in columns
-        floatvals = np.vstack(vals).astype(float)
-
-        return floatvals[:, 0], floatvals[:, 1:]
-
-    @staticmethod
-    def read_file_ncolumns_loc(path):
-        """
-        Read a Hydra-NL statistics location file with more than two columns
-
-        Parameters
-        ----------
-        path : str
-            Path to the statistics file
-        """
-        # Open file
-        if "\\" in path:
-            path = path.replace("\\", "/")
-        with open(
-            os.path.join(FileHydraNL.PACKAGE_PATH, path), "r", encoding="cp1252"
-        ) as f:
-            lines = f.readlines()
-
-        # Process data
-        vals_tot = [line.strip().split() for line in lines]
-
-        # Convert to float and split in columns
-        x = int(vals_tot[2][1][:-1])
-        y = int(vals_tot[2][2])
-
-        return x, y
+import numpy as np
+import os
+
+
+class FileHydraNL:
+    """
+    Class with staticsmethods to read Hydra-NL input files
+    """
+
+    # Path to the package
+    PACKAGE_PATH = os.path.join(
+        os.path.split(os.path.dirname(__file__))[0], "data", "statistics"
+    )
+
+    @staticmethod
+    def read_file_2columns(path):
+        """
+        Read a Hydra-NL statistics file with two columns
+
+        Parameters
+        ----------
+        path : str
+            Path to the statistics file
+        """
+        # Open file
+        if "\\" in path:
+            path = path.replace("\\", "/")
+        with open(
+            os.path.join(FileHydraNL.PACKAGE_PATH, path), "r", encoding="cp1252"
+        ) as f:
+            lines = f.readlines()
+
+        # Read non commented values from file, and split per line
+        vals = [line.strip().split() for line in lines if not line.startswith("*")]
+
+        # Convert to float and split in columns
+        kol1, kol2 = np.vstack(vals).astype(float).T
+
+        return kol1, kol2
+
+    @staticmethod
+    def read_file_ncolumns(path):
+        """
+        Read a Hydra-NL statistics file with more than two columns
+
+        Parameters
+        ----------
+        path : str
+            Path to the statistics file
+        """
+        # Open file
+        if "\\" in path:
+            path = path.replace("\\", "/")
+        with open(
+            os.path.join(FileHydraNL.PACKAGE_PATH, path), "r", encoding="cp1252"
+        ) as f:
+            lines = f.readlines()
+
+        # Read non commented values from file, and split per line
+        vals = [line.strip().split() for line in lines if not line.startswith("*")]
+
+        # Convert to float and split in columns
+        floatvals = np.vstack(vals).astype(float)
+
+        return floatvals[:, 0], floatvals[:, 1:]
+
+    @staticmethod
+    def read_file_ncolumns_loc(path):
+        """
+        Read a Hydra-NL statistics location file with more than two columns
+
+        Parameters
+        ----------
+        path : str
+            Path to the statistics file
+        """
+        # Open file
+        if "\\" in path:
+            path = path.replace("\\", "/")
+        with open(
+            os.path.join(FileHydraNL.PACKAGE_PATH, path), "r", encoding="cp1252"
+        ) as f:
+            lines = f.readlines()
+
+        # Process data
+        vals_tot = [line.strip().split() for line in lines]
+
+        # Convert to float and split in columns
+        x = int(vals_tot[2][1][:-1])
+        y = int(vals_tot[2][2])
+
+        return x, y