iqm-station-control-client 7.1.0__py3-none-any.whl → 8.0.0__py3-none-any.whl

iqm/station_control/client/iqm_server/iqm_server_client.py

@@ -49,7 +49,7 @@ from iqm.station_control.client.station_control import StationControlClient
 from iqm.station_control.interface.models import (
     DutData,
     DutFieldData,
-    JobStatus,
+    JobExecutorStatus,
     Statuses,
     SweepData,
     SweepDefinition,
@@ -292,20 +292,20 @@ def payload_to_sweep(job_payload: bytes) -> SweepDefinition:
     return sweep
 
 
-def to_job_status(job_status: proto.JobStatus) -> JobStatus:
+def to_job_status(job_status: proto.JobStatus) -> JobExecutorStatus:
     match job_status:
         case proto.JobStatus.IN_QUEUE:
-            return JobStatus.PENDING_EXECUTION
+            return JobExecutorStatus.PENDING_EXECUTION
         case proto.JobStatus.EXECUTING:
-            return JobStatus.EXECUTION_START
+            return JobExecutorStatus.EXECUTION_STARTED
         case proto.JobStatus.FAILED:
-            return JobStatus.FAILED
+            return JobExecutorStatus.FAILED
         case proto.JobStatus.COMPLETED:
-            return JobStatus.READY
+            return JobExecutorStatus.READY
         case proto.JobStatus.INTERRUPTED:
-            return JobStatus.ABORTED
+            return JobExecutorStatus.ABORTED
         case proto.JobStatus.CANCELLED:
-            return JobStatus.ABORTED
+            return JobExecutorStatus.ABORTED
     raise ValueError(f"Unknown job status: '{job_status}'")
 
 

iqm/station_control/client/serializers/sweep_serializers.py

@@ -28,7 +28,7 @@ from exa.common.sweep.database_serialization import decode_and_validate_sweeps,
 from iqm.station_control.client.serializers.datetime_serializers import deserialize_datetime, serialize_datetime
 from iqm.station_control.client.serializers.playlist_serializers import pack_playlist, unpack_playlist
 from iqm.station_control.interface.models import SweepData, SweepDefinition, SweepResults
-from iqm.station_control.interface.models.jobs import JobStatus
+from iqm.station_control.interface.models.jobs import JobExecutorStatus
 
 
 def serialize_sweep_definition(sweep_definition: SweepDefinition) -> SweepDefinitionProto:
@@ -91,7 +91,7 @@ def deserialize_sweep_data(data: dict) -> SweepData:
         modified_timestamp=deserialize_datetime(data["modified_timestamp"]),
         begin_timestamp=deserialize_datetime(data["begin_timestamp"]),
         end_timestamp=deserialize_datetime(data["end_timestamp"]),
-        job_status=JobStatus(data["job_status"]),
+        job_status=JobExecutorStatus(data["job_status"]),
     )
 
 

iqm/station_control/client/station_control.py

@@ -88,7 +88,7 @@ from iqm.station_control.interface.models import (
     SweepDefinition,
     SweepResults,
 )
-from iqm.station_control.interface.models.jobs import JobData, JobStatus
+from iqm.station_control.interface.models.jobs import JobData, JobExecutorStatus
 from iqm.station_control.interface.pydantic_base import PydanticBase
 
 logger = logging.getLogger(__name__)
@@ -684,7 +684,7 @@ class StationControlClient:
         update_progress_callback = update_progress_callback or (lambda status: None)
         try:
             job_status = self._poll_job_status_until_execution_start(job_id, update_progress_callback)
-            if JobStatus(job_status) not in JobStatus.terminal_statuses():
+            if JobExecutorStatus(job_status) not in JobExecutorStatus.terminal_statuses():
                 self._poll_job_status_until_terminal(job_id, update_progress_callback)
         except KeyboardInterrupt as exc:
             logger.info("Caught %s, revoking job %s", exc, job_id)
@@ -694,12 +694,12 @@ class StationControlClient:
 
     def _poll_job_status_until_execution_start(
         self, job_id: str, update_progress_callback: Callable[[Statuses], None]
-    ) -> JobStatus:
+    ) -> JobExecutorStatus:
         # Keep polling job status as long as it's PENDING, and update progress with `update_progress_callback`.
         max_seen_position = 0
         while True:
             job = self._poll_job(job_id)
-            if job.job_status >= JobStatus.EXECUTION_START:
+            if job.job_status >= JobExecutorStatus.EXECUTION_STARTED:
                 if max_seen_position:
                     update_progress_callback([("Progress in queue", max_seen_position, max_seen_position)])
                 return job.job_status
@@ -721,14 +721,14 @@ class StationControlClient:
         while True:
             job = self._poll_job(job_id)
             update_progress_callback(job.job_result.parallel_sweep_progress)
-            if job.job_status in JobStatus.terminal_statuses():
+            if job.job_status in JobExecutorStatus.terminal_statuses():
                 return
             sleep(1)
 
     def _poll_job(self, job_id: str) -> JobData:
         response = self._send_request(requests.get, f"jobs/{job_id}")
         job = JobData.model_validate(response.json())
-        if job.job_status == JobStatus.FAILED:
+        if job.job_status == JobExecutorStatus.FAILED:
             raise InternalServerError(f"Job: {job.job_id}\n{job.job_error}")
         return job
 

iqm/station_control/interface/models/__init__.py

@@ -15,7 +15,7 @@
 
 from iqm.station_control.interface.models.dut import DutData, DutFieldData
 from iqm.station_control.interface.models.jobs import (
-    JobStatus,
+    JobExecutorStatus,
 )
 from iqm.station_control.interface.models.observation import (
     ObservationData,

iqm/station_control/interface/models/jobs.py

@@ -26,7 +26,7 @@ from iqm.station_control.interface.pydantic_base import PydanticBase
 class TimelineEntry(PydanticBase):
     """Status and timestamp pair as described in a job timeline."""
 
-    status: JobStatus
+    status: JobExecutorStatus
     timestamp: datetime
 
 
@@ -49,16 +49,16 @@ class JobData(PydanticBase):
     """Job response data model"""
 
     job_id: UUID
-    job_status: JobStatus
+    job_status: JobExecutorStatus
     job_result: JobResult
     job_error: JobError | None
     position: int | None
 
 
 @functools.total_ordering
-class JobStatus(Enum):
+class JobExecutorStatus(Enum):
     """Enumeration of different states a job can be in. The ordering of these statuses is important,
-    and execution logic relies on it. Thus if a new status is added, ensure that it is slotted
+    and execution logic relies on it. Thus, if a new status is added, ensure that it is slotted
     in at the appropriate place. See the __lt__ implementation for further details.
     """
 
@@ -66,24 +66,24 @@ class JobStatus(Enum):
     RECEIVED = "received"
 
     # Validating the job
-    VALIDATING = "validating"
-    ACCEPTED = "accepted"
-
-    PREPARING_CALIBRATION = "preparing_calibration"
+    VALIDATION_STARTED = "validation_started"
+    VALIDATION_ENDED = "validation_ended"
 
     # Running PulLA
-    PENDING_COMPILATION = "pending_compilation"
-    COMPILING = "compiling"
-    COMPILED = "compiled"
+    FETCH_CALIBRATION_STARTED = "fetch_calibration_started"
+    FETCH_CALIBRATION_ENDED = "fetch_calibration_ended"
+    COMPILATION_STARTED = "compilation_started"
+    COMPILATION_ENDED = "compilation_ended"
 
     # Executing sweep
-    STORING_SWEEP_METADATA = "storing_sweep_metadata"
-    METADATA_STORED = "metadata_stored"
+    SAVE_SWEEP_METADATA_STARTED = "save_sweep_metadata_started"
+    SAVE_SWEEP_METADATA_ENDED = "save_sweep_metadata_ended"
     PENDING_EXECUTION = "pending_execution"
-    EXECUTION_START = "execution_start"
-    EXECUTION_END = "execution_end"
-    PENDING_POST_PROCESSING = "pending_post_processing"
-    POST_PROCESSING = "post_processing"
+    EXECUTION_STARTED = "execution_started"
+    EXECUTION_ENDED = "execution_ended"
+    POST_PROCESSING_PENDING = "post_processing_pending"
+    POST_PROCESSING_STARTED = "post_processing_started"
+    POST_PROCESSING_ENDED = "post_processing_ended"
     READY = "ready"
 
     # Job failed, can happen at any stage
@@ -101,10 +101,10 @@ class JobStatus(Enum):
     def __eq__(self, other):
         if isinstance(other, str):
             try:
-                other = JobStatus(other.lower())
+                other = JobExecutorStatus(other.lower())
             except ValueError:
                 return False
-        elif not isinstance(other, JobStatus):
+        elif not isinstance(other, JobExecutorStatus):
             return NotImplemented
         return self.name == other.name
 
@@ -112,21 +112,21 @@ class JobStatus(Enum):
         """Enable comparison according to definition order.
 
         Examples:
-            >>> JobStatus.ACCEPTED < JobStatus.COMPILED
+            >>> JobExecutorStatus.VALIDATION_ENDED < JobExecutorStatus.COMPILATION_ENDED
             True
 
         """
         if isinstance(other, str):
             try:
-                other = JobStatus(other.lower())
+                other = JobExecutorStatus(other.lower())
             except ValueError:
                 return NotImplemented
-        elif not isinstance(other, JobStatus):
+        elif not isinstance(other, JobExecutorStatus):
             return NotImplemented
-        members = list(JobStatus.__members__.values())
+        members = list(JobExecutorStatus.__members__.values())
         return members.index(self) < members.index(other)
 
     @classmethod
-    def terminal_statuses(cls) -> set[JobStatus]:
+    def terminal_statuses(cls) -> set[JobExecutorStatus]:
         """Statuses from which the execution can't continue."""
         return {cls.ABORTED, cls.FAILED, cls.READY}

iqm/station_control/interface/models/monitor.py

@@ -18,7 +18,7 @@ from __future__ import annotations
 from datetime import datetime
 from uuid import UUID
 
-from iqm.station_control.interface.models import JobStatus
+from iqm.station_control.interface.models import JobExecutorStatus
 from iqm.station_control.interface.pydantic_base import PydanticBase
 
 
@@ -26,7 +26,7 @@ class JobStateTimestamp(PydanticBase):
     """Represents a single timestamped state for a job."""
 
     job_id: UUID
-    status: JobStatus
+    status: JobExecutorStatus
     timestamp: datetime
 
 
@@ -40,6 +40,6 @@ class JobsInQueue(PydanticBase):
 class QueueState(PydanticBase):
     """Describes the state of a single job queue."""
 
-    queue: JobStatus
+    queue: JobExecutorStatus
     jobs_in_queue: int
     completed_jobs: int

iqm/station_control/interface/models/sweep.py

@@ -21,7 +21,7 @@ from iqm.models.playlist import Playlist
 
 from exa.common.data.setting_node import SettingNode
 from exa.common.sweep.util import NdSweep
-from iqm.station_control.interface.models.jobs import JobStatus
+from iqm.station_control.interface.models.jobs import JobExecutorStatus
 
 
 @dataclass(kw_only=True)
@@ -62,5 +62,5 @@ class SweepData(SweepBase):
     """Time when the sweep began in the station control."""
     end_timestamp: datetime | None
     """Time when the sweep ended in the station control."""
-    job_status: JobStatus
+    job_status: JobExecutorStatus
     """Status of sweep execution."""

{iqm_station_control_client-7.1.0.dist-info → iqm_station_control_client-8.0.0.dist-info}/METADATA

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

{iqm_station_control_client-7.1.0.dist-info → iqm_station_control_client-8.0.0.dist-info}/RECORD

@@ -1,11 +1,11 @@
 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=Aep9RaCYj_DFggUC0l7ncJUaYG-FJbumbb2PHqWvJ5c,39603
+iqm/station_control/client/station_control.py,sha256=ShSgH1lgaKE6UtcDcfwOg7vAINs_WKf2jynnL9ax50E,39661
 iqm/station_control/client/utils.py,sha256=cpS3hXEeeIXeqd_vBnnwo3JHS83FrNpG07SiTUwUx-I,1650
 iqm/station_control/client/iqm_server/__init__.py,sha256=nLsRHN1rnOKXwuzaq_liUpAYV3sis5jkyHccSdacV7U,624
 iqm/station_control/client/iqm_server/error.py,sha256=ZLV2-gxFLHZjZVkI3L5sWcBMiay7NT-ijIEvrXgVJT8,1166
 iqm/station_control/client/iqm_server/grpc_utils.py,sha256=ee42C31_JIXlP6ikZQbohgUQjqCvcSSxIW_9lm9MMV8,5727
-iqm/station_control/client/iqm_server/iqm_server_client.py,sha256=zDPfHFixd_PresI57wSs9PbiWQ9akxUkvpkOesxatas,14711
+iqm/station_control/client/iqm_server/iqm_server_client.py,sha256=6TuzBCv0yygy5B7re_nvK3NjEw_zSnP2ExBAHP0HyoY,14777
 iqm/station_control/client/iqm_server/meta_class.py,sha256=pePJ0Xy0aiJg-bZWK8D87gblq6imfXLsZHjpZkf5D9s,1399
 iqm/station_control/client/iqm_server/proto/__init__.py,sha256=mOJQ_H-NEyJMffRaDSSZeXrScHaHaHEXULv-O_OJA3A,1345
 iqm/station_control/client/iqm_server/proto/calibration_pb2.py,sha256=gum0DGmqxhbfaar8SqahmSif1pB6hgo0pVcnoi3VMUo,3017
@@ -32,23 +32,23 @@ iqm/station_control/client/serializers/playlist_serializers.py,sha256=S8RuKdqeJx
 iqm/station_control/client/serializers/run_serializers.py,sha256=4zH0I5EvvaP7wgLMprXXWa36nAPO4Lv0fPkCrDC_v-g,6698
 iqm/station_control/client/serializers/setting_node_serializer.py,sha256=m4Sbm8Qr3GiSNiE-Jh8gFEgfscfN1xxELb0vCa9cK70,1197
 iqm/station_control/client/serializers/struct_serializer.py,sha256=QztBsbRlRG_UrtpQLE3bi0WKEVn48kVB91H1g26PvqQ,3270
-iqm/station_control/client/serializers/sweep_serializers.py,sha256=pfdr36L6whk6M5koUsQsL61R1WUWiLOgQUHvwIVBUyY,5766
+iqm/station_control/client/serializers/sweep_serializers.py,sha256=MbmAei4o8jl4a8vxmkLmj5hc3fCTMbHrtejk2aLafB4,5782
 iqm/station_control/client/serializers/task_serializers.py,sha256=KTd3zdfqr9kSIiEbwsoSNiSu_hxzFfDOlLuttoyBfps,3700
 iqm/station_control/interface/__init__.py,sha256=MIQla-cBKPbZqBkp-LNyPfjiV0gzf-IFEwrMMhsnKlg,785
 iqm/station_control/interface/list_with_meta.py,sha256=GAXIDEXKeua6-2FoQ_O1tkhx-d8pBMGHaIkdvgg-cag,1185
 iqm/station_control/interface/pydantic_base.py,sha256=MVzcsH7wG1DON-qTw6KLpUDay7_b_9CDQgymVzg9HwI,1303
-iqm/station_control/interface/models/__init__.py,sha256=JeXFg-ltPOoSY6qVclUKCVyt-7okNgCptLb3h3t3JD4,1587
+iqm/station_control/interface/models/__init__.py,sha256=NpUt2IcWVVAiYZTdl6uOMk0IVWmLgJrC9D9irIZvF4c,1595
 iqm/station_control/interface/models/dut.py,sha256=dd1SpcsBe4P057jvcPqv39SjzekewwP07hThFe5ulNA,1216
-iqm/station_control/interface/models/jobs.py,sha256=HK95f3cWCawyqTaiMoGeLQEWl4hiHZDzSd63q_TXuuY,3732
-iqm/station_control/interface/models/monitor.py,sha256=9p-hg5DWrlbCIrJ0GyQ90DrHr1Yc9DCQyfpsQCm0cDE,1328
+iqm/station_control/interface/models/jobs.py,sha256=xPf7VnYMRKk5NqFXd7E8dIBAb0HwoW8AdAEQABDaTFI,4029
+iqm/station_control/interface/models/monitor.py,sha256=ItlgxtBup1hHg64uKeMKiFE7MaRRqSYdVRttsFD_XeU,1352
 iqm/station_control/interface/models/observation.py,sha256=Jce4lIsUtHRIFT3nr-cbKvh3dbR2Y_yM5x0yyvUdjF8,3261
 iqm/station_control/interface/models/observation_set.py,sha256=Ko2o3-9I38NfjNF2IQPcwfbwpkTQ3PIU7fUiSaDleX8,3031
 iqm/station_control/interface/models/run.py,sha256=m-iE3QMPQUOF7bsw8JCAM1Bd6bDVhAgxrtc_AC7rCkc,4097
 iqm/station_control/interface/models/sequence.py,sha256=uOqMwF1x-vW6UHs2WnPD3PsuSgV3a8OTAsgn_4UENLw,2723
-iqm/station_control/interface/models/sweep.py,sha256=ZV_1zcEF5_NY0nPgC75tU7s14TE1o0croBSClIVSmCE,2493
+iqm/station_control/interface/models/sweep.py,sha256=uzpbQ_nky7UxNE4c15U4oDN96f_-SgPDNVK9Sjdan3Q,2509
 iqm/station_control/interface/models/type_aliases.py,sha256=3LB9viZVi8osavY5kKF8TH1crayG7-MLjgBqXDCqL2s,1018
-iqm_station_control_client-7.1.0.dist-info/LICENSE.txt,sha256=R6Q7eUrLyoCQgWYorQ8WJmVmWKYU3dxA3jYUp0wwQAw,11332
-iqm_station_control_client-7.1.0.dist-info/METADATA,sha256=JBIbAx2UUghLhkLA2cCbIxiPUogflXtLJrEdueLAzuU,14009
-iqm_station_control_client-7.1.0.dist-info/WHEEL,sha256=y4mX-SOX4fYIkonsAGA5N0Oy-8_gI4FXw5HNI1xqvWg,91
-iqm_station_control_client-7.1.0.dist-info/top_level.txt,sha256=NB4XRfyDS6_wG9gMsyX-9LTU7kWnTQxNvkbzIxGv3-c,4
-iqm_station_control_client-7.1.0.dist-info/RECORD,,
+iqm_station_control_client-8.0.0.dist-info/LICENSE.txt,sha256=R6Q7eUrLyoCQgWYorQ8WJmVmWKYU3dxA3jYUp0wwQAw,11332
+iqm_station_control_client-8.0.0.dist-info/METADATA,sha256=q1e_eFVPPGjaV3ewVOvj72rJo4GeUKQY1me-zmLgUug,14009
+iqm_station_control_client-8.0.0.dist-info/WHEEL,sha256=y4mX-SOX4fYIkonsAGA5N0Oy-8_gI4FXw5HNI1xqvWg,91
+iqm_station_control_client-8.0.0.dist-info/top_level.txt,sha256=NB4XRfyDS6_wG9gMsyX-9LTU7kWnTQxNvkbzIxGv3-c,4
+iqm_station_control_client-8.0.0.dist-info/RECORD,,