cyberdesk 1.11.0__py3-none-any.whl → 2.1.0__py3-none-any.whl

openapi_client/cyberdesk_cloud_client/models/pool_with_machines.py

@@ -0,0 +1,162 @@
+import datetime
+from collections.abc import Mapping
+from typing import TYPE_CHECKING, Any, TypeVar, Union, cast
+from uuid import UUID
+
+from attrs import define as _attrs_define
+from attrs import field as _attrs_field
+from dateutil.parser import isoparse
+
+from ..types import UNSET, Unset
+
+if TYPE_CHECKING:
+    from ..models.machine_response import MachineResponse
+
+
+T = TypeVar("T", bound="PoolWithMachines")
+
+
+@_attrs_define
+class PoolWithMachines:
+    """Pool response with machines included
+
+    Attributes:
+        name (str):
+        id (UUID):
+        organization_id (str):
+        created_at (datetime.datetime):
+        updated_at (datetime.datetime):
+        description (Union[None, Unset, str]):
+        machine_count (Union[None, Unset, int]): Number of machines in this pool Default: 0.
+        machines (Union[Unset, list['MachineResponse']]):
+    """
+
+    name: str
+    id: UUID
+    organization_id: str
+    created_at: datetime.datetime
+    updated_at: datetime.datetime
+    description: Union[None, Unset, str] = UNSET
+    machine_count: Union[None, Unset, int] = 0
+    machines: Union[Unset, list["MachineResponse"]] = UNSET
+    additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
+
+    def to_dict(self) -> dict[str, Any]:
+        name = self.name
+
+        id = str(self.id)
+
+        organization_id = self.organization_id
+
+        created_at = self.created_at.isoformat()
+
+        updated_at = self.updated_at.isoformat()
+
+        description: Union[None, Unset, str]
+        if isinstance(self.description, Unset):
+            description = UNSET
+        else:
+            description = self.description
+
+        machine_count: Union[None, Unset, int]
+        if isinstance(self.machine_count, Unset):
+            machine_count = UNSET
+        else:
+            machine_count = self.machine_count
+
+        machines: Union[Unset, list[dict[str, Any]]] = UNSET
+        if not isinstance(self.machines, Unset):
+            machines = []
+            for machines_item_data in self.machines:
+                machines_item = machines_item_data.to_dict()
+                machines.append(machines_item)
+
+        field_dict: dict[str, Any] = {}
+        field_dict.update(self.additional_properties)
+        field_dict.update(
+            {
+                "name": name,
+                "id": id,
+                "organization_id": organization_id,
+                "created_at": created_at,
+                "updated_at": updated_at,
+            }
+        )
+        if description is not UNSET:
+            field_dict["description"] = description
+        if machine_count is not UNSET:
+            field_dict["machine_count"] = machine_count
+        if machines is not UNSET:
+            field_dict["machines"] = machines
+
+        return field_dict
+
+    @classmethod
+    def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T:
+        from ..models.machine_response import MachineResponse
+
+        d = dict(src_dict)
+        name = d.pop("name")
+
+        id = UUID(d.pop("id"))
+
+        organization_id = d.pop("organization_id")
+
+        created_at = isoparse(d.pop("created_at"))
+
+        updated_at = isoparse(d.pop("updated_at"))
+
+        def _parse_description(data: object) -> Union[None, Unset, str]:
+            if data is None:
+                return data
+            if isinstance(data, Unset):
+                return data
+            return cast(Union[None, Unset, str], data)
+
+        description = _parse_description(d.pop("description", UNSET))
+
+        def _parse_machine_count(data: object) -> Union[None, Unset, int]:
+            if data is None:
+                return data
+            if isinstance(data, Unset):
+                return data
+            return cast(Union[None, Unset, int], data)
+
+        machine_count = _parse_machine_count(d.pop("machine_count", UNSET))
+
+        machines = []
+        _machines = d.pop("machines", UNSET)
+        for machines_item_data in _machines or []:
+            machines_item = MachineResponse.from_dict(machines_item_data)
+
+            machines.append(machines_item)
+
+        pool_with_machines = cls(
+            name=name,
+            id=id,
+            organization_id=organization_id,
+            created_at=created_at,
+            updated_at=updated_at,
+            description=description,
+            machine_count=machine_count,
+            machines=machines,
+        )
+
+        pool_with_machines.additional_properties = d
+        return pool_with_machines
+
+    @property
+    def additional_keys(self) -> list[str]:
+        return list(self.additional_properties.keys())
+
+    def __getitem__(self, key: str) -> Any:
+        return self.additional_properties[key]
+
+    def __setitem__(self, key: str, value: Any) -> None:
+        self.additional_properties[key] = value
+
+    def __delitem__(self, key: str) -> None:
+        del self.additional_properties[key]
+
+    def __contains__(self, key: str) -> bool:
+        return key in self.additional_properties

openapi_client/cyberdesk_cloud_client/models/request_log_response.py

@@ -29,6 +29,7 @@ class RequestLogResponse:
         response_size_bytes (Union[None, Unset, int]):
         duration_ms (Union[None, Unset, int]):
         error_message (Union[None, Unset, str]):
+        organization_id (Union[None, Unset, str]):
     """
 
     request_id: str
@@ -43,6 +44,7 @@ class RequestLogResponse:
     response_size_bytes: Union[None, Unset, int] = UNSET
     duration_ms: Union[None, Unset, int] = UNSET
     error_message: Union[None, Unset, str] = UNSET
+    organization_id: Union[None, Unset, str] = UNSET
     additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
 
     def to_dict(self) -> dict[str, Any]:
@@ -94,6 +96,12 @@ class RequestLogResponse:
         else:
             error_message = self.error_message
 
+        organization_id: Union[None, Unset, str]
+        if isinstance(self.organization_id, Unset):
+            organization_id = UNSET
+        else:
+            organization_id = self.organization_id
+
         field_dict: dict[str, Any] = {}
         field_dict.update(self.additional_properties)
         field_dict.update(
@@ -117,6 +125,8 @@ class RequestLogResponse:
             field_dict["duration_ms"] = duration_ms
         if error_message is not UNSET:
             field_dict["error_message"] = error_message
+        if organization_id is not UNSET:
+            field_dict["organization_id"] = organization_id
 
         return field_dict
 
@@ -195,6 +205,15 @@ class RequestLogResponse:
 
         error_message = _parse_error_message(d.pop("error_message", UNSET))
 
+        def _parse_organization_id(data: object) -> Union[None, Unset, str]:
+            if data is None:
+                return data
+            if isinstance(data, Unset):
+                return data
+            return cast(Union[None, Unset, str], data)
+
+        organization_id = _parse_organization_id(d.pop("organization_id", UNSET))
+
         request_log_response = cls(
             request_id=request_id,
             method=method,
@@ -208,6 +227,7 @@ class RequestLogResponse:
             response_size_bytes=response_size_bytes,
             duration_ms=duration_ms,
             error_message=error_message,
+            organization_id=organization_id,
         )
 
         request_log_response.additional_properties = d

openapi_client/cyberdesk_cloud_client/models/run_attachment_response.py

@@ -22,13 +22,14 @@ class RunAttachmentResponse:
         content_type (str):
         attachment_type (AttachmentType):
         id (UUID):
-        user_id (UUID):
         run_id (UUID):
         size_bytes (int):
         storage_path (str):
         created_at (datetime.datetime):
         target_path (Union[None, Unset, str]):
         cleanup_imports_after_run (Union[Unset, bool]):  Default: False.
+        user_id (Union[None, UUID, Unset]):
+        organization_id (Union[None, Unset, str]):
         expires_at (Union[None, Unset, datetime.datetime]):
     """
 
@@ -36,13 +37,14 @@ class RunAttachmentResponse:
     content_type: str
     attachment_type: AttachmentType
     id: UUID
-    user_id: UUID
     run_id: UUID
     size_bytes: int
     storage_path: str
     created_at: datetime.datetime
     target_path: Union[None, Unset, str] = UNSET
     cleanup_imports_after_run: Union[Unset, bool] = False
+    user_id: Union[None, UUID, Unset] = UNSET
+    organization_id: Union[None, Unset, str] = UNSET
     expires_at: Union[None, Unset, datetime.datetime] = UNSET
     additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
 
@@ -55,8 +57,6 @@ class RunAttachmentResponse:
 
         id = str(self.id)
 
-        user_id = str(self.user_id)
-
         run_id = str(self.run_id)
 
         size_bytes = self.size_bytes
@@ -73,6 +73,20 @@ class RunAttachmentResponse:
 
         cleanup_imports_after_run = self.cleanup_imports_after_run
 
+        user_id: Union[None, Unset, str]
+        if isinstance(self.user_id, Unset):
+            user_id = UNSET
+        elif isinstance(self.user_id, UUID):
+            user_id = str(self.user_id)
+        else:
+            user_id = self.user_id
+
+        organization_id: Union[None, Unset, str]
+        if isinstance(self.organization_id, Unset):
+            organization_id = UNSET
+        else:
+            organization_id = self.organization_id
+
         expires_at: Union[None, Unset, str]
         if isinstance(self.expires_at, Unset):
             expires_at = UNSET
@@ -89,7 +103,6 @@ class RunAttachmentResponse:
                 "content_type": content_type,
                 "attachment_type": attachment_type,
                 "id": id,
-                "user_id": user_id,
                 "run_id": run_id,
                 "size_bytes": size_bytes,
                 "storage_path": storage_path,
@@ -100,6 +113,10 @@ class RunAttachmentResponse:
             field_dict["target_path"] = target_path
         if cleanup_imports_after_run is not UNSET:
             field_dict["cleanup_imports_after_run"] = cleanup_imports_after_run
+        if user_id is not UNSET:
+            field_dict["user_id"] = user_id
+        if organization_id is not UNSET:
+            field_dict["organization_id"] = organization_id
         if expires_at is not UNSET:
             field_dict["expires_at"] = expires_at
 
@@ -116,8 +133,6 @@ class RunAttachmentResponse:
 
         id = UUID(d.pop("id"))
 
-        user_id = UUID(d.pop("user_id"))
-
         run_id = UUID(d.pop("run_id"))
 
         size_bytes = d.pop("size_bytes")
@@ -137,6 +152,32 @@ class RunAttachmentResponse:
 
         cleanup_imports_after_run = d.pop("cleanup_imports_after_run", UNSET)
 
+        def _parse_user_id(data: object) -> Union[None, UUID, Unset]:
+            if data is None:
+                return data
+            if isinstance(data, Unset):
+                return data
+            try:
+                if not isinstance(data, str):
+                    raise TypeError()
+                user_id_type_0 = UUID(data)
+
+                return user_id_type_0
+            except:  # noqa: E722
+                pass
+            return cast(Union[None, UUID, Unset], data)
+
+        user_id = _parse_user_id(d.pop("user_id", UNSET))
+
+        def _parse_organization_id(data: object) -> Union[None, Unset, str]:
+            if data is None:
+                return data
+            if isinstance(data, Unset):
+                return data
+            return cast(Union[None, Unset, str], data)
+
+        organization_id = _parse_organization_id(d.pop("organization_id", UNSET))
+
         def _parse_expires_at(data: object) -> Union[None, Unset, datetime.datetime]:
             if data is None:
                 return data
@@ -159,13 +200,14 @@ class RunAttachmentResponse:
             content_type=content_type,
             attachment_type=attachment_type,
             id=id,
-            user_id=user_id,
             run_id=run_id,
             size_bytes=size_bytes,
             storage_path=storage_path,
             created_at=created_at,
             target_path=target_path,
             cleanup_imports_after_run=cleanup_imports_after_run,
+            user_id=user_id,
+            organization_id=organization_id,
             expires_at=expires_at,
         )
 

openapi_client/cyberdesk_cloud_client/models/run_bulk_create.py

@@ -24,6 +24,8 @@ class RunBulkCreate:
         count (int): Number of runs to create (max 1000)
         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 at
+            least one of these pools.
         input_values (Union['RunBulkCreateInputValuesType0', None, Unset]): Input values for workflow variables
         file_inputs (Union[None, Unset, list['FileInput']]): Files to upload to the machine
     """
@@ -31,6 +33,7 @@ class RunBulkCreate:
     workflow_id: UUID
     count: int
     machine_id: Union[None, UUID, Unset] = UNSET
+    pool_ids: Union[None, Unset, list[UUID]] = UNSET
     input_values: Union["RunBulkCreateInputValuesType0", None, Unset] = UNSET
     file_inputs: Union[None, Unset, list["FileInput"]] = UNSET
     additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
@@ -50,6 +53,18 @@ class RunBulkCreate:
         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
@@ -80,6 +95,8 @@ class RunBulkCreate:
         )
         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:
@@ -114,6 +131,28 @@ class RunBulkCreate:
 
         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["RunBulkCreateInputValuesType0", None, Unset]:
             if data is None:
                 return data
@@ -157,6 +196,7 @@ class RunBulkCreate:
             workflow_id=workflow_id,
             count=count,
             machine_id=machine_id,
+            pool_ids=pool_ids,
             input_values=input_values,
             file_inputs=file_inputs,
         )

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 at
+            least one of these pools.
         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

@@ -8,6 +8,7 @@ from attrs import field as _attrs_field
 from dateutil.parser import isoparse
 
 from ..models.run_status import RunStatus
+from ..types import UNSET, Unset
 
 if TYPE_CHECKING:
     from ..models.run_response_input_values_type_0 import RunResponseInputValuesType0
@@ -26,7 +27,6 @@ class RunResponse:
         workflow_id (UUID):
         machine_id (Union[None, UUID]):
         id (UUID):
-        user_id (UUID):
         status (RunStatus):
         error (Union[None, list[str]]):
         output_data (Union['RunResponseOutputDataType0', None]):
@@ -35,12 +35,14 @@ class RunResponse:
         run_message_history (Union[None, list['RunResponseRunMessageHistoryType0Item']]):
         input_values (Union['RunResponseInputValuesType0', None]):
         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
     machine_id: Union[None, UUID]
     id: UUID
-    user_id: UUID
     status: RunStatus
     error: Union[None, list[str]]
     output_data: Union["RunResponseOutputDataType0", None]
@@ -49,6 +51,9 @@ class RunResponse:
     run_message_history: Union[None, list["RunResponseRunMessageHistoryType0Item"]]
     input_values: Union["RunResponseInputValuesType0", None]
     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]:
@@ -65,8 +70,6 @@ class RunResponse:
 
         id = str(self.id)
 
-        user_id = str(self.user_id)
-
         status = self.status.value
 
         error: Union[None, list[str]]
@@ -114,6 +117,32 @@ class RunResponse:
 
         created_at = self.created_at.isoformat()
 
+        user_id: Union[None, Unset, str]
+        if isinstance(self.user_id, Unset):
+            user_id = UNSET
+        elif isinstance(self.user_id, UUID):
+            user_id = str(self.user_id)
+        else:
+            user_id = self.user_id
+
+        organization_id: Union[None, Unset, str]
+        if isinstance(self.organization_id, Unset):
+            organization_id = UNSET
+        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(
@@ -121,7 +150,6 @@ class RunResponse:
                 "workflow_id": workflow_id,
                 "machine_id": machine_id,
                 "id": id,
-                "user_id": user_id,
                 "status": status,
                 "error": error,
                 "output_data": output_data,
@@ -132,6 +160,12 @@ class RunResponse:
                 "created_at": created_at,
             }
         )
+        if user_id is not UNSET:
+            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
 
@@ -161,8 +195,6 @@ class RunResponse:
 
         id = UUID(d.pop("id"))
 
-        user_id = UUID(d.pop("user_id"))
-
         status = RunStatus(d.pop("status"))
 
         def _parse_error(data: object) -> Union[None, list[str]]:
@@ -264,11 +296,58 @@ class RunResponse:
 
         created_at = isoparse(d.pop("created_at"))
 
+        def _parse_user_id(data: object) -> Union[None, UUID, Unset]:
+            if data is None:
+                return data
+            if isinstance(data, Unset):
+                return data
+            try:
+                if not isinstance(data, str):
+                    raise TypeError()
+                user_id_type_0 = UUID(data)
+
+                return user_id_type_0
+            except:  # noqa: E722
+                pass
+            return cast(Union[None, UUID, Unset], data)
+
+        user_id = _parse_user_id(d.pop("user_id", UNSET))
+
+        def _parse_organization_id(data: object) -> Union[None, Unset, str]:
+            if data is None:
+                return data
+            if isinstance(data, Unset):
+                return data
+            return cast(Union[None, Unset, str], data)
+
+        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,
             id=id,
-            user_id=user_id,
             status=status,
             error=error,
             output_data=output_data,
@@ -277,6 +356,9 @@ class RunResponse:
             run_message_history=run_message_history,
             input_values=input_values,
             created_at=created_at,
+            user_id=user_id,
+            organization_id=organization_id,
+            pool_ids=pool_ids,
         )
 
         run_response.additional_properties = d