robosystems-client 0.1.12__py3-none-any.whl → 0.1.14__py3-none-any.whl

robosystems_client/api/billing/upgrade_graph_subscription_v1_graph_id_billing_subscription_upgrade_post.py

@@ -1,216 +0,0 @@
-from http import HTTPStatus
-from typing import Any, Optional, Union
-
-import httpx
-
-from ... import errors
-from ...client import AuthenticatedClient, Client
-from ...models.http_validation_error import HTTPValidationError
-from ...models.upgrade_subscription_request import UpgradeSubscriptionRequest
-from ...types import UNSET, Response, Unset
-
-
-def _get_kwargs(
-  graph_id: str,
-  *,
-  body: UpgradeSubscriptionRequest,
-  authorization: Union[None, Unset, str] = UNSET,
-  auth_token: Union[None, Unset, str] = UNSET,
-) -> dict[str, Any]:
-  headers: dict[str, Any] = {}
-  if not isinstance(authorization, Unset):
-    headers["authorization"] = authorization
-
-  cookies = {}
-  if auth_token is not UNSET:
-    cookies["auth-token"] = auth_token
-
-  _kwargs: dict[str, Any] = {
-    "method": "post",
-    "url": f"/v1/{graph_id}/billing/subscription/upgrade",
-    "cookies": cookies,
-  }
-
-  _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[Any, HTTPValidationError]]:
-  if response.status_code == 200:
-    response_200 = 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[Any, HTTPValidationError]]:
-  return Response(
-    status_code=HTTPStatus(response.status_code),
-    content=response.content,
-    headers=response.headers,
-    parsed=_parse_response(client=client, response=response),
-  )
-
-
-def sync_detailed(
-  graph_id: str,
-  *,
-  client: AuthenticatedClient,
-  body: UpgradeSubscriptionRequest,
-  authorization: Union[None, Unset, str] = UNSET,
-  auth_token: Union[None, Unset, str] = UNSET,
-) -> Response[Union[Any, HTTPValidationError]]:
-  """Upgrade Graph Subscription
-
-   Upgrade subscription for a specific graph database.
-
-  Args:
-      graph_id (str): Graph database ID
-      authorization (Union[None, Unset, str]):
-      auth_token (Union[None, Unset, str]):
-      body (UpgradeSubscriptionRequest): Request to upgrade a graph database subscription.
-
-  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[Any, HTTPValidationError]]
-  """
-
-  kwargs = _get_kwargs(
-    graph_id=graph_id,
-    body=body,
-    authorization=authorization,
-    auth_token=auth_token,
-  )
-
-  response = client.get_httpx_client().request(
-    **kwargs,
-  )
-
-  return _build_response(client=client, response=response)
-
-
-def sync(
-  graph_id: str,
-  *,
-  client: AuthenticatedClient,
-  body: UpgradeSubscriptionRequest,
-  authorization: Union[None, Unset, str] = UNSET,
-  auth_token: Union[None, Unset, str] = UNSET,
-) -> Optional[Union[Any, HTTPValidationError]]:
-  """Upgrade Graph Subscription
-
-   Upgrade subscription for a specific graph database.
-
-  Args:
-      graph_id (str): Graph database ID
-      authorization (Union[None, Unset, str]):
-      auth_token (Union[None, Unset, str]):
-      body (UpgradeSubscriptionRequest): Request to upgrade a graph database subscription.
-
-  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[Any, HTTPValidationError]
-  """
-
-  return sync_detailed(
-    graph_id=graph_id,
-    client=client,
-    body=body,
-    authorization=authorization,
-    auth_token=auth_token,
-  ).parsed
-
-
-async def asyncio_detailed(
-  graph_id: str,
-  *,
-  client: AuthenticatedClient,
-  body: UpgradeSubscriptionRequest,
-  authorization: Union[None, Unset, str] = UNSET,
-  auth_token: Union[None, Unset, str] = UNSET,
-) -> Response[Union[Any, HTTPValidationError]]:
-  """Upgrade Graph Subscription
-
-   Upgrade subscription for a specific graph database.
-
-  Args:
-      graph_id (str): Graph database ID
-      authorization (Union[None, Unset, str]):
-      auth_token (Union[None, Unset, str]):
-      body (UpgradeSubscriptionRequest): Request to upgrade a graph database subscription.
-
-  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[Any, HTTPValidationError]]
-  """
-
-  kwargs = _get_kwargs(
-    graph_id=graph_id,
-    body=body,
-    authorization=authorization,
-    auth_token=auth_token,
-  )
-
-  response = await client.get_async_httpx_client().request(**kwargs)
-
-  return _build_response(client=client, response=response)
-
-
-async def asyncio(
-  graph_id: str,
-  *,
-  client: AuthenticatedClient,
-  body: UpgradeSubscriptionRequest,
-  authorization: Union[None, Unset, str] = UNSET,
-  auth_token: Union[None, Unset, str] = UNSET,
-) -> Optional[Union[Any, HTTPValidationError]]:
-  """Upgrade Graph Subscription
-
-   Upgrade subscription for a specific graph database.
-
-  Args:
-      graph_id (str): Graph database ID
-      authorization (Union[None, Unset, str]):
-      auth_token (Union[None, Unset, str]):
-      body (UpgradeSubscriptionRequest): Request to upgrade a graph database subscription.
-
-  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[Any, HTTPValidationError]
-  """
-
-  return (
-    await asyncio_detailed(
-      graph_id=graph_id,
-      client=client,
-      body=body,
-      authorization=authorization,
-      auth_token=auth_token,
-    )
-  ).parsed

robosystems_client/models/backup_export_request.py

@@ -1,72 +0,0 @@
-from collections.abc import Mapping
-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="BackupExportRequest")
-
-
-@_attrs_define
-class BackupExportRequest:
-  """Request model for exporting a backup.
-
-  Attributes:
-      backup_id (str): ID of backup to export
-      export_format (Union[Unset, str]): Export format - only 'original' is supported (compressed .kuzu file) Default:
-          'original'.
-  """
-
-  backup_id: str
-  export_format: Union[Unset, str] = "original"
-  additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
-
-  def to_dict(self) -> dict[str, Any]:
-    backup_id = self.backup_id
-
-    export_format = self.export_format
-
-    field_dict: dict[str, Any] = {}
-    field_dict.update(self.additional_properties)
-    field_dict.update(
-      {
-        "backup_id": backup_id,
-      }
-    )
-    if export_format is not UNSET:
-      field_dict["export_format"] = export_format
-
-    return field_dict
-
-  @classmethod
-  def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T:
-    d = dict(src_dict)
-    backup_id = d.pop("backup_id")
-
-    export_format = d.pop("export_format", UNSET)
-
-    backup_export_request = cls(
-      backup_id=backup_id,
-      export_format=export_format,
-    )
-
-    backup_export_request.additional_properties = d
-    return backup_export_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

robosystems_client/models/credit_check_request.py

@@ -1,82 +0,0 @@
-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="CreditCheckRequest")
-
-
-@_attrs_define
-class CreditCheckRequest:
-  """Request to check credit balance.
-
-  Attributes:
-      operation_type (str): Type of operation to check
-      base_cost (Union[None, Unset, float]): Custom base cost (uses default if not provided)
-  """
-
-  operation_type: str
-  base_cost: Union[None, Unset, float] = UNSET
-  additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
-
-  def to_dict(self) -> dict[str, Any]:
-    operation_type = self.operation_type
-
-    base_cost: Union[None, Unset, float]
-    if isinstance(self.base_cost, Unset):
-      base_cost = UNSET
-    else:
-      base_cost = self.base_cost
-
-    field_dict: dict[str, Any] = {}
-    field_dict.update(self.additional_properties)
-    field_dict.update(
-      {
-        "operation_type": operation_type,
-      }
-    )
-    if base_cost is not UNSET:
-      field_dict["base_cost"] = base_cost
-
-    return field_dict
-
-  @classmethod
-  def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T:
-    d = dict(src_dict)
-    operation_type = d.pop("operation_type")
-
-    def _parse_base_cost(data: object) -> Union[None, Unset, float]:
-      if data is None:
-        return data
-      if isinstance(data, Unset):
-        return data
-      return cast(Union[None, Unset, float], data)
-
-    base_cost = _parse_base_cost(d.pop("base_cost", UNSET))
-
-    credit_check_request = cls(
-      operation_type=operation_type,
-      base_cost=base_cost,
-    )
-
-    credit_check_request.additional_properties = d
-    return credit_check_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

robosystems_client/models/upgrade_subscription_request.py

@@ -1,82 +0,0 @@
-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="UpgradeSubscriptionRequest")
-
-
-@_attrs_define
-class UpgradeSubscriptionRequest:
-  """Request to upgrade a graph database subscription.
-
-  Attributes:
-      plan_name (str):
-      payment_method_id (Union[None, Unset, str]):
-  """
-
-  plan_name: str
-  payment_method_id: Union[None, Unset, str] = UNSET
-  additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
-
-  def to_dict(self) -> dict[str, Any]:
-    plan_name = self.plan_name
-
-    payment_method_id: Union[None, Unset, str]
-    if isinstance(self.payment_method_id, Unset):
-      payment_method_id = UNSET
-    else:
-      payment_method_id = self.payment_method_id
-
-    field_dict: dict[str, Any] = {}
-    field_dict.update(self.additional_properties)
-    field_dict.update(
-      {
-        "plan_name": plan_name,
-      }
-    )
-    if payment_method_id is not UNSET:
-      field_dict["payment_method_id"] = payment_method_id
-
-    return field_dict
-
-  @classmethod
-  def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T:
-    d = dict(src_dict)
-    plan_name = d.pop("plan_name")
-
-    def _parse_payment_method_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)
-
-    payment_method_id = _parse_payment_method_id(d.pop("payment_method_id", UNSET))
-
-    upgrade_subscription_request = cls(
-      plan_name=plan_name,
-      payment_method_id=payment_method_id,
-    )
-
-    upgrade_subscription_request.additional_properties = d
-    return upgrade_subscription_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