tsp 1.7.1__py3-none-any.whl → 1.7.7__py3-none-any.whl

tsp/data/example_permos.csv

@@ -0,0 +1,8 @@
+time,0,0.5,1,1.5,2.3,3.6,5.09,7.08,9.57,12.56,20.04
+2002-11-18 00:00:00,NA,-0.38333,-0.18833,-0.27,-0.25167,-0.45667,-0.68167,-0.67333,-0.76333,-0.67667,-0.53167
+2002-11-19 00:00:00,NA,-0.53667,-0.1925,-0.2575,-0.24667,-0.45667,-0.6825,-0.685,-0.75083,-0.68083,-0.5375
+2002-11-20 00:00:00,NA,-0.61083,-0.21167,-0.26333,-0.24333,-0.46417,-0.67833,-0.66833,-0.7525,-0.6825,-0.54
+2002-11-21 00:00:00,NA,-0.46333,-0.2,-0.26333,-0.245,-0.45917,-0.66917,-0.67333,-0.74833,-0.68167,-0.54083
+2002-11-22 00:00:00,NA,-0.35583,-0.195,-0.2625,-0.2425,-0.46333,-0.66417,-0.67333,-0.75,-0.68417,-0.54
+2002-11-23 00:00:00,NA,-0.33333,-0.19833,-0.265,-0.2425,-0.46333,-0.66,-0.66833,-0.75,-0.68667,-0.53917
+2002-11-24 00:00:00,NA,-0.32333,-0.20083,-0.2675,-0.2425,-0.46667,-0.65833,-0.66167,-0.7475,-0.68417,-0.54

tsp/data/test_geotop_has_space.txt

@@ -0,0 +1,5 @@
+Date12[DDMMYYYYhhmm],Simulation_Period,Run,IDpoint,100.000000 
+01/12/2003 00:00,0,1,32,-1.000000 
+01/12/2003 01:00,1,1,32,-1.624711 
+01/12/2003 02:00,1,1,32,-2.366138 
+01/12/2003 03:00,1,1,32,-3.010153 

tsp/dataloggers/AbstractReader.py

@@ -1,43 +1,43 @@
-from abc import ABCMeta, abstractmethod
-import pandas as pd
-
-class AbstractReader(object):
-    __metaclass__ = ABCMeta
-
-    DATA = pd.DataFrame()
-    META = dict()
-    
-    def __init__(self, datefmt=None):
-        """
-        The generic class used to read logger data. Subclasses must be defined depending on what kind of instrument the data comes from. However, each reader behaves the same way. They all have the `.read()` method that takes the file name as an argument. You can create a reader first and then call the `.read()` method, or just run everything on one line. 
-
-        Attributes
-        ----------
-        DATA : pd.DataFrame
-            Reader objects should store csv data in a pandas dataframe in the `DATA` attribute.  Column titles should mostly be left unchanged with the exception of the datetime column which is renamed to `TIME`. TIME should always the first column in the dataframe.
-        META : dict
-            Where possible, any metadata that is found in the file is stored in a `META` attribute.
-
-        Notes
-        -----
-        Datalogger metadata will differ between instruments. However, there are many commonalities. The following is an alphabetical list of metadata that should use common terminology and formatting in the `META` dictionary for each datalogger type. 
-        
-        .. code-block:: python
-
-            latitude : (CF) latitude where data was collected
-            longitude : (CF) longitude where data was collected
-
-
-            Abbrevations:
-            CF - Climate and Forecast Conventions
-
-        """
-        if datefmt:
-            self.DATEFMT = datefmt
-
-    @abstractmethod
-    def read(self, file) -> "pd.DataFrame":
-        """read data from a file"""
-
-    def get_data(self):
-        return self.DATA
+from abc import ABCMeta, abstractmethod
+import pandas as pd
+
+class AbstractReader(object):
+    __metaclass__ = ABCMeta
+
+    DATA = pd.DataFrame()
+    META = dict()
+    
+    def __init__(self, datefmt=None):
+        """
+        The generic class used to read logger data. Subclasses must be defined depending on what kind of instrument the data comes from. However, each reader behaves the same way. They all have the `.read()` method that takes the file name as an argument. You can create a reader first and then call the `.read()` method, or just run everything on one line. 
+
+        Attributes
+        ----------
+        DATA : pd.DataFrame
+            Reader objects should store csv data in a pandas dataframe in the `DATA` attribute.  Column titles should mostly be left unchanged with the exception of the datetime column which is renamed to `TIME`. TIME should always the first column in the dataframe.
+        META : dict
+            Where possible, any metadata that is found in the file is stored in a `META` attribute.
+
+        Notes
+        -----
+        Datalogger metadata will differ between instruments. However, there are many commonalities. The following is an alphabetical list of metadata that should use common terminology and formatting in the `META` dictionary for each datalogger type. 
+        
+        .. code-block:: python
+
+            latitude : (CF) latitude where data was collected
+            longitude : (CF) longitude where data was collected
+
+
+            Abbrevations:
+            CF - Climate and Forecast Conventions
+
+        """
+        if datefmt:
+            self.DATEFMT = datefmt
+
+    @abstractmethod
+    def read(self, file) -> "pd.DataFrame":
+        """read data from a file"""
+
+    def get_data(self):
+        return self.DATA

tsp/dataloggers/FG2.py

@@ -1,111 +1,111 @@
-import pandas as pd
-import re
-import warnings
-
-from .AbstractReader import AbstractReader
-
-
-class FG2(AbstractReader):
-    DATEFMT = "%d.%m.%Y %H:%M:%S"
-    DELIMITER = ","
-    HK = re.compile("^HK")
-
-    def __init__(self):
-        """File reader for FG2
-        
-        FG2 is the newer software for GeoPrecision instruments
-        
-        Examples
-        --------
-        .. code-block:: python
-
-            from teaspoon.dataloggers import FG2
-            from pkg_resources import resource_filename
-            fg2_file = resource_filename("teaspoon", "dataloggers/test_files/FG2_399.csv")
-
-            # Read an FG2 file
-            FG2().read(fg2_file)
-        """
-        pass
-
-    def read(self, file: str) -> "pd.DataFrame":
-        """Read a FG2 geoprecision file
-
-        Parameters
-        ----------
-        file : str
-            Path to an FG2 Geoprecision datalogger export
-
-        Returns
-        -------
-        pandas.DataFrame
-            A dataframe containing the data
-        """
-        self.META['raw'] = list()
-        
-        data = list()
-
-        with open(file, "r") as f:
-            for line in f:
-                if self.__is_header(line):
-                    delimiters = line.count(self.DELIMITER)
-                    columns = line.strip().split(self.DELIMITER)
-
-                elif self._is_observation(line):
-                    line = line.strip()
-                    line += self.DELIMITER * (delimiters - line.count(self.DELIMITER))
-                    data.append(line.split(self.DELIMITER))
-
-                else:
-                    self.META['raw'].append(line)
-
-        self.DATA = pd.DataFrame(data, columns=columns)
-        
-        for col in self.DATA.columns:
-            if col == "TIME":
-                continue
-            
-            try:
-                self.DATA[col] = pd.to_numeric(self.DATA[col], errors='raise')
-            except ValueError:
-                warnings.warn("Could not successfully convert all data to numeric. Some data may be missing")
-                self.DATA[col] = pd.to_numeric(self.DATA[col], errors='coerce')
-
-        self.DATA["TIME"] = pd.to_datetime(self.DATA["TIME"], format=self.DATEFMT)
-        self.DATA = self.DATA.drop(["NO"], axis=1)
-        self.DATA = self.drop_hk(self.DATA)
-
-        self.META.update(parse_fg2_meta_lines(self.META['raw']))
-
-        return self.DATA
-
-    def _is_metadata(self, line) -> bool:
-        match = re.search("^<.*>$", line)
-        return bool(match)
-
-    def _is_observation(self, line: str) -> bool:
-        match = re.search(fr"^\d*{self.DELIMITER}\d\d.\d\d", line)
-        return bool(match) 
-
-    def __is_header(self, line: str) -> bool:
-        match = re.search(f"NO{self.DELIMITER}TIME", line)
-        return bool(match)
-
-    def _is_hk(self, name: str) -> bool:
-        if self.HK.match(name):
-            return True
-        return False
-
-    def drop_hk(self, df: "pd.DataFrame") -> "pd.DataFrame":
-        return df.drop([c for c in df if self._is_hk(c)], axis=1)
-
-
-def parse_fg2_meta_lines(meta: "list[str]") -> dict:
-    parsed = dict()
-    serial = re.compile(r"LOGGER: \$([\w]{6})")
-
-    for line in meta:
-        if serial.match(line):
-            parsed["logger_serial_number"] = serial.match(line).group(1)
-
+import pandas as pd
+import re
+import warnings
+
+from .AbstractReader import AbstractReader
+
+
+class FG2(AbstractReader):
+    DATEFMT = "%d.%m.%Y %H:%M:%S"
+    DELIMITER = ","
+    HK = re.compile("^HK")
+
+    def __init__(self):
+        """File reader for FG2
+        
+        FG2 is the newer software for GeoPrecision instruments
+        
+        Examples
+        --------
+        .. code-block:: python
+
+            from teaspoon.dataloggers import FG2
+            from pkg_resources import resource_filename
+            fg2_file = resource_filename("teaspoon", "dataloggers/test_files/FG2_399.csv")
+
+            # Read an FG2 file
+            FG2().read(fg2_file)
+        """
+        pass
+
+    def read(self, file: str) -> "pd.DataFrame":
+        """Read a FG2 geoprecision file
+
+        Parameters
+        ----------
+        file : str
+            Path to an FG2 Geoprecision datalogger export
+
+        Returns
+        -------
+        pandas.DataFrame
+            A dataframe containing the data
+        """
+        self.META['raw'] = list()
+        
+        data = list()
+
+        with open(file, "r") as f:
+            for line in f:
+                if self.__is_header(line):
+                    delimiters = line.count(self.DELIMITER)
+                    columns = line.strip().split(self.DELIMITER)
+
+                elif self._is_observation(line):
+                    line = line.strip()
+                    line += self.DELIMITER * (delimiters - line.count(self.DELIMITER))
+                    data.append(line.split(self.DELIMITER))
+
+                else:
+                    self.META['raw'].append(line)
+
+        self.DATA = pd.DataFrame(data, columns=columns)
+        
+        for col in self.DATA.columns:
+            if col == "TIME":
+                continue
+            
+            try:
+                self.DATA[col] = pd.to_numeric(self.DATA[col], errors='raise')
+            except ValueError:
+                warnings.warn("Could not successfully convert all data to numeric. Some data may be missing")
+                self.DATA[col] = pd.to_numeric(self.DATA[col], errors='coerce')
+
+        self.DATA["TIME"] = pd.to_datetime(self.DATA["TIME"], format=self.DATEFMT)
+        self.DATA = self.DATA.drop(["NO"], axis=1)
+        self.DATA = self.drop_hk(self.DATA)
+
+        self.META.update(parse_fg2_meta_lines(self.META['raw']))
+
+        return self.DATA
+
+    def _is_metadata(self, line) -> bool:
+        match = re.search("^<.*>$", line)
+        return bool(match)
+
+    def _is_observation(self, line: str) -> bool:
+        match = re.search(fr"^\d*{self.DELIMITER}\d\d.\d\d", line)
+        return bool(match) 
+
+    def __is_header(self, line: str) -> bool:
+        match = re.search(f"NO{self.DELIMITER}TIME", line)
+        return bool(match)
+
+    def _is_hk(self, name: str) -> bool:
+        if self.HK.match(name):
+            return True
+        return False
+
+    def drop_hk(self, df: "pd.DataFrame") -> "pd.DataFrame":
+        return df.drop([c for c in df if self._is_hk(c)], axis=1)
+
+
+def parse_fg2_meta_lines(meta: "list[str]") -> dict:
+    parsed = dict()
+    serial = re.compile(r"LOGGER: \$([\w]{6})")
+
+    for line in meta:
+        if serial.match(line):
+            parsed["logger_serial_number"] = serial.match(line).group(1)
+
     return parsed

tsp/dataloggers/GP5W.py

@@ -1,115 +1,115 @@
-
-import pandas as pd
-import re
-import warnings
-
-from .AbstractReader import AbstractReader
-
-
-class GP5W(AbstractReader):
-    DATEFMT = "%d.%m.%Y %H:%M:%S"
-    HK = re.compile("^#HK")
-
-    def __init__(self):
-        """File reader for GP5W 
-
-        GP5W is the older software for GeoPrecision instruments
-
-        Examples
-        --------
-
-        .. code-block:: python
-
-            from teaspoon.dataloggers import GP5W
-            from pkg_resources import resource_filename
-            gp5w_file = resource_filename("teaspoon", "dataloggers/test_files/GP5W_260.csv")
-
-            # To read a file, you might first create a reader object 
-            reader = GP5W()
-            reader.read(gp5w_file)
-
-            # Or instead, read the data in one line
-            data = GP5W().read(gp5w_file)
-        """
-        super().__init__()
-
-    def read(self, file: str) -> "pd.DataFrame":
-        """Read a GP5W geoprecision file
-
-        Parameters
-        ----------
-        file : str
-            Path to an GP5W Geoprecision datalogger export
-
-        Returns
-        -------
-        pandas.DataFrame
-            A dataframe containing the data
-
-        """
-        self.META['raw'] = list()
-
-        data = list()
-        
-        with open(file, "r") as f:
-            for line in f:
-                if self.__is_header(line):
-                    delimiters = line.count(",")
-                    columns = line.strip().split(",")
-
-                elif self._is_observation(line):
-                    line = line.strip()
-                    line += "," * (delimiters - line.count(","))
-                    data.append(line.split(","))
-
-                else:
-                    self.META['raw'].append(line)
-
-        self.DATA = pd.DataFrame(data, columns=columns)
-        
-        for col in self.DATA.columns:
-            if col == "Time":
-                continue
-            
-            try:
-                self.DATA[col] = pd.to_numeric(self.DATA[col], errors='raise')
-            except ValueError:
-                warnings.warn("Could not successfully convert all data to numeric. Some data may be missing")
-                self.DATA[col] = pd.to_numeric(self.DATA[col], errors='coerce')
-
-        self.DATA["Time"] = pd.to_datetime(self.DATA["Time"], format=self.DATEFMT)
-        self.DATA.rename(columns={"Time":"TIME"}, inplace=True)
-
-        self.DATA = self.DATA.drop(["No"], axis=1)
-        self.DATA = self.drop_hk(self.DATA)
-
-        self.META.update(parse_gp5w_meta_lines(self.META['raw']))
-
-        return self.DATA
-
-    def _is_observation(self, line: str) -> bool:
-        match = re.search(r"\d*,\d{2}\.\d{2}\.\d{4}", line)
-        return bool(match)
-
-    def __is_header(self, line: str) -> bool:
-        match = re.search("No,Time", line)
-        return bool(match)
-
-    def _is_hk(self, name: str) -> bool:
-        if self.HK.match(name):
-            return True
-        return False
-
-    def drop_hk(self, df: "pd.DataFrame") -> "pd.DataFrame":
-        return df.drop([c for c in df if self._is_hk(c)], axis=1)
-
-
-def parse_gp5w_meta_lines(meta: "list[str]") -> dict:
-    parsed = dict()
-    serial = re.compile(r"Logger: \#([\w]{6})")
-
-    for line in meta:
-        if serial.match(line):
-            parsed["logger_serial_number"] = serial.match(line).group(1)
-
+
+import pandas as pd
+import re
+import warnings
+
+from .AbstractReader import AbstractReader
+
+
+class GP5W(AbstractReader):
+    DATEFMT = "%d.%m.%Y %H:%M:%S"
+    HK = re.compile("^#HK")
+
+    def __init__(self):
+        """File reader for GP5W 
+
+        GP5W is the older software for GeoPrecision instruments
+
+        Examples
+        --------
+
+        .. code-block:: python
+
+            from teaspoon.dataloggers import GP5W
+            from pkg_resources import resource_filename
+            gp5w_file = resource_filename("teaspoon", "dataloggers/test_files/GP5W_260.csv")
+
+            # To read a file, you might first create a reader object 
+            reader = GP5W()
+            reader.read(gp5w_file)
+
+            # Or instead, read the data in one line
+            data = GP5W().read(gp5w_file)
+        """
+        super().__init__()
+
+    def read(self, file: str) -> "pd.DataFrame":
+        """Read a GP5W geoprecision file
+
+        Parameters
+        ----------
+        file : str
+            Path to an GP5W Geoprecision datalogger export
+
+        Returns
+        -------
+        pandas.DataFrame
+            A dataframe containing the data
+
+        """
+        self.META['raw'] = list()
+
+        data = list()
+        
+        with open(file, "r") as f:
+            for line in f:
+                if self.__is_header(line):
+                    delimiters = line.count(",")
+                    columns = line.strip().split(",")
+
+                elif self._is_observation(line):
+                    line = line.strip()
+                    line += "," * (delimiters - line.count(","))
+                    data.append(line.split(","))
+
+                else:
+                    self.META['raw'].append(line)
+
+        self.DATA = pd.DataFrame(data, columns=columns)
+        
+        for col in self.DATA.columns:
+            if col == "Time":
+                continue
+            
+            try:
+                self.DATA[col] = pd.to_numeric(self.DATA[col], errors='raise')
+            except ValueError:
+                warnings.warn("Could not successfully convert all data to numeric. Some data may be missing")
+                self.DATA[col] = pd.to_numeric(self.DATA[col], errors='coerce')
+
+        self.DATA["Time"] = pd.to_datetime(self.DATA["Time"], format=self.DATEFMT)
+        self.DATA.rename(columns={"Time":"TIME"}, inplace=True)
+
+        self.DATA = self.DATA.drop(["No"], axis=1)
+        self.DATA = self.drop_hk(self.DATA)
+
+        self.META.update(parse_gp5w_meta_lines(self.META['raw']))
+
+        return self.DATA
+
+    def _is_observation(self, line: str) -> bool:
+        match = re.search(r"\d*,\d{2}\.\d{2}\.\d{4}", line)
+        return bool(match)
+
+    def __is_header(self, line: str) -> bool:
+        match = re.search("No,Time", line)
+        return bool(match)
+
+    def _is_hk(self, name: str) -> bool:
+        if self.HK.match(name):
+            return True
+        return False
+
+    def drop_hk(self, df: "pd.DataFrame") -> "pd.DataFrame":
+        return df.drop([c for c in df if self._is_hk(c)], axis=1)
+
+
+def parse_gp5w_meta_lines(meta: "list[str]") -> dict:
+    parsed = dict()
+    serial = re.compile(r"Logger: \#([\w]{6})")
+
+    for line in meta:
+        if serial.match(line):
+            parsed["logger_serial_number"] = serial.match(line).group(1)
+
     return parsed

tsp/dataloggers/Geoprecision.py

@@ -1,34 +1,34 @@
-from typing import Optional, Type
-
-from tsp.dataloggers.AbstractReader import AbstractReader
-from tsp.dataloggers.FG2 import FG2
-from tsp.dataloggers.GP5W import GP5W
-
-import re
-
-firmwares = {GP5W: re.compile("GP5W", re.IGNORECASE),
-             FG2: re.compile("FG2", re.IGNORECASE)}
-
-def detect_geoprecision_type(file: str) -> "Optional[Type[AbstractReader]]":
-    """ Detect whether a geoprecision file uses from a 'GP5W' or 'FG2' firmware and return a 
-
-    Parameters
-    ----------
-    file : str
-        Path to a geoprecision file.
-
-    Returns
-    -------
-    Optional[Type[AbstractReader]]
-        An appropriate file reader. If the file corresponds to neither GP5W or FG2, None is returned.
-    """
-    with open(file, 'r') as f:
-        
-        header = f.readline()
-        
-        for g, pattern in firmwares.items():
-            if pattern.search(header):
-                return g
-
-    return None
-
+from typing import Optional, Type
+
+from tsp.dataloggers.AbstractReader import AbstractReader
+from tsp.dataloggers.FG2 import FG2
+from tsp.dataloggers.GP5W import GP5W
+
+import re
+
+firmwares = {GP5W: re.compile("GP5W", re.IGNORECASE),
+             FG2: re.compile("FG2", re.IGNORECASE)}
+
+def detect_geoprecision_type(file: str) -> "Optional[Type[AbstractReader]]":
+    """ Detect whether a geoprecision file uses from a 'GP5W' or 'FG2' firmware and return a 
+
+    Parameters
+    ----------
+    file : str
+        Path to a geoprecision file.
+
+    Returns
+    -------
+    Optional[Type[AbstractReader]]
+        An appropriate file reader. If the file corresponds to neither GP5W or FG2, None is returned.
+    """
+    with open(file, 'r') as f:
+        
+        header = f.readline()
+        
+        for g, pattern in firmwares.items():
+            if pattern.search(header):
+                return g
+
+    return None
+