qctrl-workflow-client 1.1.0__tar.gz → 1.2.0__tar.gz

{qctrl_workflow_client-1.1.0 → qctrl_workflow_client-1.2.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: qctrl-workflow-client
-Version: 1.1.0
+Version: 1.2.0
 Summary: Q-CTRL Client
 Home-page: https://q-ctrl.com
 License: https://q-ctrl.com/terms

{qctrl_workflow_client-1.1.0 → qctrl_workflow_client-1.2.0}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "qctrl-workflow-client"
-version = "1.1.0"
+version = "1.2.0"
 description = "Q-CTRL Client"
 license = "https://q-ctrl.com/terms"
 authors = ["Q-CTRL <support@q-ctrl.com>"]

{qctrl_workflow_client-1.1.0 → qctrl_workflow_client-1.2.0}/qctrlworkflowclient/__init__.py

@@ -11,7 +11,7 @@
 # distributed under the License is distributed on an "AS IS" BASIS. See the
 # License for the specific language.
 
-__version__ = "1.1.0"
+__version__ = "1.2.0"
 
 from .defaults import (
     get_authenticated_client_for_product,

{qctrl_workflow_client-1.1.0 → qctrl_workflow_client-1.2.0}/qctrlworkflowclient/router/api.py

@@ -575,7 +575,7 @@ class ApiRouter(BaseRouter):
         Wait until the requested number of machines are online.
         """
 
-        number_of_machines_online = self._get_number_of_machines_online()
+        number_of_machines_online = self.get_machine_status()["online"]
 
         def machines(count: int) -> str:
             if count == 1:
@@ -591,9 +591,9 @@ class ApiRouter(BaseRouter):
             "requested": number_of_machines_requested,
         }
 
-    def _get_number_of_machines_online(self) -> int:
+    def get_machine_status(self) -> dict[str, int]:
         """
-        Return the number of machines that are online.
+        Return the current state of all machines.
         """
         _query = gql.gql(
             """
@@ -601,8 +601,12 @@ class ApiRouter(BaseRouter):
                 tenant(organizationId:$organizationId) {
                     tenant {
                         currentInstances {
+                            initializing
+                            offline
                             online
-                        }
+                            pending
+                            terminating
+                            }
                     }
                     errors {
                         message
@@ -616,10 +620,7 @@ class ApiRouter(BaseRouter):
             _query,
             {"organizationId": self._get_configured_organization().organization_id},
         )
-        number_of_machines_online = response["tenant"]["tenant"]["currentInstances"][
-            "online"
-        ]
-        return number_of_machines_online
+        return response["tenant"]["tenant"]["currentInstances"]
 
     def activity_monitor(
         self,