cyberdesk 2.0.0__py3-none-any.whl → 2.1.1__py3-none-any.whl

openapi_client/cyberdesk_cloud_client/models/run_create.py

@@ -23,12 +23,15 @@ class RunCreate:
         workflow_id (UUID):
         machine_id (Union[None, UUID, Unset]): Machine ID. If not provided, an available machine will be automatically
             selected.
+        pool_ids (Union[None, Unset, list[UUID]]): Pool IDs to filter available machines. Machine must belong to all of
+            these pools (intersection). Ignored when machine_id is provided.
         input_values (Union['RunCreateInputValuesType0', None, Unset]): Input values for workflow variables
         file_inputs (Union[None, Unset, list['FileInput']]): Files to upload to the machine
     """
 
     workflow_id: UUID
     machine_id: Union[None, UUID, Unset] = UNSET
+    pool_ids: Union[None, Unset, list[UUID]] = UNSET
     input_values: Union["RunCreateInputValuesType0", None, Unset] = UNSET
     file_inputs: Union[None, Unset, list["FileInput"]] = UNSET
     additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
@@ -46,6 +49,18 @@ class RunCreate:
         else:
             machine_id = self.machine_id
 
+        pool_ids: Union[None, Unset, list[str]]
+        if isinstance(self.pool_ids, Unset):
+            pool_ids = UNSET
+        elif isinstance(self.pool_ids, list):
+            pool_ids = []
+            for pool_ids_type_0_item_data in self.pool_ids:
+                pool_ids_type_0_item = str(pool_ids_type_0_item_data)
+                pool_ids.append(pool_ids_type_0_item)
+
+        else:
+            pool_ids = self.pool_ids
+
         input_values: Union[None, Unset, dict[str, Any]]
         if isinstance(self.input_values, Unset):
             input_values = UNSET
@@ -75,6 +90,8 @@ class RunCreate:
         )
         if machine_id is not UNSET:
             field_dict["machine_id"] = machine_id
+        if pool_ids is not UNSET:
+            field_dict["pool_ids"] = pool_ids
         if input_values is not UNSET:
             field_dict["input_values"] = input_values
         if file_inputs is not UNSET:
@@ -107,6 +124,28 @@ class RunCreate:
 
         machine_id = _parse_machine_id(d.pop("machine_id", UNSET))
 
+        def _parse_pool_ids(data: object) -> Union[None, Unset, list[UUID]]:
+            if data is None:
+                return data
+            if isinstance(data, Unset):
+                return data
+            try:
+                if not isinstance(data, list):
+                    raise TypeError()
+                pool_ids_type_0 = []
+                _pool_ids_type_0 = data
+                for pool_ids_type_0_item_data in _pool_ids_type_0:
+                    pool_ids_type_0_item = UUID(pool_ids_type_0_item_data)
+
+                    pool_ids_type_0.append(pool_ids_type_0_item)
+
+                return pool_ids_type_0
+            except:  # noqa: E722
+                pass
+            return cast(Union[None, Unset, list[UUID]], data)
+
+        pool_ids = _parse_pool_ids(d.pop("pool_ids", UNSET))
+
         def _parse_input_values(data: object) -> Union["RunCreateInputValuesType0", None, Unset]:
             if data is None:
                 return data
@@ -149,6 +188,7 @@ class RunCreate:
         run_create = cls(
             workflow_id=workflow_id,
             machine_id=machine_id,
+            pool_ids=pool_ids,
             input_values=input_values,
             file_inputs=file_inputs,
         )

openapi_client/cyberdesk_cloud_client/models/run_response.py

@@ -37,6 +37,7 @@ class RunResponse:
         created_at (datetime.datetime):
         user_id (Union[None, UUID, Unset]):
         organization_id (Union[None, Unset, str]):
+        pool_ids (Union[None, Unset, list[UUID]]):
     """
 
     workflow_id: UUID
@@ -52,6 +53,7 @@ class RunResponse:
     created_at: datetime.datetime
     user_id: Union[None, UUID, Unset] = UNSET
     organization_id: Union[None, Unset, str] = UNSET
+    pool_ids: Union[None, Unset, list[UUID]] = UNSET
     additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
 
     def to_dict(self) -> dict[str, Any]:
@@ -129,6 +131,18 @@ class RunResponse:
         else:
             organization_id = self.organization_id
 
+        pool_ids: Union[None, Unset, list[str]]
+        if isinstance(self.pool_ids, Unset):
+            pool_ids = UNSET
+        elif isinstance(self.pool_ids, list):
+            pool_ids = []
+            for pool_ids_type_0_item_data in self.pool_ids:
+                pool_ids_type_0_item = str(pool_ids_type_0_item_data)
+                pool_ids.append(pool_ids_type_0_item)
+
+        else:
+            pool_ids = self.pool_ids
+
         field_dict: dict[str, Any] = {}
         field_dict.update(self.additional_properties)
         field_dict.update(
@@ -150,6 +164,8 @@ class RunResponse:
             field_dict["user_id"] = user_id
         if organization_id is not UNSET:
             field_dict["organization_id"] = organization_id
+        if pool_ids is not UNSET:
+            field_dict["pool_ids"] = pool_ids
 
         return field_dict
 
@@ -306,6 +322,28 @@ class RunResponse:
 
         organization_id = _parse_organization_id(d.pop("organization_id", UNSET))
 
+        def _parse_pool_ids(data: object) -> Union[None, Unset, list[UUID]]:
+            if data is None:
+                return data
+            if isinstance(data, Unset):
+                return data
+            try:
+                if not isinstance(data, list):
+                    raise TypeError()
+                pool_ids_type_0 = []
+                _pool_ids_type_0 = data
+                for pool_ids_type_0_item_data in _pool_ids_type_0:
+                    pool_ids_type_0_item = UUID(pool_ids_type_0_item_data)
+
+                    pool_ids_type_0.append(pool_ids_type_0_item)
+
+                return pool_ids_type_0
+            except:  # noqa: E722
+                pass
+            return cast(Union[None, Unset, list[UUID]], data)
+
+        pool_ids = _parse_pool_ids(d.pop("pool_ids", UNSET))
+
         run_response = cls(
             workflow_id=workflow_id,
             machine_id=machine_id,
@@ -320,6 +358,7 @@ class RunResponse:
             created_at=created_at,
             user_id=user_id,
             organization_id=organization_id,
+            pool_ids=pool_ids,
         )
 
         run_response.additional_properties = d