pychemstation 0.10.7__py3-none-any.whl → 0.10.8__py3-none-any.whl

pychemstation/analysis/base_spectrum.py

@@ -2,6 +2,7 @@ import logging
 import os
 import pickle
 from abc import ABC, abstractmethod
+from typing import Optional, Union
 
 import matplotlib.pyplot as plt
 import numpy as np
@@ -43,16 +44,15 @@ class AbstractSpectrum(ABC):
         "baseline",
     }
 
-    def __init__(self, path=None, autosaving=True):
+    def __init__(
+        self, path: Optional[Union[str, bool]] = None, autosaving: Optional[bool] = True
+    ):
         """Default constructor, loads properties into instance namespace.
 
         Can be redefined in ancestor classes.
 
-        Args:
-            path (Union[str, bool], optional): Valid path to save data to.
-                If omitted, uses ".//spectrum". If False - no folder created.
-            autosaving (bool, optional): If the True (default) will save the
-                spectrum when the new one is loaded. Will drop otherwise.
+        :param path: Valid path to save data to.  If omitted, uses ".//spectrum". If False - no folder created.
+        :param autosaving: If the True (default) will save the  spectrum when the new one is loaded. Will drop otherwise.
         """
 
         self.autosaving = autosaving
@@ -70,10 +70,10 @@ class AbstractSpectrum(ABC):
             self.path = os.path.join(".", "spectrum")
             os.makedirs(self.path, exist_ok=True)
         else:
-            try:
+            if isinstance(path, str):
                 os.makedirs(path, exist_ok=True)
                 self.path = path
-            except TypeError:  # type(path) -> bool
+            else:
                 self.path = "."
 
         # creating logger
@@ -86,15 +86,14 @@ class AbstractSpectrum(ABC):
         self.__init__(path=self.path, autosaving=self.autosaving)
 
     @abstractmethod
-    def load_spectrum(self, x, y, timestamp):
+    def load_spectrum(self, x: np.ndarray, y: np.ndarray, timestamp: float):
         """Loads the spectral data.
 
         This method must be redefined in ancestor classes.
 
-        Args:
-            x (:obj: np.array): An array with data to be plotted as "x" axis.
-            y (:obj: np.array): An array with data to be plotted as "y" axis.
-            timestamp (float): Timestamp to the corresponding spectrum.
+        :param x: An array with data to be plotted as "x" axis.
+        :param y: An array with data to be plotted as "y" axis.
+        :param timestamp: Timestamp to the corresponding spectrum.
         """
 
         try:
@@ -107,8 +106,8 @@ class AbstractSpectrum(ABC):
                 self.save_data()
             self._dump()
 
-        self.x = x
-        self.y = y
+        self.x: np.ndarray = x
+        self.y: np.ndarray = y
         self.timestamp = timestamp
 
     def save_data(self, filename=None, verbose=False):

pychemstation/analysis/chromatogram.py

@@ -3,7 +3,7 @@
 import os
 import time
 from dataclasses import dataclass
-from typing import Dict
+from typing import Dict, Tuple
 
 import numpy as np
 
@@ -80,15 +80,14 @@ class AgilentHPLCChromatogram(AbstractSpectrum):
 
     ### PUBLIC METHODS TO LOAD RAW DATA ###
 
-    def extract_rawdata(self, experiment_dir: str, channel: str):
-        """
-        Reads raw data from Chemstation .CH files.
+    def extract_rawdata(
+        self, experiment_dir: str, channel: str
+    ) -> Tuple[np.ndarray, np.ndarray]:
+        """Reads raw data from Chemstation .CH files.
 
-        Args:
-            experiment_dir: .D directory with the .CH files
+        :param experiment_dir: .D directory with the .CH files
 
-        Returns:
-            np.array(times), np.array(values)   Raw chromatogram data
+        :returns: np.array(times), np.array(values)   Raw chromatogram data
         """
         filename = os.path.join(experiment_dir, f"DAD1{channel}")
         npz_file = filename + ".npz"

pychemstation/analysis/process_report.py

@@ -65,8 +65,7 @@ class ReportProcessor(abc.ABC):
 
 class CSVProcessor(ReportProcessor):
     def __init__(self, path: str):
-        """
-        Class to process reports in CSV form.
+        """Class to process reports in CSV form.
 
         :param path: the parent folder that contains the CSV report(s) to parse.
         """
@@ -84,7 +83,9 @@ class CSVProcessor(ReportProcessor):
                 if "00" in name and file_extension.lower() == "csv":
                     prefix, _, _ = name.partition("00")
                     return prefix
-        raise FileNotFoundError("Couldn't find the prefix for CSV")
+        raise FileNotFoundError(
+            "Couldn't find the prefix for CSV, please make sure the post-run settings generate a CSV."
+        )
 
     def report_contains(self, labels: List[str], want: List[str]):
         for label in labels:
@@ -104,8 +105,7 @@ class CSVProcessor(ReportProcessor):
         return all_labels_seen
 
     def process_report(self) -> Result[AgilentReport, AnyStr]:
-        """
-        Method to parse details from CSV report.
+        """Method to parse details from CSV report.
 
         :return: subset of complete report details, specifically the sample location, solvents in pumps,
          and list of peaks at each wavelength channel.
@@ -176,9 +176,7 @@ class CSVProcessor(ReportProcessor):
 
 
 class TXTProcessor(ReportProcessor):
-    """
-    Regex matches for column and unit combinations, courtesy of Veronica Lai.
-    """
+    """Regex matches for column and unit combinations, courtesy of Veronica Lai."""
 
     _column_re_dictionary = {
         "Peak": {  # peak index
@@ -212,8 +210,7 @@ class TXTProcessor(ReportProcessor):
         max_ret_time: int = 999,
         target_wavelength_range=None,
     ):
-        """
-        Class to process reports in CSV form.
+        """Class to process reports in CSV form.
 
         :param path: the parent folder that contains the CSV report(s) to parse.
         :param min_ret_time: peaks after this value (min) will be returned
@@ -228,8 +225,7 @@ class TXTProcessor(ReportProcessor):
         super().__init__(path)
 
     def process_report(self) -> Result[AgilentReport, Union[AnyStr, Exception]]:
-        """
-        Method to parse details from CSV report.
+        """Method to parse details from CSV report.
         If you want more functionality, use `aghplctools`.
         `from aghplctools.ingestion.text import pull_hplc_area_from_txt`
         `signals = pull_hplc_area_from_txt(file_path)`
@@ -281,8 +277,7 @@ class TXTProcessor(ReportProcessor):
             return Err(e)
 
     def parse_area_report(self, report_text: str) -> Dict:
-        """
-        Interprets report text and parses the area report section, converting it to dictionary.
+        """Interprets report text and parses the area report section, converting it to dictionary.
         Courtesy of Veronica Lai.
 
         :param report_text: plain text version of the report.
@@ -340,8 +335,7 @@ class TXTProcessor(ReportProcessor):
         return signals
 
     def build_peak_regex(self, signal_table: str) -> Pattern[str] | None:
-        """
-        Builds a peak regex from a signal table. Courtesy of Veronica Lai.
+        """Builds a peak regex from a signal table. Courtesy of Veronica Lai.
 
         :param signal_table: block of lines associated with an area table
         :return: peak line regex object (<=3.6 _sre.SRE_PATTERN, >=3.7 re.Pattern)

pychemstation/control/README.md

@@ -70,7 +70,7 @@ seq_table = SequenceTable(
             inj_source=InjectionSource.MANUAL
         ),
         SequenceEntry(
-            vial_location=TenVialColumn.ONE,
+            vial_location=VialBar.ONE,
             method="General-Poroshell",
             num_inj=1,
             inj_vol=1,

pychemstation/control/controllers/__init__.py

@@ -4,5 +4,6 @@
 
 from .comm import CommunicationController
 from . import data_aq
+from . import devices
 
-__all__ = ["CommunicationController", "data_aq"]
+__all__ = ["CommunicationController", "data_aq", "devices"]

pychemstation/control/controllers/comm.py

@@ -15,18 +15,23 @@ from typing import Optional, Union, Tuple, List
 
 from result import Err, Ok, Result
 
+from ...utils.abc_tables.abc_comm import ABCCommunicationController
 from ...utils.macro import (
     Command,
     HPLCErrorStatus,
     Status,
     str_to_status,
 )
-from .abc_tables.abc_comm import ABCCommunicationController
 
 
 class CommunicationController(ABCCommunicationController):
-    """
-    Class that communicates with Agilent using Macros
+    """Class that communicates with Agilent using Macros
+
+    :param comm_dir: the complete directory path that was used in the MACRO file, common file that pychemstation and Chemstation use to communicate.
+    :param cmd_file: name of the write file that pychemstation writes MACROs to, in `comm_dir`
+    :param reply_file: name of the read file that Chemstation replies to, in `comm_dir
+    :param offline: whether or not communication with Chemstation is to be established
+    :param debug: if True, prints all send MACROs to an out.txt file
     """
 
     def __init__(
@@ -37,12 +42,6 @@ class CommunicationController(ABCCommunicationController):
         offline: bool = False,
         debug: bool = False,
     ):
-        """
-        :param comm_dir:
-        :param cmd_file: Name of command file
-        :param reply_file: Name of reply file
-        :param debug: whether to save log of sent commands
-        """
         super().__init__(comm_dir, cmd_file, reply_file, offline, debug)
 
     def get_num_val(self, cmd: str) -> Union[int, float]:
@@ -151,21 +150,28 @@ class CommunicationController(ABCCommunicationController):
 
     def get_chemstation_dirs(self) -> Tuple[str, str, List[str]]:
         method_dir, sequence_dir, data_dirs = None, None, None
-        self.send(Command.GET_METHOD_DIR)
-        res = self.receive()
-        if res.is_ok():
-            method_dir = res.ok_value.string_response
-        self.send(Command.GET_SEQUENCE_DIR)
-        res = self.receive()
-        if res.is_ok():
-            sequence_dir = res.ok_value.string_response
-        self.send(Command.GET_DATA_DIRS)
-        res = self.receive()
-        if res.is_ok():
-            data_dirs = res.ok().string_response.split("|")
-        if method_dir and sequence_dir and data_dirs:
-            return method_dir, sequence_dir, data_dirs
-        else:
-            raise ValueError(
-                "Please provide the method, sequence and data directories, could not be found."
-            )
+        for _ in range(10):
+            self.send(Command.GET_METHOD_DIR)
+            res = self.receive()
+            if res.is_ok():
+                method_dir = res.ok_value.string_response
+            self.send(Command.GET_SEQUENCE_DIR)
+            res = self.receive()
+            if res.is_ok():
+                sequence_dir = res.ok_value.string_response
+            self.send(Command.GET_DATA_DIRS)
+            res = self.receive()
+            if res.is_ok():
+                data_dirs = res.ok().string_response.split("|")
+            if method_dir and sequence_dir and data_dirs:
+                if not sequence_dir[0].isalpha():
+                    sequence_dir = "C:" + sequence_dir
+                if not method_dir[0].isalpha():
+                    method_dir = "C:" + method_dir
+                for i, data_dir in enumerate(data_dirs):
+                    if not data_dir[0].isalpha():
+                        data_dirs[i] = "C:" + data_dir
+                return method_dir, sequence_dir, data_dirs
+        raise ValueError(
+            f"One of the method: {method_dir}, sequence: {sequence_dir} or data directories: {data_dirs} could not be found, please provide your own."
+        )