iqm-station-control-client 3.2__py3-none-any.whl → 3.4__py3-none-any.whl

iqm/station_control/client/station_control.py

@@ -29,7 +29,12 @@ from packaging.version import Version, parse
 import requests
 
 from exa.common.data.setting_node import SettingNode
-from exa.common.errors.exa_error import RequestError
+from exa.common.errors.server_errors import (
+    STATUS_CODE_TO_ERROR_MAPPING,
+    InternalServerError,
+    NotFoundError,
+    StationControlError,
+)
 from exa.common.qcm_data.qcm_data_client import QCMDataClient
 from iqm.station_control.client.list_models import (
     DutFieldDataList,
@@ -159,6 +164,12 @@ class StationControlClient:
         response = self._send_request(requests.get, "configuration")
         return response.json()
 
+    @cache
+    def get_exa_configuration(self) -> str:
+        """Return the recommended EXA configuration from the server."""
+        response = self._send_request(requests.get, "exa/configuration")
+        return response.content.decode("utf-8")
+
     def get_or_create_software_version_set(self, software_version_set: SoftwareVersionSet) -> int:
         """Get software version set ID from the database, or create if it doesn't exist."""
         # FIXME: We don't have information if the object was created or fetched. Thus, server always responds 200 (OK).
@@ -180,8 +191,8 @@ class StationControlClient:
         """Get a raw chip design record matching the given chip label."""
         try:
             response = self._send_request(requests.get, f"chip-design-records/{dut_label}")
-        except RequestError as err:
-            if str(err).find("An error occurred on the server with the error code 404") >= 0 and self._qcm_data_client:
+        except StationControlError as err:
+            if isinstance(err, NotFoundError) and self._qcm_data_client:
                 return self._qcm_data_client.get_chip_design_record(dut_label)
             raise err
         return response.json()
@@ -224,7 +235,7 @@ class StationControlClient:
             in monolithic mode or successful submission to the task queue in remote mode.
 
         Raises:
-            RequestError if submitting a sweep failed.
+            ExaError if submitting a sweep failed.
 
         """
         data = serialize_sweep_task_request(sweep_definition, queue_name="sweeps")
@@ -247,7 +258,7 @@ class StationControlClient:
 
     def delete_sweep(self, sweep_id: uuid.UUID) -> None:
         """Delete sweep in the database."""
-        self._send_request(requests.delete, f"sweeps/{sweep_id}/delete")
+        self._send_request(requests.delete, f"sweeps/{sweep_id}")
 
     def get_sweep_results(self, sweep_id: uuid.UUID) -> SweepResults:
         """Get N-dimensional sweep results from the database."""
@@ -644,7 +655,7 @@ class StationControlClient:
     def _poll_task(self, task_id: str) -> dict:
         task = self._send_request(requests.get, f"tasks/{task_id}").json()
         if task["task_status"] == "FAILURE":
-            raise RequestError(f"Server-side error. Details: {task}")
+            raise InternalServerError(task)
         return task
 
     def _send_request(
@@ -675,9 +686,9 @@ class StationControlClient:
                 error_message = response_json["detail"]
             except json.JSONDecodeError:
                 error_message = response.text
-            raise RequestError(
-                f"An error occurred on the server with the error code {response.status_code}: {error_message}"
-            )
+
+            error_class = STATUS_CODE_TO_ERROR_MAPPING[response.status_code]
+            raise error_class(error_message)
         return response
 
     def _check_api_versions(self):

{iqm_station_control_client-3.2.dist-info → iqm_station_control_client-3.4.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: iqm-station-control-client
-Version: 3.2
+Version: 3.4
 Summary: Python client for communicating with Station Control Service
 Author-email: IQM Finland Oy <info@meetiqm.com>
 License: Apache License

{iqm_station_control_client-3.2.dist-info → iqm_station_control_client-3.4.dist-info}/RECORD

@@ -1,6 +1,6 @@
 iqm/station_control/client/__init__.py,sha256=BmBIBdZa10r-IWCFzZ1-0DG6GQKPIXqGXltfXop4ZeQ,942
 iqm/station_control/client/list_models.py,sha256=SjD0DbCrM9z1SSuGoQS83lyJmDLuMOatpJUoW8itW9s,2335
-iqm/station_control/client/station_control.py,sha256=pvSGWObxTs61Iyiv7pcl9SDLvKijkaZZYzJC8VoGIgo,33584
+iqm/station_control/client/station_control.py,sha256=IvxsQUiSS_jov5UkvHlBcak3184ZBpx-KKNFFMpuwj8,33820
 iqm/station_control/client/utils.py,sha256=3VVLCXt6rcKNQc4HerlMxpxkRfTFMYCwOQHF2DS1OG8,1143
 iqm/station_control/client/serializers/__init__.py,sha256=8os3EGOtNTRFaviZdGwDyMt9GUpM3ZP7icPKAxOg1qg,1438
 iqm/station_control/client/serializers/channel_property_serializer.py,sha256=ChlX8B-blM5hjv3pUExHOd-vE3O_myPwILu36KZYYNU,7121
@@ -22,8 +22,8 @@ iqm/station_control/interface/models/run.py,sha256=m-iE3QMPQUOF7bsw8JCAM1Bd6bDVh
 iqm/station_control/interface/models/sequence.py,sha256=uOqMwF1x-vW6UHs2WnPD3PsuSgV3a8OTAsgn_4UENLw,2723
 iqm/station_control/interface/models/sweep.py,sha256=6SQ4Ty4_Rm1KTeR7YfrLmwyD-AnNE495LMxYu8dM4Ko,2947
 iqm/station_control/interface/models/type_aliases.py,sha256=3LB9viZVi8osavY5kKF8TH1crayG7-MLjgBqXDCqL2s,1018
-iqm_station_control_client-3.2.dist-info/LICENSE.txt,sha256=R6Q7eUrLyoCQgWYorQ8WJmVmWKYU3dxA3jYUp0wwQAw,11332
-iqm_station_control_client-3.2.dist-info/METADATA,sha256=TuwaCjnFL20R5QlwAgqpvBwwjoOPn8Jep4plUQZnsBo,13971
-iqm_station_control_client-3.2.dist-info/WHEEL,sha256=y4mX-SOX4fYIkonsAGA5N0Oy-8_gI4FXw5HNI1xqvWg,91
-iqm_station_control_client-3.2.dist-info/top_level.txt,sha256=NB4XRfyDS6_wG9gMsyX-9LTU7kWnTQxNvkbzIxGv3-c,4
-iqm_station_control_client-3.2.dist-info/RECORD,,
+iqm_station_control_client-3.4.dist-info/LICENSE.txt,sha256=R6Q7eUrLyoCQgWYorQ8WJmVmWKYU3dxA3jYUp0wwQAw,11332
+iqm_station_control_client-3.4.dist-info/METADATA,sha256=fypBAAypV7BKCp55g-WflzENsu_FzuuRaw1-RpJfCCg,13971
+iqm_station_control_client-3.4.dist-info/WHEEL,sha256=y4mX-SOX4fYIkonsAGA5N0Oy-8_gI4FXw5HNI1xqvWg,91
+iqm_station_control_client-3.4.dist-info/top_level.txt,sha256=NB4XRfyDS6_wG9gMsyX-9LTU7kWnTQxNvkbzIxGv3-c,4
+iqm_station_control_client-3.4.dist-info/RECORD,,