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

openapi_client/cyberdesk_cloud_client/api/pools/update_pool_v1_pools_pool_id_patch.py

@@ -0,0 +1,186 @@
+from http import HTTPStatus
+from typing import Any, Optional, Union
+from uuid import UUID
+
+import httpx
+
+from ... import errors
+from ...client import AuthenticatedClient, Client
+from ...models.http_validation_error import HTTPValidationError
+from ...models.pool_response import PoolResponse
+from ...models.pool_update import PoolUpdate
+from ...types import Response
+
+
+def _get_kwargs(
+    pool_id: UUID,
+    *,
+    body: PoolUpdate,
+) -> dict[str, Any]:
+    headers: dict[str, Any] = {}
+
+    _kwargs: dict[str, Any] = {
+        "method": "patch",
+        "url": f"/v1/pools/{pool_id}",
+    }
+
+    _kwargs["json"] = body.to_dict()
+
+    headers["Content-Type"] = "application/json"
+
+    _kwargs["headers"] = headers
+    return _kwargs
+
+
+def _parse_response(
+    *, client: Union[AuthenticatedClient, Client], response: httpx.Response
+) -> Optional[Union[HTTPValidationError, PoolResponse]]:
+    if response.status_code == 200:
+        response_200 = PoolResponse.from_dict(response.json())
+
+        return response_200
+    if response.status_code == 422:
+        response_422 = HTTPValidationError.from_dict(response.json())
+
+        return response_422
+    if client.raise_on_unexpected_status:
+        raise errors.UnexpectedStatus(response.status_code, response.content)
+    else:
+        return None
+
+
+def _build_response(
+    *, client: Union[AuthenticatedClient, Client], response: httpx.Response
+) -> Response[Union[HTTPValidationError, PoolResponse]]:
+    return Response(
+        status_code=HTTPStatus(response.status_code),
+        content=response.content,
+        headers=response.headers,
+        parsed=_parse_response(client=client, response=response),
+    )
+
+
+def sync_detailed(
+    pool_id: UUID,
+    *,
+    client: AuthenticatedClient,
+    body: PoolUpdate,
+) -> Response[Union[HTTPValidationError, PoolResponse]]:
+    """Update Pool
+
+     Update a pool's details.
+
+    Args:
+        pool_id (UUID):
+        body (PoolUpdate): Schema for updating a pool
+
+    Raises:
+        errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
+        httpx.TimeoutException: If the request takes longer than Client.timeout.
+
+    Returns:
+        Response[Union[HTTPValidationError, PoolResponse]]
+    """
+
+    kwargs = _get_kwargs(
+        pool_id=pool_id,
+        body=body,
+    )
+
+    response = client.get_httpx_client().request(
+        **kwargs,
+    )
+
+    return _build_response(client=client, response=response)
+
+
+def sync(
+    pool_id: UUID,
+    *,
+    client: AuthenticatedClient,
+    body: PoolUpdate,
+) -> Optional[Union[HTTPValidationError, PoolResponse]]:
+    """Update Pool
+
+     Update a pool's details.
+
+    Args:
+        pool_id (UUID):
+        body (PoolUpdate): Schema for updating a pool
+
+    Raises:
+        errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
+        httpx.TimeoutException: If the request takes longer than Client.timeout.
+
+    Returns:
+        Union[HTTPValidationError, PoolResponse]
+    """
+
+    return sync_detailed(
+        pool_id=pool_id,
+        client=client,
+        body=body,
+    ).parsed
+
+
+async def asyncio_detailed(
+    pool_id: UUID,
+    *,
+    client: AuthenticatedClient,
+    body: PoolUpdate,
+) -> Response[Union[HTTPValidationError, PoolResponse]]:
+    """Update Pool
+
+     Update a pool's details.
+
+    Args:
+        pool_id (UUID):
+        body (PoolUpdate): Schema for updating a pool
+
+    Raises:
+        errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
+        httpx.TimeoutException: If the request takes longer than Client.timeout.
+
+    Returns:
+        Response[Union[HTTPValidationError, PoolResponse]]
+    """
+
+    kwargs = _get_kwargs(
+        pool_id=pool_id,
+        body=body,
+    )
+
+    response = await client.get_async_httpx_client().request(**kwargs)
+
+    return _build_response(client=client, response=response)
+
+
+async def asyncio(
+    pool_id: UUID,
+    *,
+    client: AuthenticatedClient,
+    body: PoolUpdate,
+) -> Optional[Union[HTTPValidationError, PoolResponse]]:
+    """Update Pool
+
+     Update a pool's details.
+
+    Args:
+        pool_id (UUID):
+        body (PoolUpdate): Schema for updating a pool
+
+    Raises:
+        errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
+        httpx.TimeoutException: If the request takes longer than Client.timeout.
+
+    Returns:
+        Union[HTTPValidationError, PoolResponse]
+    """
+
+    return (
+        await asyncio_detailed(
+            pool_id=pool_id,
+            client=client,
+            body=body,
+        )
+    ).parsed

openapi_client/cyberdesk_cloud_client/models/__init__.py

@@ -32,6 +32,8 @@ from .http_validation_error import HTTPValidationError
 from .keyboard_key_request import KeyboardKeyRequest
 from .keyboard_type_request import KeyboardTypeRequest
 from .machine_create import MachineCreate
+from .machine_pool_assignment import MachinePoolAssignment
+from .machine_pool_update import MachinePoolUpdate
 from .machine_response import MachineResponse
 from .machine_status import MachineStatus
 from .machine_update import MachineUpdate
@@ -41,10 +43,15 @@ from .mouse_position import MousePosition
 from .paginated_response import PaginatedResponse
 from .paginated_response_connection_response import PaginatedResponseConnectionResponse
 from .paginated_response_machine_response import PaginatedResponseMachineResponse
+from .paginated_response_pool_response import PaginatedResponsePoolResponse
 from .paginated_response_run_attachment_response import PaginatedResponseRunAttachmentResponse
 from .paginated_response_run_response import PaginatedResponseRunResponse
 from .paginated_response_trajectory_response import PaginatedResponseTrajectoryResponse
 from .paginated_response_workflow_response import PaginatedResponseWorkflowResponse
+from .pool_create import PoolCreate
+from .pool_response import PoolResponse
+from .pool_update import PoolUpdate
+from .pool_with_machines import PoolWithMachines
 from .power_shell_exec_request import PowerShellExecRequest
 from .power_shell_session_request import PowerShellSessionRequest
 from .powershell_exec_v1_computer_machine_id_shell_powershell_exec_post_response_powershell_exec_v1_computer_machine_id_shell_powershell_exec_post import (
@@ -109,6 +116,8 @@ __all__ = (
     "KeyboardKeyRequest",
     "KeyboardTypeRequest",
     "MachineCreate",
+    "MachinePoolAssignment",
+    "MachinePoolUpdate",
     "MachineResponse",
     "MachineStatus",
     "MachineUpdate",
@@ -118,10 +127,15 @@ __all__ = (
     "PaginatedResponse",
     "PaginatedResponseConnectionResponse",
     "PaginatedResponseMachineResponse",
+    "PaginatedResponsePoolResponse",
     "PaginatedResponseRunAttachmentResponse",
     "PaginatedResponseRunResponse",
     "PaginatedResponseTrajectoryResponse",
     "PaginatedResponseWorkflowResponse",
+    "PoolCreate",
+    "PoolResponse",
+    "PoolUpdate",
+    "PoolWithMachines",
     "PowerShellExecRequest",
     "PowershellExecV1ComputerMachineIdShellPowershellExecPostResponsePowershellExecV1ComputerMachineIdShellPowershellExecPost",
     "PowerShellSessionRequest",

openapi_client/cyberdesk_cloud_client/models/machine_pool_assignment.py

@@ -0,0 +1,69 @@
+from collections.abc import Mapping
+from typing import Any, TypeVar
+from uuid import UUID
+
+from attrs import define as _attrs_define
+from attrs import field as _attrs_field
+
+T = TypeVar("T", bound="MachinePoolAssignment")
+
+
+@_attrs_define
+class MachinePoolAssignment:
+    """Schema for assigning machines to pools
+
+    Attributes:
+        machine_ids (list[UUID]): List of machine IDs to assign to the pool
+    """
+
+    machine_ids: list[UUID]
+    additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
+
+    def to_dict(self) -> dict[str, Any]:
+        machine_ids = []
+        for machine_ids_item_data in self.machine_ids:
+            machine_ids_item = str(machine_ids_item_data)
+            machine_ids.append(machine_ids_item)
+
+        field_dict: dict[str, Any] = {}
+        field_dict.update(self.additional_properties)
+        field_dict.update(
+            {
+                "machine_ids": machine_ids,
+            }
+        )
+
+        return field_dict
+
+    @classmethod
+    def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T:
+        d = dict(src_dict)
+        machine_ids = []
+        _machine_ids = d.pop("machine_ids")
+        for machine_ids_item_data in _machine_ids:
+            machine_ids_item = UUID(machine_ids_item_data)
+
+            machine_ids.append(machine_ids_item)
+
+        machine_pool_assignment = cls(
+            machine_ids=machine_ids,
+        )
+
+        machine_pool_assignment.additional_properties = d
+        return machine_pool_assignment
+
+    @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/machine_pool_update.py

@@ -0,0 +1,69 @@
+from collections.abc import Mapping
+from typing import Any, TypeVar
+from uuid import UUID
+
+from attrs import define as _attrs_define
+from attrs import field as _attrs_field
+
+T = TypeVar("T", bound="MachinePoolUpdate")
+
+
+@_attrs_define
+class MachinePoolUpdate:
+    """Schema for updating a machine's pool assignments
+
+    Attributes:
+        pool_ids (list[UUID]): List of pool IDs to assign the machine to
+    """
+
+    pool_ids: list[UUID]
+    additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
+
+    def to_dict(self) -> dict[str, Any]:
+        pool_ids = []
+        for pool_ids_item_data in self.pool_ids:
+            pool_ids_item = str(pool_ids_item_data)
+            pool_ids.append(pool_ids_item)
+
+        field_dict: dict[str, Any] = {}
+        field_dict.update(self.additional_properties)
+        field_dict.update(
+            {
+                "pool_ids": pool_ids,
+            }
+        )
+
+        return field_dict
+
+    @classmethod
+    def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T:
+        d = dict(src_dict)
+        pool_ids = []
+        _pool_ids = d.pop("pool_ids")
+        for pool_ids_item_data in _pool_ids:
+            pool_ids_item = UUID(pool_ids_item_data)
+
+            pool_ids.append(pool_ids_item)
+
+        machine_pool_update = cls(
+            pool_ids=pool_ids,
+        )
+
+        machine_pool_update.additional_properties = d
+        return machine_pool_update
+
+    @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/machine_response.py

@@ -1,6 +1,6 @@
 import datetime
 from collections.abc import Mapping
-from typing import Any, TypeVar, Union, cast
+from typing import TYPE_CHECKING, Any, TypeVar, Union, cast
 from uuid import UUID
 
 from attrs import define as _attrs_define
@@ -10,6 +10,10 @@ from dateutil.parser import isoparse
 from ..models.machine_status import MachineStatus
 from ..types import UNSET, Unset
 
+if TYPE_CHECKING:
+    from ..models.pool_response import PoolResponse
+
+
 T = TypeVar("T", bound="MachineResponse")
 
 
@@ -31,6 +35,7 @@ class MachineResponse:
         os_info (Union[None, Unset, str]):
         user_id (Union[None, UUID, Unset]):
         organization_id (Union[None, Unset, str]):
+        pools (Union[None, Unset, list['PoolResponse']]):
     """
 
     fingerprint: str
@@ -46,6 +51,7 @@ class MachineResponse:
     os_info: Union[None, Unset, str] = UNSET
     user_id: Union[None, UUID, Unset] = UNSET
     organization_id: Union[None, Unset, str] = UNSET
+    pools: Union[None, Unset, list["PoolResponse"]] = UNSET
     additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
 
     def to_dict(self) -> dict[str, Any]:
@@ -101,6 +107,18 @@ class MachineResponse:
         else:
             organization_id = self.organization_id
 
+        pools: Union[None, Unset, list[dict[str, Any]]]
+        if isinstance(self.pools, Unset):
+            pools = UNSET
+        elif isinstance(self.pools, list):
+            pools = []
+            for pools_type_0_item_data in self.pools:
+                pools_type_0_item = pools_type_0_item_data.to_dict()
+                pools.append(pools_type_0_item)
+
+        else:
+            pools = self.pools
+
         field_dict: dict[str, Any] = {}
         field_dict.update(self.additional_properties)
         field_dict.update(
@@ -126,11 +144,15 @@ class MachineResponse:
             field_dict["user_id"] = user_id
         if organization_id is not UNSET:
             field_dict["organization_id"] = organization_id
+        if pools is not UNSET:
+            field_dict["pools"] = pools
 
         return field_dict
 
     @classmethod
     def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T:
+        from ..models.pool_response import PoolResponse
+
         d = dict(src_dict)
         fingerprint = d.pop("fingerprint")
 
@@ -208,6 +230,28 @@ class MachineResponse:
 
         organization_id = _parse_organization_id(d.pop("organization_id", UNSET))
 
+        def _parse_pools(data: object) -> Union[None, Unset, list["PoolResponse"]]:
+            if data is None:
+                return data
+            if isinstance(data, Unset):
+                return data
+            try:
+                if not isinstance(data, list):
+                    raise TypeError()
+                pools_type_0 = []
+                _pools_type_0 = data
+                for pools_type_0_item_data in _pools_type_0:
+                    pools_type_0_item = PoolResponse.from_dict(pools_type_0_item_data)
+
+                    pools_type_0.append(pools_type_0_item)
+
+                return pools_type_0
+            except:  # noqa: E722
+                pass
+            return cast(Union[None, Unset, list["PoolResponse"]], data)
+
+        pools = _parse_pools(d.pop("pools", UNSET))
+
         machine_response = cls(
             fingerprint=fingerprint,
             id=id,
@@ -222,6 +266,7 @@ class MachineResponse:
             os_info=os_info,
             user_id=user_id,
             organization_id=organization_id,
+            pools=pools,
         )
 
         machine_response.additional_properties = d

openapi_client/cyberdesk_cloud_client/models/paginated_response_pool_response.py

@@ -0,0 +1,97 @@
+from collections.abc import Mapping
+from typing import TYPE_CHECKING, Any, TypeVar
+
+from attrs import define as _attrs_define
+from attrs import field as _attrs_field
+
+if TYPE_CHECKING:
+    from ..models.pool_response import PoolResponse
+
+
+T = TypeVar("T", bound="PaginatedResponsePoolResponse")
+
+
+@_attrs_define
+class PaginatedResponsePoolResponse:
+    """
+    Attributes:
+        items (list['PoolResponse']):
+        total (int):
+        skip (int):
+        limit (int):
+    """
+
+    items: list["PoolResponse"]
+    total: int
+    skip: int
+    limit: int
+    additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
+
+    def to_dict(self) -> dict[str, Any]:
+        items = []
+        for items_item_data in self.items:
+            items_item = items_item_data.to_dict()
+            items.append(items_item)
+
+        total = self.total
+
+        skip = self.skip
+
+        limit = self.limit
+
+        field_dict: dict[str, Any] = {}
+        field_dict.update(self.additional_properties)
+        field_dict.update(
+            {
+                "items": items,
+                "total": total,
+                "skip": skip,
+                "limit": limit,
+            }
+        )
+
+        return field_dict
+
+    @classmethod
+    def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T:
+        from ..models.pool_response import PoolResponse
+
+        d = dict(src_dict)
+        items = []
+        _items = d.pop("items")
+        for items_item_data in _items:
+            items_item = PoolResponse.from_dict(items_item_data)
+
+            items.append(items_item)
+
+        total = d.pop("total")
+
+        skip = d.pop("skip")
+
+        limit = d.pop("limit")
+
+        paginated_response_pool_response = cls(
+            items=items,
+            total=total,
+            skip=skip,
+            limit=limit,
+        )
+
+        paginated_response_pool_response.additional_properties = d
+        return paginated_response_pool_response
+
+    @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/pool_create.py

@@ -0,0 +1,82 @@
+from collections.abc import Mapping
+from typing import Any, TypeVar, Union, cast
+
+from attrs import define as _attrs_define
+from attrs import field as _attrs_field
+
+from ..types import UNSET, Unset
+
+T = TypeVar("T", bound="PoolCreate")
+
+
+@_attrs_define
+class PoolCreate:
+    """Schema for creating a pool
+
+    Attributes:
+        name (str):
+        description (Union[None, Unset, str]):
+    """
+
+    name: str
+    description: Union[None, Unset, str] = UNSET
+    additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
+
+    def to_dict(self) -> dict[str, Any]:
+        name = self.name
+
+        description: Union[None, Unset, str]
+        if isinstance(self.description, Unset):
+            description = UNSET
+        else:
+            description = self.description
+
+        field_dict: dict[str, Any] = {}
+        field_dict.update(self.additional_properties)
+        field_dict.update(
+            {
+                "name": name,
+            }
+        )
+        if description is not UNSET:
+            field_dict["description"] = description
+
+        return field_dict
+
+    @classmethod
+    def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T:
+        d = dict(src_dict)
+        name = d.pop("name")
+
+        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))
+
+        pool_create = cls(
+            name=name,
+            description=description,
+        )
+
+        pool_create.additional_properties = d
+        return pool_create
+
+    @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