mxcubecore 1.384.0__py3-none-any.whl → 1.386.0__py3-none-any.whl

mxcubecore/HardwareObjects/ICATLIMS.py

@@ -326,9 +326,9 @@ class ICATLIMS(AbstractLims):
                 )
                 if sample_information is not None:
                     if len(HWR.beamline.session.get_full_path("", "")) > 0:
-                        destination_folder = HWR.beamline.session.get_full_path("", "")[
-                            0
-                        ]
+                        destination_folder = (
+                            HWR.beamline.session.get_base_process_directory()
+                        )
                         msg = "Download restource: "
                         msg += f"sample_sheet_id={sample_sheet_id} "
                         msg += f"destination_folder={destination_folder}"
@@ -820,21 +820,20 @@ class ICATLIMS(AbstractLims):
             msg = f"Sample {sample_id} not found"
             logging.getLogger("HWR").debug(msg)
 
-        start_time = datacollection_dict.get("startTime", "")
-        end_time = datacollection_dict.get("endTime", "")
+        start_time = datacollection_dict.get("collection_start_time", "")
+        end_time = datetime.now(ZoneInfo("Europe/Paris")).isoformat()
 
-        try:
-            dt_aware = datetime.strptime(start_time, "%Y-%m-%d %H:%M:%S").replace(
-                tzinfo=ZoneInfo("Europe/Paris")
-            )
-            dt_aware_end = datetime.strptime(end_time, "%Y-%m-%d %H:%M:%S").replace(
-                tzinfo=ZoneInfo("Europe/Paris")
-            )
+        if start_time:
+            try:
+                dt_aware = datetime.strptime(start_time, "%Y-%m-%d %H:%M:%S").replace(
+                    tzinfo=ZoneInfo("Europe/Paris")
+                )
+                start_time = dt_aware.isoformat(timespec="microseconds")
+            except (ValueError, TypeError):
+                logging.getLogger("HWR").exception("Cannot parse start time")
+        else:
+            start_time = datetime.now(ZoneInfo("Europe/Paris")).isoformat()
 
-            start_time = dt_aware.isoformat(timespec="microseconds")
-            end_time = dt_aware_end.isoformat(timespec="microseconds")
-        except TypeError:
-            logging.getLogger("HWR").exception("Cannot parse start and end time")
         bsx, bsy, shape, _ = HWR.beamline.beam.get_value()
         flux_end = datacollection_dict.get("flux_end") or HWR.beamline.flux.get_value()
 
@@ -887,6 +886,29 @@ class ICATLIMS(AbstractLims):
 
             directory = Path(energyscan_dict["scanFileFullPath"]).parent
 
+            start_time = energyscan_dict.get("startTime", "")
+            end_time = energyscan_dict.get("endTime", "")
+
+            if start_time:
+                try:
+                    dt_aware = datetime.strptime(
+                        start_time, "%Y-%m-%d %H:%M:%S"
+                    ).replace(tzinfo=ZoneInfo("Europe/Paris"))
+                    start_time = dt_aware.isoformat(timespec="microseconds")
+                    metadata.update({"startDate": start_time})
+                except (ValueError, TypeError):
+                    logging.getLogger("HWR").exception("Cannot parse start time")
+
+            if end_time:
+                try:
+                    dt_aware = datetime.strptime(end_time, "%Y-%m-%d %H:%M:%S").replace(
+                        tzinfo=ZoneInfo("Europe/Paris")
+                    )
+                    end_time = dt_aware.isoformat(timespec="microseconds")
+                    metadata.update({"endDate": end_time})
+                except (ValueError, TypeError):
+                    logging.getLogger("HWR").exception("Cannot parse start time")
+
             metadata.update(
                 {
                     "scanType": "energy_scan",
@@ -940,6 +962,29 @@ class ICATLIMS(AbstractLims):
 
             directory = Path(xfespectrum_dict["filename"]).parent
 
+            start_time = xfespectrum_dict.get("startTime", "")
+            end_time = xfespectrum_dict.get("endTime", "")
+
+            if start_time:
+                try:
+                    dt_aware = datetime.strptime(
+                        start_time, "%Y-%m-%d %H:%M:%S"
+                    ).replace(tzinfo=ZoneInfo("Europe/Paris"))
+                    start_time = dt_aware.isoformat(timespec="microseconds")
+                    metadata.update({"startDate": start_time})
+                except (ValueError, TypeError):
+                    logging.getLogger("HWR").exception("Cannot parse start time")
+
+            if end_time:
+                try:
+                    dt_aware = datetime.strptime(end_time, "%Y-%m-%d %H:%M:%S").replace(
+                        tzinfo=ZoneInfo("Europe/Paris")
+                    )
+                    end_time = dt_aware.isoformat(timespec="microseconds")
+                    metadata.update({"endDate": end_time})
+                except (ValueError, TypeError):
+                    logging.getLogger("HWR").exception("Cannot parse end time")
+
             metadata.update(
                 {
                     "scanType": "xrf",
@@ -1045,7 +1090,7 @@ class ICATLIMS(AbstractLims):
             try:
                 token = self.icat_session["sessionId"]
                 url = f"{self.url}/catalogue/{token}/files/download?sampleId={sample_id}&resourceId={resource.id}"
-                response = requests.get(url, stream=True, timeout=3)
+                response = requests.get(url, stream=True, timeout=30)
                 response.raise_for_status()
 
                 file_path = resource_folder / resource.filename
@@ -1110,15 +1155,6 @@ class ICATLIMS(AbstractLims):
             proposal = f"{HWR.beamline.session.proposal_code}"
             proposal += f"{HWR.beamline.session.proposal_number}"
 
-            try:
-                dt_aware = datetime.strptime(
-                    datacollection_dict.get("collection_start_time"),
-                    "%Y-%m-%d %H:%M:%S",
-                ).replace(tzinfo=ZoneInfo("Europe/Paris"))
-                start_time = dt_aware.isoformat(timespec="microseconds")
-            except RuntimeError:
-                logger.warning("Failed to parse start and end time")
-
             metadata.update(
                 {
                     "MX_dataCollectionId": datacollection_dict.get("collection_id"),
@@ -1149,7 +1185,6 @@ class ICATLIMS(AbstractLims):
                     ),
                     "MX_position_id": workflow_params.get("workflow_position_id"),
                     "group_by": workflow_params.get("workflow_group_by"),
-                    "startDate": start_time,
                 }
             )
 
@@ -1171,7 +1206,7 @@ class ICATLIMS(AbstractLims):
             )
 
             try:
-                metadata["lims"] = HWR.beamline.lims.get_active_lims()
+                metadata["lims"] = HWR.beamline.lims.get_active_lims().name
             except Exception:
                 logger.exception("Failed to read get_active_lims.")
 

mxcubecore/HardwareObjects/abstract/AbstractSampleChanger.py

@@ -312,6 +312,98 @@ class SampleChanger(Container, HardwareObject):
         """
         return self.status
 
+    def get_contents_as_dict(self) -> dict:
+        """
+        Build and return the hierarchical structure of the sample changer contents.
+
+        Returns:
+            dict: A nested dictionary describing the sample changer contents.
+                {
+                    "name": <str>,  # root sample changer address
+                    "room_temperature_mode": <bool> (optional),
+                    "children": [
+                        {
+                            "name": <str>,       # element address
+                            "status": <str>,     # "Loaded", "Used", "Present", or ""
+                            "id": <str>,         # element ID
+                            "selected": <bool>,  # whether element is selected
+                            "children": [ ... ]  # nested elements, same structure
+                        },
+                        ...
+                    ]
+                }
+        """
+        contents = {"name": self.get_address()}
+
+        if hasattr(self, "get_room_temperature_mode"):
+            contents["room_temperature_mode"] = self.get_room_temperature_mode()
+
+        for element in self.get_components():
+            if element.is_present():
+                self._add_element(contents, element)
+
+        return contents
+
+    def _get_status(self, element) -> str:
+        """
+        Determine the status string for a sample changer element.
+
+        Args:
+            element: The element object to check.
+
+        Returns:
+            str: One of:
+                - "Loaded":  element is a leaf and currently loaded
+                - "Used":    element is a leaf and has been loaded before
+                - "Present": element is present in the changer
+                - "":        element is not present
+        """
+        if element.is_leaf():
+            if element.is_loaded():
+                return "Loaded"
+            if element.has_been_loaded():
+                return "Used"
+            return ""
+        return "Present" if element.is_present() else ""
+
+    def _get_id(self, element) -> str:
+        """
+        Get the unique identifier (ID or token) for a sample changer element.
+
+        Args:
+            element: The element to get the ID for.
+
+        Returns:
+            str: The token (if root sample changer and available),
+                 the element ID (if available), or an empty string.
+        """
+        if element == self:
+            token = element.get_token()
+            return token if token else ""
+
+        return element.get_id() or ""
+
+    def _add_element(self, parent, element) -> dict:
+        """
+        Recursively add an element and its children into the contents dictionary.
+
+        Args:
+            parent (dict): The parent dictionary to append the element to.
+            element: The element object to add.
+        """
+        new_element = {
+            "name": element.get_address(),
+            "status": self._get_status(element),
+            "id": self._get_id(element),
+            "selected": element.is_selected(),
+        }
+
+        parent.setdefault("children", []).append(new_element)
+
+        if not element.is_leaf():
+            for child in element.get_components():
+                self._add_element(new_element, child)
+
     @property
     def progress_message(self) -> str:
         """

{mxcubecore-1.384.0.dist-info → mxcubecore-1.386.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: mxcubecore
-Version: 1.384.0
+Version: 1.386.0
 Summary: Core libraries for the MXCuBE application
 License: LGPL-3.0-or-later
 Keywords: mxcube,mxcube3,mxcubecore

{mxcubecore-1.384.0.dist-info → mxcubecore-1.386.0.dist-info}/RECORD

@@ -196,7 +196,7 @@ mxcubecore/HardwareObjects/GrobMotor.py,sha256=l5A9Us_ceLXlhEF6RNwyXQsI56fJiobWh
 mxcubecore/HardwareObjects/GrobSampleChanger.py,sha256=LMetcL45fWrwP4C8rtENrEDNydelLpJ77SD1JC5C3go,8113
 mxcubecore/HardwareObjects/Harvester.py,sha256=0TYUXmz-Pmfd7Dhz7ToUGHzJRuJmnga8-4BK4B0KGQA,26524
 mxcubecore/HardwareObjects/HarvesterMaintenance.py,sha256=8s4yHDEFG-C1WYyW_RlwrFPSpc8o5hGi14aQuFQFrHs,9405
-mxcubecore/HardwareObjects/ICATLIMS.py,sha256=irvsn5xLy0CZpheFG_UGUzFO1nIKn5Mz3GVGZt9c2ic,53129
+mxcubecore/HardwareObjects/ICATLIMS.py,sha256=YMlha9sg6U1EWSiKZ1Uy8fbud9XdfYNH9-oHESwPD34,54814
 mxcubecore/HardwareObjects/ISARAMaint.py,sha256=I8LHXK6wCfzixsxWmmcqWlrdaL3AOX91XmVeAwT7GPk,8959
 mxcubecore/HardwareObjects/LNLS/EPICSActuator.py,sha256=3dDj6aXyTz03m8osdA6udq_a4bch-KNcUMrXSNvbG5Q,2587
 mxcubecore/HardwareObjects/LNLS/EPICSMotor.py,sha256=lRTc1t32rJKNTIcnKCECrqo2m9BhlYvLq9NWg4Y0w0Q,3576
@@ -359,7 +359,7 @@ mxcubecore/HardwareObjects/abstract/AbstractNState.py,sha256=JZvD_ZpQu5U8ycaWFii
 mxcubecore/HardwareObjects/abstract/AbstractOnlineProcessing.py,sha256=vs_fJsDDLwivu1kYHA7RKJXj1aTH5V6G2tEy7MYGw4c,34104
 mxcubecore/HardwareObjects/abstract/AbstractProcedure.py,sha256=kW2SzN5kYYCeOS2IzpzQ9VAjYAoRU7ZtyDxqrOb8Pbg,7849
 mxcubecore/HardwareObjects/abstract/AbstractResolution.py,sha256=eLoymf-i637FbwfZwOtTmKOeSqCQjE2OnpSMdl8oIEg,8524
-mxcubecore/HardwareObjects/abstract/AbstractSampleChanger.py,sha256=h9j7XF8KrRtVQpthF_424fsEIR2LqsEZLNPrLaQB4TI,26718
+mxcubecore/HardwareObjects/abstract/AbstractSampleChanger.py,sha256=f_D8YSoHPHkVLtGBobKEw79oMDf4rBL1y4Io3bqYnIw,29968
 mxcubecore/HardwareObjects/abstract/AbstractSampleView.py,sha256=-E9cTMW-gw2vY91OeL2V5wg1icduLeYFOX50vo2_NEA,7774
 mxcubecore/HardwareObjects/abstract/AbstractShutter.py,sha256=OeHSr3N9x-a9QmGdQEu3usN23TYspJGDnZjAkShqk3w,2798
 mxcubecore/HardwareObjects/abstract/AbstractSlits.py,sha256=1fymQFlxJ9naOVT9A8THszEkHm7VxQSGJJWPQV0v3Hs,3839
@@ -467,8 +467,8 @@ mxcubecore/utils/conversion.py,sha256=G1bk2Mi2ZwGbZa5pEeiFaKWxhSVXVGqu1L9_SioyUO
 mxcubecore/utils/qt_import.py,sha256=0lPmqok_oYQZ059kJCq7RWdg490T8YKyRvoZGyWDy4M,14486
 mxcubecore/utils/tango.py,sha256=vwEVrIrWKEFaeaJUz3xbaC7XWHY8ZeJ-pfcSrTfZPIE,2114
 mxcubecore/utils/units.py,sha256=Gh7ovTUN00XBMUoyDG5W7akCx1pROL-M6pK2z1ouemg,1361
-mxcubecore-1.384.0.dist-info/COPYING,sha256=u-Mc8zCecwyo4YoP8UulmzCiZZ_MmCLROd_NBtOcRj0,35148
-mxcubecore-1.384.0.dist-info/COPYING.LESSER,sha256=46mU2C5kSwOnkqkw9XQAJlhBL2JAf1_uCD8lVcXyMRg,7652
-mxcubecore-1.384.0.dist-info/METADATA,sha256=66iGNybEYtqpHBGq_qJ5xpvzlYErk-ys5D7XCMh86OY,4259
-mxcubecore-1.384.0.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
-mxcubecore-1.384.0.dist-info/RECORD,,
+mxcubecore-1.386.0.dist-info/COPYING,sha256=u-Mc8zCecwyo4YoP8UulmzCiZZ_MmCLROd_NBtOcRj0,35148
+mxcubecore-1.386.0.dist-info/COPYING.LESSER,sha256=46mU2C5kSwOnkqkw9XQAJlhBL2JAf1_uCD8lVcXyMRg,7652
+mxcubecore-1.386.0.dist-info/METADATA,sha256=oh50leeNzHaCAdv4fg8O96k0ewFL-ijez3eQW0QsbLw,4259
+mxcubecore-1.386.0.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
+mxcubecore-1.386.0.dist-info/RECORD,,