pychemstation 0.5.5__tar.gz → 0.5.6.dev1__tar.gz

{pychemstation-0.5.5 → pychemstation-0.5.6.dev1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: pychemstation
-Version: 0.5.5
+Version: 0.5.6.dev1
 Summary: Library to interact with Chemstation software, primarily used in Hein lab
 Home-page: https://gitlab.com/heingroup/device-api/pychemstation
 Author: Lucy Hao

{pychemstation-0.5.5 → pychemstation-0.5.6.dev1}/pychemstation/control/controllers/method.py

@@ -51,9 +51,10 @@ class MethodController(TableController):
         rows = self.get_num_rows()
         if rows.is_ok():
             timetable_rows = [self.get_row(r + 1) for r in range(int(rows.ok_value.num_response))]
-            return MethodTimetable(
+            self.table_state = MethodTimetable(
                 first_row=self.get_method_params(),
                 subsequent_rows=timetable_rows)
+            return self.table_state
         else:
             raise RuntimeError(rows.err_value)
 
@@ -88,6 +89,7 @@ class MethodController(TableController):
         if res.is_ok():
             parsed_response = res.ok_value.string_response
             assert parsed_response == f"{method_name}.M", "Switching Methods failed."
+        self.table_state = None
 
     def load_from_disk(self, method_name: str) -> MethodTimetable:
         """
@@ -118,7 +120,7 @@ class MethodController(TableController):
                     elif solvent.channel == "Channel_B":
                         organic_modifier = solvent
 
-            return MethodTimetable(
+            self.table_state = MethodTimetable(
                 first_row=HPLCMethodParams(
                     organic_modifier=organic_modifier.percentage,
                     flow=method.flow,
@@ -135,6 +137,7 @@ class MethodController(TableController):
                 organic_modifier=organic_modifier,
                 modifier_a=aq_modifier
             )
+            return self.table_state
         else:
             raise FileNotFoundError
 
@@ -242,7 +245,7 @@ class MethodController(TableController):
         """
         self.send(Command.STOP_METHOD_CMD)
 
-    def run(self, experiment_name: str):
+    def run(self, experiment_name: str, stall_while_running: bool = True):
         """
         This is the preferred method to trigger a run.
         Starts the currently selected method, storing data
@@ -261,14 +264,21 @@ class MethodController(TableController):
             folder_name = f"{experiment_name}_{timestamp}.D"
             self.data_files.append(os.path.join(self.data_dir, folder_name))
 
-        run_completed = self.check_hplc_done_running()
-
-        if not run_completed.is_ok():
-            raise RuntimeError("Run did not complete as expected")
+            if stall_while_running:
+                run_completed = self.check_hplc_done_running(method=self.table_state)
+                if run_completed.is_ok():
+                    self.data_files[-1] = run_completed.value
+                else:
+                    raise RuntimeError("Run error has occurred.")
+            else:
+                self.data_files[-1].dir = self.fuzzy_match_most_recent_folder(folder_name).ok_value
 
     def retrieve_recent_data_files(self) -> str:
         return self.data_files[-1]
 
-    def get_data(self) -> AgilentChannelChromatogramData:
-        self.get_spectrum(self.data_files[-1])
+    def get_data(self, custom_path: Optional[str] = None) -> AgilentChannelChromatogramData:
+        if not custom_path:
+            self.get_spectrum(self.data_files[-1])
+        else:
+            self.get_spectrum(custom_path)
         return AgilentChannelChromatogramData(**self.spectra)

{pychemstation-0.5.5 → pychemstation-0.5.6.dev1}/pychemstation/control/controllers/sequence.py

@@ -1,3 +1,6 @@
+from typing import  Optional
+
+
 import os
 import time
 
@@ -24,9 +27,10 @@ class SequenceController(TableController):
         seq_name = self.receive()
 
         if rows.is_ok() and seq_name.is_ok():
-            return SequenceTable(
+            self.table_state = SequenceTable(
                 name=seq_name.ok_value.string_response.partition(".S")[0],
                 rows=[self.get_row(r + 1) for r in range(int(rows.ok_value.num_response))])
+            return self.table_state
         raise RuntimeError(rows.err_value)
 
     def get_row(self, row: int) -> SequenceEntry:
@@ -60,6 +64,7 @@ class SequenceController(TableController):
         parsed_response = self.receive().value.string_response
 
         assert parsed_response == f"{seq_name}.S", "Switching sequence failed."
+        self.table_state = None
 
     def edit(self, sequence_table: SequenceTable):
         """
@@ -166,7 +171,7 @@ class SequenceController(TableController):
 
         self.send(Command.SAVE_SEQUENCE_CMD)
 
-    def run(self):
+    def run(self, stall_while_running: bool = True):
         """
         Starts the currently loaded sequence, storing data
         under the <data_dir>/<sequence table name> folder.
@@ -181,19 +186,21 @@ class SequenceController(TableController):
             self.data_files.append(SequenceDataFiles(dir=folder_name,
                                                      sequence_name=seq_table.name))
 
-        run_completed = self.check_hplc_done_running(sequence=seq_table)
-
-        if run_completed.is_ok():
-            self.data_files[-1].dir = run_completed.value
-        else:
-            raise RuntimeError("Run error has occured.")
+            if stall_while_running:
+                run_completed = self.check_hplc_done_running(sequence=seq_table)
+                if run_completed.is_ok():
+                    self.data_files[-1].dir = run_completed.value
+                else:
+                    raise RuntimeError("Run error has occurred.")
+            else:
+                self.data_files[-1].dir = self.fuzzy_match_most_recent_folder(folder_name).ok_value
 
     def retrieve_recent_data_files(self):
         sequence_data_files: SequenceDataFiles = self.data_files[-1]
         return sequence_data_files.dir
 
-    def get_data(self) -> list[AgilentChannelChromatogramData]:
-        parent_dir = self.data_files[-1].dir
+    def get_data(self, custom_path:Optional[str] = None ) -> list[AgilentChannelChromatogramData]:
+        parent_dir = self.data_files[-1].dir if not custom_path else custom_path
         subdirs = [x[0] for x in os.walk(self.data_dir)]
         potential_folders = sorted(list(filter(lambda d: parent_dir in d, subdirs)))
         self.data_files[-1].child_dirs = [f for f in potential_folders if
@@ -202,10 +209,5 @@ class SequenceController(TableController):
         spectra: list[AgilentChannelChromatogramData] = []
         for row in self.data_files[-1].child_dirs:
             self.get_spectrum(row)
-            spectra.append(
-                AgilentChannelChromatogramData(
-                    A=self.spectra["A"],
-                    B=self.spectra["B"],
-                    C=self.spectra["C"],
-                    D=self.spectra["D"]))
+            spectra.append(AgilentChannelChromatogramData(**self.spectra))
         return spectra

{pychemstation-0.5.5 → pychemstation-0.5.6.dev1}/pychemstation/control/controllers/table_controller.py

@@ -6,7 +6,6 @@ Authors: Lucy Hao
 
 import abc
 import os
-import time
 from typing import Union, Optional
 
 import polling
@@ -25,6 +24,7 @@ class TableController(abc.ABC):
     def __init__(self, controller: CommunicationController, src: str, data_dir: str, table: Table):
         self.controller = controller
         self.table = table
+        self.table_state : Optional[TableController] = None
 
         if os.path.isdir(src):
             self.src: str = src
@@ -120,7 +120,7 @@ class TableController(abc.ABC):
         self.send(TableOperation.CREATE_TABLE.value.format(register=self.table.register,
                                                            table_name=self.table.name))
 
-    def get_num_rows(self) -> Result[int, str]:
+    def get_num_rows(self) -> Result[Response, str]:
         self.send(TableOperation.GET_NUM_ROWS.value.format(register=self.table.register,
                                                            table_name=self.table.name,
                                                            col_name=RegisterFlag.NUM_ROWS))
@@ -139,8 +139,8 @@ class TableController(abc.ABC):
     def check_hplc_is_running(self) -> bool:
         started_running = polling.poll(
             lambda: isinstance(self.controller.get_status(), HPLCRunningStatus),
-            step=30,
-            max_tries=10)
+            step=5,
+            max_tries=100)
         return started_running
 
     def check_hplc_done_running(self,
@@ -154,7 +154,7 @@ class TableController(abc.ABC):
         """
         timeout = 10 * 60
         if method:
-            timeout = ((method.first_row.maximum_run_time + 2) * 60)
+            timeout = ((method.first_row.maximum_run_time + 5) * 60)
         if sequence:
             timeout *= len(sequence.rows)
 
@@ -162,26 +162,36 @@ class TableController(abc.ABC):
         finished_run = polling.poll(
             lambda: self.controller.check_if_running(),
             timeout=timeout,
-            step=60
-        )
-
-        if finished_run:
-            if os.path.exists(most_recent_folder):
-                return Ok(most_recent_folder)
-            else:
-                subdirs = [x[0] for x in os.walk(self.data_dir)]
-                potential_folders = sorted(list(filter(lambda d: most_recent_folder in d, subdirs)))
-                parent_dirs = []
-                for folder in potential_folders:
-                    path = os.path.normpath(folder)
-                    split_folder = path.split(os.sep)
-                    if most_recent_folder in split_folder[-1]:
-                        parent_dirs.append(folder)
-                parent_dir = sorted(parent_dirs, reverse=True)[0]
-                return Ok(parent_dir)
+            step=18)
+
+        check_folder = self.fuzzy_match_most_recent_folder(most_recent_folder)
+        if check_folder.is_ok() and finished_run:
+            return check_folder
+        elif check_folder.is_ok():
+            finished_run = polling.poll(
+                lambda: self.controller.check_if_running(),
+                timeout=timeout,
+                step=12)
+            if finished_run:
+                return check_folder
         else:
             return Err("Run did not complete as expected")
 
+    def fuzzy_match_most_recent_folder(self, most_recent_folder) -> Result[str, str]:
+        if os.path.exists(most_recent_folder):
+            return Ok(most_recent_folder)
+
+        subdirs = [x[0] for x in os.walk(self.data_dir)]
+        potential_folders = sorted(list(filter(lambda d: most_recent_folder in d, subdirs)))
+        parent_dirs = []
+        for folder in potential_folders:
+            path = os.path.normpath(folder)
+            split_folder = path.split(os.sep)
+            if most_recent_folder in split_folder[-1]:
+                parent_dirs.append(folder)
+        parent_dir = sorted(parent_dirs, reverse=True)[0]
+        return Ok(parent_dir)
+
     @abc.abstractmethod
     def retrieve_recent_data_files(self):
         pass

{pychemstation-0.5.5 → pychemstation-0.5.6.dev1}/pychemstation/utils/chromatogram.py

@@ -16,7 +16,7 @@ ACQUISITION_PARAMETERS = "acq.txt"
 
 # format used in acquisition parameters
 TIME_FORMAT = "%Y-%m-%d %H-%M-%S"
-SEQUENCE_TIME_FORMAT = "%Y-%m-%d %H"
+SEQUENCE_TIME_FORMAT = "%Y-%m-%d %H-%M"
 
 
 class AgilentHPLCChromatogram(AbstractSpectrum):

{pychemstation-0.5.5 → pychemstation-0.5.6.dev1}/pychemstation.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: pychemstation
-Version: 0.5.5
+Version: 0.5.6.dev1
 Summary: Library to interact with Chemstation software, primarily used in Hein lab
 Home-page: https://gitlab.com/heingroup/device-api/pychemstation
 Author: Lucy Hao

{pychemstation-0.5.5 → pychemstation-0.5.6.dev1}/setup.py

@@ -5,7 +5,7 @@ with open("README.md", "r") as fh:
 
 setuptools.setup(
     name="pychemstation",
-    version="0.5.5",
+    version="0.5.6.dev1",
     author="Lucy Hao",
     author_email="lhao03@student.ubc.ca",
     description="Library to interact with Chemstation software, primarily used in Hein lab",

{pychemstation-0.5.5 → pychemstation-0.5.6.dev1}/tests/test_comb.py

@@ -28,14 +28,14 @@ class TestCombinations(unittest.TestCase):
                 first_row=HPLCMethodParams(
                     organic_modifier=5,
                     flow=0.65,
-                    maximum_run_time=2),
+                    maximum_run_time=1),
                 subsequent_rows=[
                     TimeTableEntry(
                         start_time=0.10,
                         organic_modifer=5,
                         flow=0.34),
                     TimeTableEntry(
-                        start_time=1,
+                        start_time=0.5,
                         organic_modifer=98,
                         flow=0.55)])
             self.hplc_controller.edit_method(rand_method, save=True)