blaxel 0.1.9rc36__py3-none-any.whl → 0.1.10__py3-none-any.whl

blaxel/sandbox/client/client.py

@@ -0,0 +1,162 @@
+import ssl
+from typing import Any, Optional, Union
+
+import httpx
+from attrs import define, evolve, field
+
+
+@define
+class Client:
+    """A Client which has been authenticated for use on secured endpoints
+
+    The following are accepted as keyword arguments and will be used to construct httpx Clients internally:
+
+        ``base_url``: The base URL for the API, all requests are made to a relative path to this URL
+
+        ``cookies``: A dictionary of cookies to be sent with every request
+
+        ``headers``: A dictionary of headers to be sent with every request
+
+        ``auth``: An implementation of httpx.Auth to use for authentication
+
+        ``timeout``: The maximum amount of a time a request can take. API functions will raise
+        httpx.TimeoutException if this is exceeded.
+
+        ``verify_ssl``: Whether or not to verify the SSL certificate of the API server. This should be True in production,
+        but can be set to False for testing purposes.
+
+        ``follow_redirects``: Whether or not to follow redirects. Default value is False.
+
+        ``httpx_args``: A dictionary of additional arguments to be passed to the ``httpx.Client`` and ``httpx.AsyncClient`` constructor.
+
+
+    Attributes:
+        raise_on_unexpected_status: Whether or not to raise an errors.UnexpectedStatus if the API returns a
+            status code that was not documented in the source OpenAPI document. Can also be provided as a keyword
+            argument to the constructor.
+        auth: Auth to use for authentication
+    """
+
+    raise_on_unexpected_status: bool = field(default=True, kw_only=True)
+    _base_url: str = field(alias="base_url", default="")
+    _cookies: dict[str, str] = field(factory=dict, kw_only=True, alias="cookies")
+    _headers: dict[str, str] = field(factory=dict, kw_only=True, alias="headers")
+    _auth: httpx.Auth = field(default=None, alias="auth")
+    _timeout: Optional[httpx.Timeout] = field(default=None, kw_only=True, alias="timeout")
+    _verify_ssl: Union[str, bool, ssl.SSLContext] = field(default=True, kw_only=True, alias="verify_ssl")
+    _follow_redirects: bool = field(default=False, kw_only=True, alias="follow_redirects")
+    _httpx_args: dict[str, Any] = field(factory=dict, kw_only=True, alias="httpx_args")
+    _client: Optional[httpx.Client] = field(default=None, init=False)
+    _async_client: Optional[httpx.AsyncClient] = field(default=None, init=False)
+
+    def with_base_url(self, base_url: str) -> "Client":
+        """Get a new client matching this one with a new base URL"""
+        self._base_url = base_url
+        if self._client is not None:
+            self._client.base_url = base_url
+        if self._async_client is not None:
+            self._async_client.base_url = base_url
+        return evolve(self, base_url=base_url)
+
+    def with_headers(self, headers: dict[str, str]) -> "Client":
+        """Get a new client matching this one with additional headers"""
+        self._headers = headers
+        if self._client is not None:
+            self._client.headers.update(headers)
+        if self._async_client is not None:
+            self._async_client.headers.update(headers)
+        return evolve(self, headers={**self._headers, **headers})
+
+    def with_cookies(self, cookies: dict[str, str]) -> "Client":
+        """Get a new client matching this one with additional cookies"""
+        self._cookies = cookies
+        if self._client is not None:
+            self._client.cookies.update(cookies)
+        if self._async_client is not None:
+            self._async_client.cookies.update(cookies)
+        return evolve(self, cookies={**self._cookies, **cookies})
+
+    def with_timeout(self, timeout: httpx.Timeout) -> "Client":
+        """Get a new client matching this one with a new timeout (in seconds)"""
+        self._timeout = timeout
+        if self._client is not None:
+            self._client.timeout = timeout
+        if self._async_client is not None:
+            self._async_client.timeout = timeout
+        return evolve(self, timeout=timeout)
+
+    def with_auth(self, auth: httpx.Auth) -> "Client":
+        """Get a new client matching this one with a new provider"""
+        self._auth = auth
+        if self._client is not None:
+            self._client.auth = auth
+        if self._async_client is not None:
+            self._async_client.auth = auth
+        return evolve(self, auth=auth)
+
+    def set_httpx_client(self, client: httpx.Client) -> "Client":
+        """Manually set the underlying httpx.Client
+
+        **NOTE**: This will override any other settings on the client, including cookies, headers, and timeout.
+        """
+        self._client = client
+        return self
+
+    def get_httpx_client(self) -> httpx.Client:
+        """Get the underlying httpx.Client, constructing a new one if not previously set"""
+        if self._client is None:
+            self._client = httpx.Client(
+                base_url=self._base_url,
+                cookies=self._cookies,
+                headers=self._headers,
+                timeout=self._timeout,
+                verify=self._verify_ssl,
+                follow_redirects=self._follow_redirects,
+                auth=self._auth,
+                **self._httpx_args,
+            )
+        return self._client
+
+    def __enter__(self) -> "Client":
+        """Enter a context manager for self.client—you cannot enter twice (see httpx docs)"""
+        self.get_httpx_client().__enter__()
+        return self
+
+    def __exit__(self, *args: Any, **kwargs: Any) -> None:
+        """Exit a context manager for internal httpx.Client (see httpx docs)"""
+        self.get_httpx_client().__exit__(*args, **kwargs)
+
+    def set_async_httpx_client(self, async_client: httpx.AsyncClient) -> "Client":
+        """Manually the underlying httpx.AsyncClient
+
+        **NOTE**: This will override any other settings on the client, including cookies, headers, and timeout.
+        """
+        self._async_client = async_client
+        return self
+
+    def get_async_httpx_client(self) -> httpx.AsyncClient:
+        """Get the underlying httpx.AsyncClient, constructing a new one if not previously set"""
+        if self._async_client is None:
+            self._async_client = httpx.AsyncClient(
+                base_url=self._base_url,
+                cookies=self._cookies,
+                headers=self._headers,
+                timeout=self._timeout,
+                verify=self._verify_ssl,
+                follow_redirects=self._follow_redirects,
+                auth=self._auth,
+                **self._httpx_args,
+            )
+        return self._async_client
+
+    async def __aenter__(self) -> "Client":
+        """Enter a context manager for underlying httpx.AsyncClient—you cannot enter twice (see httpx docs)"""
+        await self.get_async_httpx_client().__aenter__()
+        return self
+
+    async def __aexit__(self, *args: Any, **kwargs: Any) -> None:
+        """Exit a context manager for underlying httpx.AsyncClient (see httpx docs)"""
+        await self.get_async_httpx_client().__aexit__(*args, **kwargs)
+
+
+client = Client()

blaxel/sandbox/client/errors.py

@@ -0,0 +1,16 @@
+"""Contains shared errors types that can be raised from API functions"""
+
+
+class UnexpectedStatus(Exception):
+    """Raised by api functions when the response status an undocumented status and Client.raise_on_unexpected_status is True"""
+
+    def __init__(self, status_code: int, content: bytes):
+        self.status_code = status_code
+        self.content = content
+
+        super().__init__(
+            f"Unexpected status code: {status_code}\n\nResponse content:\n{content.decode(errors='ignore')}"
+        )
+
+
+__all__ = ["UnexpectedStatus"]

blaxel/sandbox/client/models/__init__.py

@@ -0,0 +1,41 @@
+"""Contains all the data models used in inputs/outputs"""
+
+from .delete_network_process_pid_monitor_response_200 import (
+    DeleteNetworkProcessPidMonitorResponse200,
+)
+from .directory import Directory
+from .error_response import ErrorResponse
+from .file import File
+from .file_request import FileRequest
+from .file_with_content import FileWithContent
+from .get_network_process_pid_ports_response_200 import GetNetworkProcessPidPortsResponse200
+from .get_process_identifier_logs_response_200 import GetProcessIdentifierLogsResponse200
+from .get_process_identifier_logs_stream_response_200 import (
+    GetProcessIdentifierLogsStreamResponse200,
+)
+from .port_monitor_request import PortMonitorRequest
+from .post_network_process_pid_monitor_response_200 import PostNetworkProcessPidMonitorResponse200
+from .process_kill_request import ProcessKillRequest
+from .process_request import ProcessRequest
+from .process_response import ProcessResponse
+from .subdirectory import Subdirectory
+from .success_response import SuccessResponse
+
+__all__ = (
+    "DeleteNetworkProcessPidMonitorResponse200",
+    "Directory",
+    "ErrorResponse",
+    "File",
+    "FileRequest",
+    "FileWithContent",
+    "GetNetworkProcessPidPortsResponse200",
+    "GetProcessIdentifierLogsResponse200",
+    "GetProcessIdentifierLogsStreamResponse200",
+    "PortMonitorRequest",
+    "PostNetworkProcessPidMonitorResponse200",
+    "ProcessKillRequest",
+    "ProcessRequest",
+    "ProcessResponse",
+    "Subdirectory",
+    "SuccessResponse",
+)

blaxel/sandbox/client/models/delete_network_process_pid_monitor_response_200.py

@@ -0,0 +1,45 @@
+from typing import Any, TypeVar
+
+from attrs import define as _attrs_define
+from attrs import field as _attrs_field
+
+T = TypeVar("T", bound="DeleteNetworkProcessPidMonitorResponse200")
+
+
+@_attrs_define
+class DeleteNetworkProcessPidMonitorResponse200:
+    """ """
+
+    additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
+
+    def to_dict(self) -> dict[str, Any]:
+        field_dict: dict[str, Any] = {}
+        field_dict.update(self.additional_properties)
+
+        return field_dict
+
+    @classmethod
+    def from_dict(cls: type[T], src_dict: dict[str, Any]) -> T:
+        if not src_dict:
+            return None
+        d = src_dict.copy()
+        delete_network_process_pid_monitor_response_200 = cls()
+
+        delete_network_process_pid_monitor_response_200.additional_properties = d
+        return delete_network_process_pid_monitor_response_200
+
+    @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

blaxel/sandbox/client/models/directory.py

@@ -0,0 +1,112 @@
+from typing import TYPE_CHECKING, Any, TypeVar, Union
+
+from attrs import define as _attrs_define
+from attrs import field as _attrs_field
+
+from ..types import UNSET, Unset
+
+if TYPE_CHECKING:
+    from ..models.file import File
+    from ..models.subdirectory import Subdirectory
+
+
+T = TypeVar("T", bound="Directory")
+
+
+@_attrs_define
+class Directory:
+    """
+    Attributes:
+        files (Union[Unset, list['File']]):
+        path (Union[Unset, str]):
+        subdirectories (Union[Unset, list['Subdirectory']]): @name Subdirectories
+    """
+
+    files: Union[Unset, list["File"]] = UNSET
+    path: Union[Unset, str] = UNSET
+    subdirectories: Union[Unset, list["Subdirectory"]] = UNSET
+    additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
+
+    def to_dict(self) -> dict[str, Any]:
+        files: Union[Unset, list[dict[str, Any]]] = UNSET
+        if not isinstance(self.files, Unset):
+            files = []
+            for files_item_data in self.files:
+                if type(files_item_data) == dict:
+                    files_item = files_item_data
+                else:
+                    files_item = files_item_data.to_dict()
+                files.append(files_item)
+
+        path = self.path
+
+        subdirectories: Union[Unset, list[dict[str, Any]]] = UNSET
+        if not isinstance(self.subdirectories, Unset):
+            subdirectories = []
+            for subdirectories_item_data in self.subdirectories:
+                if type(subdirectories_item_data) == dict:
+                    subdirectories_item = subdirectories_item_data
+                else:
+                    subdirectories_item = subdirectories_item_data.to_dict()
+                subdirectories.append(subdirectories_item)
+
+        field_dict: dict[str, Any] = {}
+        field_dict.update(self.additional_properties)
+        field_dict.update({})
+        if files is not UNSET:
+            field_dict["files"] = files
+        if path is not UNSET:
+            field_dict["path"] = path
+        if subdirectories is not UNSET:
+            field_dict["subdirectories"] = subdirectories
+
+        return field_dict
+
+    @classmethod
+    def from_dict(cls: type[T], src_dict: dict[str, Any]) -> T:
+        from ..models.file import File
+        from ..models.subdirectory import Subdirectory
+
+        if not src_dict:
+            return None
+        d = src_dict.copy()
+        files = []
+        _files = d.pop("files", UNSET)
+        for files_item_data in _files or []:
+            files_item = File.from_dict(files_item_data)
+
+            files.append(files_item)
+
+        path = d.pop("path", UNSET)
+
+        subdirectories = []
+        _subdirectories = d.pop("subdirectories", UNSET)
+        for subdirectories_item_data in _subdirectories or []:
+            subdirectories_item = Subdirectory.from_dict(subdirectories_item_data)
+
+            subdirectories.append(subdirectories_item)
+
+        directory = cls(
+            files=files,
+            path=path,
+            subdirectories=subdirectories,
+        )
+
+        directory.additional_properties = d
+        return directory
+
+    @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

blaxel/sandbox/client/models/error_response.py

@@ -0,0 +1,60 @@
+from typing import Any, TypeVar, Union
+
+from attrs import define as _attrs_define
+from attrs import field as _attrs_field
+
+from ..types import UNSET, Unset
+
+T = TypeVar("T", bound="ErrorResponse")
+
+
+@_attrs_define
+class ErrorResponse:
+    """
+    Attributes:
+        error (Union[Unset, str]):  Example: Error message.
+    """
+
+    error: Union[Unset, str] = UNSET
+    additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
+
+    def to_dict(self) -> dict[str, Any]:
+        error = self.error
+
+        field_dict: dict[str, Any] = {}
+        field_dict.update(self.additional_properties)
+        field_dict.update({})
+        if error is not UNSET:
+            field_dict["error"] = error
+
+        return field_dict
+
+    @classmethod
+    def from_dict(cls: type[T], src_dict: dict[str, Any]) -> T:
+        if not src_dict:
+            return None
+        d = src_dict.copy()
+        error = d.pop("error", UNSET)
+
+        error_response = cls(
+            error=error,
+        )
+
+        error_response.additional_properties = d
+        return error_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

blaxel/sandbox/client/models/file.py

@@ -0,0 +1,105 @@
+from typing import Any, TypeVar, Union
+
+from attrs import define as _attrs_define
+from attrs import field as _attrs_field
+
+from ..types import UNSET, Unset
+
+T = TypeVar("T", bound="File")
+
+
+@_attrs_define
+class File:
+    """
+    Attributes:
+        group (Union[Unset, str]):
+        last_modified (Union[Unset, str]):
+        owner (Union[Unset, str]):
+        path (Union[Unset, str]):
+        permissions (Union[Unset, str]):
+        size (Union[Unset, int]):
+    """
+
+    group: Union[Unset, str] = UNSET
+    last_modified: Union[Unset, str] = UNSET
+    owner: Union[Unset, str] = UNSET
+    path: Union[Unset, str] = UNSET
+    permissions: Union[Unset, str] = UNSET
+    size: Union[Unset, int] = UNSET
+    additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
+
+    def to_dict(self) -> dict[str, Any]:
+        group = self.group
+
+        last_modified = self.last_modified
+
+        owner = self.owner
+
+        path = self.path
+
+        permissions = self.permissions
+
+        size = self.size
+
+        field_dict: dict[str, Any] = {}
+        field_dict.update(self.additional_properties)
+        field_dict.update({})
+        if group is not UNSET:
+            field_dict["group"] = group
+        if last_modified is not UNSET:
+            field_dict["lastModified"] = last_modified
+        if owner is not UNSET:
+            field_dict["owner"] = owner
+        if path is not UNSET:
+            field_dict["path"] = path
+        if permissions is not UNSET:
+            field_dict["permissions"] = permissions
+        if size is not UNSET:
+            field_dict["size"] = size
+
+        return field_dict
+
+    @classmethod
+    def from_dict(cls: type[T], src_dict: dict[str, Any]) -> T:
+        if not src_dict:
+            return None
+        d = src_dict.copy()
+        group = d.pop("group", UNSET)
+
+        last_modified = d.pop("lastModified", UNSET)
+
+        owner = d.pop("owner", UNSET)
+
+        path = d.pop("path", UNSET)
+
+        permissions = d.pop("permissions", UNSET)
+
+        size = d.pop("size", UNSET)
+
+        file = cls(
+            group=group,
+            last_modified=last_modified,
+            owner=owner,
+            path=path,
+            permissions=permissions,
+            size=size,
+        )
+
+        file.additional_properties = d
+        return file
+
+    @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

blaxel/sandbox/client/models/file_request.py

@@ -0,0 +1,78 @@
+from typing import Any, TypeVar, Union
+
+from attrs import define as _attrs_define
+from attrs import field as _attrs_field
+
+from ..types import UNSET, Unset
+
+T = TypeVar("T", bound="FileRequest")
+
+
+@_attrs_define
+class FileRequest:
+    """
+    Attributes:
+        content (Union[Unset, str]):  Example: file contents here.
+        is_directory (Union[Unset, bool]):
+        permissions (Union[Unset, str]):  Example: 0644.
+    """
+
+    content: Union[Unset, str] = UNSET
+    is_directory: Union[Unset, bool] = UNSET
+    permissions: Union[Unset, str] = UNSET
+    additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
+
+    def to_dict(self) -> dict[str, Any]:
+        content = self.content
+
+        is_directory = self.is_directory
+
+        permissions = self.permissions
+
+        field_dict: dict[str, Any] = {}
+        field_dict.update(self.additional_properties)
+        field_dict.update({})
+        if content is not UNSET:
+            field_dict["content"] = content
+        if is_directory is not UNSET:
+            field_dict["isDirectory"] = is_directory
+        if permissions is not UNSET:
+            field_dict["permissions"] = permissions
+
+        return field_dict
+
+    @classmethod
+    def from_dict(cls: type[T], src_dict: dict[str, Any]) -> T:
+        if not src_dict:
+            return None
+        d = src_dict.copy()
+        content = d.pop("content", UNSET)
+
+        is_directory = d.pop("isDirectory", UNSET)
+
+        permissions = d.pop("permissions", UNSET)
+
+        file_request = cls(
+            content=content,
+            is_directory=is_directory,
+            permissions=permissions,
+        )
+
+        file_request.additional_properties = d
+        return file_request
+
+    @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