cirro-api-client 0.2.2__tar.gz → 0.2.3__tar.gz

{cirro_api_client-0.2.2 → cirro_api_client-0.2.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: cirro_api_client
-Version: 0.2.2
+Version: 0.2.3
 Summary: A client library for accessing Cirro
 Home-page: https://github.com/CirroBio/Cirro-client-python
 License: MIT

cirro_api_client-0.2.3/cirro_api_client/v1/api/compute_environment/create_compute_environment.py

@@ -0,0 +1,186 @@
+from http import HTTPStatus
+from typing import Any, Dict, Optional
+
+import httpx
+
+from ... import errors
+from ...client import Client
+from ...models.compute_environment_configuration_input import ComputeEnvironmentConfigurationInput
+from ...models.create_response import CreateResponse
+from ...types import Response
+
+
+def _get_kwargs(
+    project_id: str,
+    *,
+    body: ComputeEnvironmentConfigurationInput,
+) -> Dict[str, Any]:
+    headers: Dict[str, Any] = {}
+
+    _kwargs: Dict[str, Any] = {
+        "method": "post",
+        "url": f"/projects/{project_id}/compute-environments",
+    }
+
+    _body = body.to_dict()
+
+    _kwargs["json"] = _body
+    headers["Content-Type"] = "application/json"
+
+    _kwargs["headers"] = headers
+    return _kwargs
+
+
+def _parse_response(*, client: Client, response: httpx.Response) -> Optional[CreateResponse]:
+    if response.status_code == HTTPStatus.CREATED:
+        response_201 = CreateResponse.from_dict(response.json())
+
+        return response_201
+
+    errors.handle_error_response(response, client.raise_on_unexpected_status)
+
+
+def _build_response(*, client: Client, response: httpx.Response) -> Response[CreateResponse]:
+    return Response(
+        status_code=HTTPStatus(response.status_code),
+        content=response.content,
+        headers=response.headers,
+        parsed=_parse_response(client=client, response=response),
+    )
+
+
+def sync_detailed(
+    project_id: str,
+    *,
+    client: Client,
+    body: ComputeEnvironmentConfigurationInput,
+) -> Response[CreateResponse]:
+    """Create compute environment
+
+     Create a new compute environment for a project
+
+    Args:
+        project_id (str):
+        body (ComputeEnvironmentConfigurationInput):
+        client (Client): instance of the API client
+
+    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[CreateResponse]
+    """
+
+    kwargs = _get_kwargs(
+        project_id=project_id,
+        body=body,
+    )
+
+    response = client.get_httpx_client().request(
+        auth=client.get_auth(),
+        **kwargs,
+    )
+
+    return _build_response(client=client, response=response)
+
+
+def sync(
+    project_id: str,
+    *,
+    client: Client,
+    body: ComputeEnvironmentConfigurationInput,
+) -> Optional[CreateResponse]:
+    """Create compute environment
+
+     Create a new compute environment for a project
+
+    Args:
+        project_id (str):
+        body (ComputeEnvironmentConfigurationInput):
+        client (Client): instance of the API client
+
+    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:
+        CreateResponse
+    """
+
+    try:
+        return sync_detailed(
+            project_id=project_id,
+            client=client,
+            body=body,
+        ).parsed
+    except errors.NotFoundException:
+        return None
+
+
+async def asyncio_detailed(
+    project_id: str,
+    *,
+    client: Client,
+    body: ComputeEnvironmentConfigurationInput,
+) -> Response[CreateResponse]:
+    """Create compute environment
+
+     Create a new compute environment for a project
+
+    Args:
+        project_id (str):
+        body (ComputeEnvironmentConfigurationInput):
+        client (Client): instance of the API client
+
+    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[CreateResponse]
+    """
+
+    kwargs = _get_kwargs(
+        project_id=project_id,
+        body=body,
+    )
+
+    response = await client.get_async_httpx_client().request(auth=client.get_auth(), **kwargs)
+
+    return _build_response(client=client, response=response)
+
+
+async def asyncio(
+    project_id: str,
+    *,
+    client: Client,
+    body: ComputeEnvironmentConfigurationInput,
+) -> Optional[CreateResponse]:
+    """Create compute environment
+
+     Create a new compute environment for a project
+
+    Args:
+        project_id (str):
+        body (ComputeEnvironmentConfigurationInput):
+        client (Client): instance of the API client
+
+    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:
+        CreateResponse
+    """
+
+    try:
+        return (
+            await asyncio_detailed(
+                project_id=project_id,
+                client=client,
+                body=body,
+            )
+        ).parsed
+    except errors.NotFoundException:
+        return None

cirro_api_client-0.2.3/cirro_api_client/v1/api/compute_environment/delete_compute_environment.py

@@ -0,0 +1,105 @@
+from http import HTTPStatus
+from typing import Any, Dict, Optional
+
+import httpx
+
+from ... import errors
+from ...client import Client
+from ...types import Response
+
+
+def _get_kwargs(
+    project_id: str,
+    compute_environment_id: str,
+) -> Dict[str, Any]:
+    _kwargs: Dict[str, Any] = {
+        "method": "delete",
+        "url": f"/projects/{project_id}/compute-environments/{compute_environment_id}",
+    }
+
+    return _kwargs
+
+
+def _parse_response(*, client: Client, response: httpx.Response) -> Optional[Any]:
+    if response.status_code == HTTPStatus.OK:
+        return None
+
+    errors.handle_error_response(response, client.raise_on_unexpected_status)
+
+
+def _build_response(*, client: Client, response: httpx.Response) -> Response[Any]:
+    return Response(
+        status_code=HTTPStatus(response.status_code),
+        content=response.content,
+        headers=response.headers,
+        parsed=_parse_response(client=client, response=response),
+    )
+
+
+def sync_detailed(
+    project_id: str,
+    compute_environment_id: str,
+    *,
+    client: Client,
+) -> Response[Any]:
+    """Delete compute environment
+
+     Delete a compute environment for a project
+
+    Args:
+        project_id (str):
+        compute_environment_id (str):
+        client (Client): instance of the API client
+
+    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[Any]
+    """
+
+    kwargs = _get_kwargs(
+        project_id=project_id,
+        compute_environment_id=compute_environment_id,
+    )
+
+    response = client.get_httpx_client().request(
+        auth=client.get_auth(),
+        **kwargs,
+    )
+
+    return _build_response(client=client, response=response)
+
+
+async def asyncio_detailed(
+    project_id: str,
+    compute_environment_id: str,
+    *,
+    client: Client,
+) -> Response[Any]:
+    """Delete compute environment
+
+     Delete a compute environment for a project
+
+    Args:
+        project_id (str):
+        compute_environment_id (str):
+        client (Client): instance of the API client
+
+    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[Any]
+    """
+
+    kwargs = _get_kwargs(
+        project_id=project_id,
+        compute_environment_id=compute_environment_id,
+    )
+
+    response = await client.get_async_httpx_client().request(auth=client.get_auth(), **kwargs)
+
+    return _build_response(client=client, response=response)

cirro_api_client-0.2.3/cirro_api_client/v1/api/compute_environment/update_compute_environment.py

@@ -0,0 +1,122 @@
+from http import HTTPStatus
+from typing import Any, Dict, Optional
+
+import httpx
+
+from ... import errors
+from ...client import Client
+from ...models.compute_environment_configuration_input import ComputeEnvironmentConfigurationInput
+from ...types import Response
+
+
+def _get_kwargs(
+    project_id: str,
+    compute_environment_id: str,
+    *,
+    body: ComputeEnvironmentConfigurationInput,
+) -> Dict[str, Any]:
+    headers: Dict[str, Any] = {}
+
+    _kwargs: Dict[str, Any] = {
+        "method": "put",
+        "url": f"/projects/{project_id}/compute-environments/{compute_environment_id}",
+    }
+
+    _body = body.to_dict()
+
+    _kwargs["json"] = _body
+    headers["Content-Type"] = "application/json"
+
+    _kwargs["headers"] = headers
+    return _kwargs
+
+
+def _parse_response(*, client: Client, response: httpx.Response) -> Optional[Any]:
+    if response.status_code == HTTPStatus.OK:
+        return None
+
+    errors.handle_error_response(response, client.raise_on_unexpected_status)
+
+
+def _build_response(*, client: Client, response: httpx.Response) -> Response[Any]:
+    return Response(
+        status_code=HTTPStatus(response.status_code),
+        content=response.content,
+        headers=response.headers,
+        parsed=_parse_response(client=client, response=response),
+    )
+
+
+def sync_detailed(
+    project_id: str,
+    compute_environment_id: str,
+    *,
+    client: Client,
+    body: ComputeEnvironmentConfigurationInput,
+) -> Response[Any]:
+    """Update compute environment
+
+     Update a compute environment for a project
+
+    Args:
+        project_id (str):
+        compute_environment_id (str):
+        body (ComputeEnvironmentConfigurationInput):
+        client (Client): instance of the API client
+
+    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[Any]
+    """
+
+    kwargs = _get_kwargs(
+        project_id=project_id,
+        compute_environment_id=compute_environment_id,
+        body=body,
+    )
+
+    response = client.get_httpx_client().request(
+        auth=client.get_auth(),
+        **kwargs,
+    )
+
+    return _build_response(client=client, response=response)
+
+
+async def asyncio_detailed(
+    project_id: str,
+    compute_environment_id: str,
+    *,
+    client: Client,
+    body: ComputeEnvironmentConfigurationInput,
+) -> Response[Any]:
+    """Update compute environment
+
+     Update a compute environment for a project
+
+    Args:
+        project_id (str):
+        compute_environment_id (str):
+        body (ComputeEnvironmentConfigurationInput):
+        client (Client): instance of the API client
+
+    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[Any]
+    """
+
+    kwargs = _get_kwargs(
+        project_id=project_id,
+        compute_environment_id=compute_environment_id,
+        body=body,
+    )
+
+    response = await client.get_async_httpx_client().request(auth=client.get_auth(), **kwargs)
+
+    return _build_response(client=client, response=response)

{cirro_api_client-0.2.2 → cirro_api_client-0.2.3}/cirro_api_client/v1/models/__init__.py

@@ -29,6 +29,8 @@ from .cloud_account import CloudAccount
 from .cloud_account_type import CloudAccountType
 from .column_definition import ColumnDefinition
 from .compute_environment_configuration import ComputeEnvironmentConfiguration
+from .compute_environment_configuration_input import ComputeEnvironmentConfigurationInput
+from .compute_environment_configuration_input_properties import ComputeEnvironmentConfigurationInputProperties
 from .compute_environment_configuration_properties import ComputeEnvironmentConfigurationProperties
 from .contact import Contact
 from .create_notebook_instance_request import CreateNotebookInstanceRequest
@@ -158,6 +160,8 @@ __all__ = (
     "CloudAccountType",
     "ColumnDefinition",
     "ComputeEnvironmentConfiguration",
+    "ComputeEnvironmentConfigurationInput",
+    "ComputeEnvironmentConfigurationInputProperties",
     "ComputeEnvironmentConfigurationProperties",
     "Contact",
     "CreateNotebookInstanceRequest",

cirro_api_client-0.2.3/cirro_api_client/v1/models/compute_environment_configuration_input.py

@@ -0,0 +1,112 @@
+from typing import TYPE_CHECKING, Any, Dict, List, Type, TypeVar, Union, cast
+
+from attrs import define as _attrs_define
+from attrs import field as _attrs_field
+
+from ..types import UNSET, Unset
+
+if TYPE_CHECKING:
+    from ..models.compute_environment_configuration_input_properties import (
+        ComputeEnvironmentConfigurationInputProperties,
+    )
+
+
+T = TypeVar("T", bound="ComputeEnvironmentConfigurationInput")
+
+
+@_attrs_define
+class ComputeEnvironmentConfigurationInput:
+    """
+    Attributes:
+        name (str):
+        agent_id (Union[None, Unset, str]):
+        properties (Union['ComputeEnvironmentConfigurationInputProperties', None, Unset]):
+    """
+
+    name: str
+    agent_id: Union[None, Unset, str] = UNSET
+    properties: Union["ComputeEnvironmentConfigurationInputProperties", None, Unset] = UNSET
+    additional_properties: Dict[str, Any] = _attrs_field(init=False, factory=dict)
+
+    def to_dict(self) -> Dict[str, Any]:
+        from ..models.compute_environment_configuration_input_properties import (
+            ComputeEnvironmentConfigurationInputProperties,
+        )
+
+        name = self.name
+
+        agent_id: Union[None, Unset, str]
+        if isinstance(self.agent_id, Unset):
+            agent_id = UNSET
+        else:
+            agent_id = self.agent_id
+
+        properties: Union[Dict[str, Any], None, Unset]
+        if isinstance(self.properties, Unset):
+            properties = UNSET
+        elif isinstance(self.properties, ComputeEnvironmentConfigurationInputProperties):
+            properties = self.properties.to_dict()
+        else:
+            properties = self.properties
+
+        field_dict: Dict[str, Any] = {}
+        field_dict.update(self.additional_properties)
+        field_dict.update(
+            {
+                "name": name,
+            }
+        )
+        if agent_id is not UNSET:
+            field_dict["agentId"] = agent_id
+        if properties is not UNSET:
+            field_dict["properties"] = properties
+
+        return field_dict
+
+    @classmethod
+    def from_dict(cls: Type[T], src_dict: Dict[str, Any]) -> T:
+        from ..models.compute_environment_configuration_input_properties import (
+            ComputeEnvironmentConfigurationInputProperties,
+        )
+
+        d = src_dict.copy()
+        name = d.pop("name")
+
+        def _parse_agent_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)
+
+        agent_id = _parse_agent_id(d.pop("agentId", UNSET))
+
+        def _parse_properties(data: object) -> Union["ComputeEnvironmentConfigurationInputProperties", None, Unset]:
+            if data is None:
+                return data
+            if isinstance(data, Unset):
+                return data
+            try:
+                if not isinstance(data, dict):
+                    raise TypeError()
+                properties_type_0 = ComputeEnvironmentConfigurationInputProperties.from_dict(data)
+
+                return properties_type_0
+            except:  # noqa: E722
+                pass
+            return cast(Union["ComputeEnvironmentConfigurationInputProperties", None, Unset], data)
+
+        properties = _parse_properties(d.pop("properties", UNSET))
+
+        compute_environment_configuration_input = cls(
+            name=name,
+            agent_id=agent_id,
+            properties=properties,
+        )
+
+        compute_environment_configuration_input.additional_properties = d
+        return compute_environment_configuration_input
+
+    @property
+    def additional_keys(self) -> List[str]:
+        return list(self.additional_properties.keys())

cirro_api_client-0.2.3/cirro_api_client/v1/models/compute_environment_configuration_input_properties.py

@@ -0,0 +1,32 @@
+from typing import Any, Dict, List, Type, TypeVar
+
+from attrs import define as _attrs_define
+from attrs import field as _attrs_field
+
+T = TypeVar("T", bound="ComputeEnvironmentConfigurationInputProperties")
+
+
+@_attrs_define
+class ComputeEnvironmentConfigurationInputProperties:
+    """ """
+
+    additional_properties: Dict[str, str] = _attrs_field(init=False, factory=dict)
+
+    def to_dict(self) -> Dict[str, Any]:
+        field_dict: Dict[str, Any] = {}
+        field_dict.update(self.additional_properties)
+        field_dict.update({})
+
+        return field_dict
+
+    @classmethod
+    def from_dict(cls: Type[T], src_dict: Dict[str, Any]) -> T:
+        d = src_dict.copy()
+        compute_environment_configuration_input_properties = cls()
+
+        compute_environment_configuration_input_properties.additional_properties = d
+        return compute_environment_configuration_input_properties
+
+    @property
+    def additional_keys(self) -> List[str]:
+        return list(self.additional_properties.keys())

{cirro_api_client-0.2.2 → cirro_api_client-0.2.3}/cirro_api_client/v1/models/task.py

@@ -18,6 +18,7 @@ class Task:
         native_job_id (str):
         status (str):
         requested_at (datetime.datetime):
+        started_at (Union[Unset, datetime.datetime]):
         stopped_at (Union[Unset, datetime.datetime]):
         container_image (Union[Unset, str]):
         command_line (Union[Unset, str]):
@@ -28,6 +29,7 @@ class Task:
     native_job_id: str
     status: str
     requested_at: datetime.datetime
+    started_at: Union[Unset, datetime.datetime] = UNSET
     stopped_at: Union[Unset, datetime.datetime] = UNSET
     container_image: Union[Unset, str] = UNSET
     command_line: Union[Unset, str] = UNSET
@@ -43,6 +45,10 @@ class Task:
 
         requested_at = self.requested_at.isoformat()
 
+        started_at: Union[Unset, str] = UNSET
+        if not isinstance(self.started_at, Unset):
+            started_at = self.started_at.isoformat()
+
         stopped_at: Union[Unset, str] = UNSET
         if not isinstance(self.stopped_at, Unset):
             stopped_at = self.stopped_at.isoformat()
@@ -63,6 +69,8 @@ class Task:
                 "requestedAt": requested_at,
             }
         )
+        if started_at is not UNSET:
+            field_dict["startedAt"] = started_at
         if stopped_at is not UNSET:
             field_dict["stoppedAt"] = stopped_at
         if container_image is not UNSET:
@@ -85,6 +93,13 @@ class Task:
 
         requested_at = isoparse(d.pop("requestedAt"))
 
+        _started_at = d.pop("startedAt", UNSET)
+        started_at: Union[Unset, datetime.datetime]
+        if isinstance(_started_at, Unset):
+            started_at = UNSET
+        else:
+            started_at = isoparse(_started_at)
+
         _stopped_at = d.pop("stoppedAt", UNSET)
         stopped_at: Union[Unset, datetime.datetime]
         if isinstance(_stopped_at, Unset):
@@ -103,6 +118,7 @@ class Task:
             native_job_id=native_job_id,
             status=status,
             requested_at=requested_at,
+            started_at=started_at,
             stopped_at=stopped_at,
             container_image=container_image,
             command_line=command_line,

{cirro_api_client-0.2.2 → cirro_api_client-0.2.3}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "cirro_api_client"
-version = "0.2.2"
+version = "0.2.3"
 description = "A client library for accessing Cirro"
 authors = ["Cirro <support@cirro.bio>"]
 license = "MIT"