anyscale 0.24.88__py3-none-any.whl → 0.25.5__py3-none-any.whl

anyscale/controllers/service_account_controller.py

@@ -1,168 +0,0 @@
-from typing import List, Optional
-
-import click
-from rich import box
-from rich.console import Console
-from rich.table import Table
-
-from anyscale.cli_logger import BlockLogger
-from anyscale.client.openapi_client.models import (
-    AnyscaleServiceAccount,
-    ServerSessionToken,
-)
-from anyscale.controllers.base_controller import BaseController
-
-
-WARNING_COLOR = "bold red"
-DEFAULT_OVERFLOW = "fold"
-DEFAULT_COL_WIDTH = 36
-ONE_HUNDRED_YEARS_IN_SECONDS = 3153600000
-
-
-class ServiceAccountController(BaseController):
-    def __init__(self):
-        super().__init__()
-        self.log = BlockLogger()
-        self.console = Console()
-
-    def _get_service_account_identifier(
-        self, email: Optional[str], name: Optional[str]
-    ) -> str:
-        # Logic can be simplified but kept verbose for clarity and typing
-        if not email and not name:
-            raise click.ClickException("Either email or name must be provided.")
-        if email and name:
-            raise click.ClickException("Only one of email or name can be provided.")
-        if email:
-            return email
-        elif name:
-            return name
-
-        raise click.ClickException("Internal server error. Please contact support.")
-
-    def _validate_exactly_one_service_account_per_email_or_name(
-        self, service_accounts: List[AnyscaleServiceAccount], identifier: str,
-    ):
-        if len(service_accounts) == 0:
-            raise click.ClickException(f"No service account {identifier} found.")
-
-        if len(service_accounts) > 1:
-            raise click.ClickException(
-                f"Internal server error when fetching service account {identifier}. Please contact support."
-            )
-
-    def _print_new_api_key(self, api_key: ServerSessionToken):
-        self.console.print(
-            "The following API token for the service account will only appear once:\n",
-            style=WARNING_COLOR,
-        )
-        self.console.print(api_key.server_session_id)
-
-    def create_service_account(self, name: str):
-        service_account = self.api_client.create_service_account_api_v2_users_service_accounts_post(
-            name=name
-        ).result
-        api_key = self.api_client.create_api_key_api_v2_users_create_api_key_post(
-            api_key_parameters={
-                "user_id": service_account.user_id,
-                "duration": ONE_HUNDRED_YEARS_IN_SECONDS,
-            }
-        ).result
-
-        self.console.print(f"\nService account {name} created successfully.")
-        self._print_new_api_key(api_key)
-
-    def create_new_service_account_api_key(
-        self, email: Optional[str], name: Optional[str]
-    ):
-        service_accounts = (
-            self.api_client.list_organization_collaborators_api_v2_organization_collaborators_get(
-                is_service_account=True, email=email, name=name
-            )
-        ).results
-        self._validate_exactly_one_service_account_per_email_or_name(
-            service_accounts, self._get_service_account_identifier(email, name)
-        )
-        sa = service_accounts[0]
-        api_key = self.api_client.create_api_key_api_v2_users_create_api_key_post(
-            api_key_parameters={
-                "user_id": sa.user_id,
-                "duration": ONE_HUNDRED_YEARS_IN_SECONDS,
-            }
-        ).result
-
-        self._print_new_api_key(api_key)
-
-    def _print_service_account_table(
-        self, service_accounts: List[AnyscaleServiceAccount]
-    ):
-        table = Table(box=box.MINIMAL, header_style="bright_cyan")
-        table.add_column("NAME", width=DEFAULT_COL_WIDTH, overflow=DEFAULT_OVERFLOW)
-        table.add_column(
-            "CREATED AT",
-            style="dim",
-            width=DEFAULT_COL_WIDTH,
-            overflow=DEFAULT_OVERFLOW,
-        )
-        table.add_column(
-            "ORGANIZATION ROLE", width=DEFAULT_COL_WIDTH, overflow=DEFAULT_OVERFLOW
-        )
-        table.add_column("EMAIL", width=80, overflow=DEFAULT_OVERFLOW)
-        for sa in service_accounts:
-            table.add_row(
-                sa.name,
-                sa.created_at.strftime("%Y/%m/%d"),
-                sa.permission_level,
-                sa.email,
-            )
-
-        self.console.print(table)
-
-    def list_service_accounts(self, max_items: int):
-        service_accounts = self.api_client.list_organization_collaborators_api_v2_organization_collaborators_get(
-            is_service_account=True
-        ).results
-
-        self._print_service_account_table(service_accounts[:max_items])
-
-    def delete_service_account(self, email: Optional[str], name: Optional[str]):
-        service_accounts = self.api_client.list_organization_collaborators_api_v2_organization_collaborators_get(
-            is_service_account=True, email=email, name=name
-        ).results
-        identifier = self._get_service_account_identifier(email, name)
-        self._validate_exactly_one_service_account_per_email_or_name(
-            service_accounts, identifier
-        )
-        sa = service_accounts[0]
-        self.api_client.remove_organization_collaborator_api_v2_organization_collaborators_identity_id_delete(
-            identity_id=sa.id
-        )
-        self.console.print(f"Service account {identifier} deleted successfully.")
-
-    def rotate_service_account_api_keys(
-        self, email: Optional[str], name: Optional[str]
-    ):
-        service_accounts = (
-            self.api_client.list_organization_collaborators_api_v2_organization_collaborators_get(
-                is_service_account=True, email=email, name=name
-            )
-        ).results
-
-        identifier = self._get_service_account_identifier(email, name)
-        self._validate_exactly_one_service_account_per_email_or_name(
-            service_accounts, identifier
-        )
-        sa = service_accounts[0]
-        self.api_client.rotate_api_key_for_user_api_v2_organization_collaborators_rotate_api_key_for_user_user_id_post(
-            sa.user_id
-        )
-        api_key = self.api_client.create_api_key_api_v2_users_create_api_key_post(
-            api_key_parameters={
-                "user_id": sa.user_id,
-                "duration": ONE_HUNDRED_YEARS_IN_SECONDS,
-            }
-        ).result
-        self.console.print(
-            f"\nAll API keys for service account {identifier} rotated successfully."
-        )
-        self._print_new_api_key(api_key)

anyscale/sdk/anyscale_client/models/log_level_types.py

@@ -1,100 +0,0 @@
-# coding: utf-8
-
-"""
-    Anyscale API
-
-    No description provided (generated by Openapi Generator https://github.com/openapitools/openapi-generator)  # noqa: E501
-
-    The version of the OpenAPI document: 0.1.0
-    Generated by: https://openapi-generator.tech
-"""
-
-
-import pprint
-import re  # noqa: F401
-
-import six
-
-from anyscale_client.configuration import Configuration
-
-
-class LogLevelTypes(object):
-    """NOTE: This class is auto generated by OpenAPI Generator.
-    Ref: https://openapi-generator.tech
-
-    Do not edit the class manually.
-    """
-
-    """
-    allowed enum values
-    """
-    INFO = "info"
-    ERROR = "error"
-
-    allowable_values = [INFO, ERROR]  # noqa: E501
-
-    """
-    Attributes:
-      openapi_types (dict): The key is attribute name
-                            and the value is attribute type.
-      attribute_map (dict): The key is attribute name
-                            and the value is json key in definition.
-    """
-    openapi_types = {
-    }
-
-    attribute_map = {
-    }
-
-    def __init__(self, local_vars_configuration=None):  # noqa: E501
-        """LogLevelTypes - a model defined in OpenAPI"""  # noqa: E501
-        if local_vars_configuration is None:
-            local_vars_configuration = Configuration()
-        self.local_vars_configuration = local_vars_configuration
-        self.discriminator = None
-
-    def to_dict(self):
-        """Returns the model properties as a dict"""
-        result = {}
-
-        for attr, _ in six.iteritems(self.openapi_types):
-            value = getattr(self, attr)
-            if isinstance(value, list):
-                result[attr] = list(map(
-                    lambda x: x.to_dict() if hasattr(x, "to_dict") else x,
-                    value
-                ))
-            elif hasattr(value, "to_dict"):
-                result[attr] = value.to_dict()
-            elif isinstance(value, dict):
-                result[attr] = dict(map(
-                    lambda item: (item[0], item[1].to_dict())
-                    if hasattr(item[1], "to_dict") else item,
-                    value.items()
-                ))
-            else:
-                result[attr] = value
-
-        return result
-
-    def to_str(self):
-        """Returns the string representation of the model"""
-        return pprint.pformat(self.to_dict())
-
-    def __repr__(self):
-        """For `print` and `pprint`"""
-        return self.to_str()
-
-    def __eq__(self, other):
-        """Returns true if both objects are equal"""
-        if not isinstance(other, LogLevelTypes):
-            return False
-
-        return self.to_dict() == other.to_dict()
-
-    def __ne__(self, other):
-        """Returns true if both objects are not equal"""
-        if not isinstance(other, LogLevelTypes):
-            return True
-
-        return self.to_dict() != other.to_dict()

anyscale/sdk/anyscale_client/models/session_event.py

@@ -1,267 +0,0 @@
-# coding: utf-8
-
-"""
-    Anyscale API
-
-    No description provided (generated by Openapi Generator https://github.com/openapitools/openapi-generator)  # noqa: E501
-
-    The version of the OpenAPI document: 0.1.0
-    Generated by: https://openapi-generator.tech
-"""
-
-
-import pprint
-import re  # noqa: F401
-
-import six
-
-from anyscale_client.configuration import Configuration
-
-
-class SessionEvent(object):
-    """NOTE: This class is auto generated by OpenAPI Generator.
-    Ref: https://openapi-generator.tech
-
-    Do not edit the class manually.
-    """
-
-    """
-    Attributes:
-      openapi_types (dict): The key is attribute name
-                            and the value is attribute type.
-      attribute_map (dict): The key is attribute name
-                            and the value is json key in definition.
-    """
-    openapi_types = {
-        'event_type': 'str',
-        'log_level': 'LogLevelTypes',
-        'timestamp': 'datetime',
-        'description': 'str',
-        'cause': 'SessionEventCause',
-        'id': 'str'
-    }
-
-    attribute_map = {
-        'event_type': 'event_type',
-        'log_level': 'log_level',
-        'timestamp': 'timestamp',
-        'description': 'description',
-        'cause': 'cause',
-        'id': 'id'
-    }
-
-    def __init__(self, event_type=None, log_level=None, timestamp=None, description=None, cause=None, id=None, local_vars_configuration=None):  # noqa: E501
-        """SessionEvent - a model defined in OpenAPI"""  # noqa: E501
-        if local_vars_configuration is None:
-            local_vars_configuration = Configuration()
-        self.local_vars_configuration = local_vars_configuration
-
-        self._event_type = None
-        self._log_level = None
-        self._timestamp = None
-        self._description = None
-        self._cause = None
-        self._id = None
-        self.discriminator = None
-
-        self.event_type = event_type
-        self.log_level = log_level
-        self.timestamp = timestamp
-        self.description = description
-        if cause is not None:
-            self.cause = cause
-        self.id = id
-
-    @property
-    def event_type(self):
-        """Gets the event_type of this SessionEvent.  # noqa: E501
-
-        The category of session event.  # noqa: E501
-
-        :return: The event_type of this SessionEvent.  # noqa: E501
-        :rtype: str
-        """
-        return self._event_type
-
-    @event_type.setter
-    def event_type(self, event_type):
-        """Sets the event_type of this SessionEvent.
-
-        The category of session event.  # noqa: E501
-
-        :param event_type: The event_type of this SessionEvent.  # noqa: E501
-        :type: str
-        """
-        if self.local_vars_configuration.client_side_validation and event_type is None:  # noqa: E501
-            raise ValueError("Invalid value for `event_type`, must not be `None`")  # noqa: E501
-
-        self._event_type = event_type
-
-    @property
-    def log_level(self):
-        """Gets the log_level of this SessionEvent.  # noqa: E501
-
-        The severity of the session event.  # noqa: E501
-
-        :return: The log_level of this SessionEvent.  # noqa: E501
-        :rtype: LogLevelTypes
-        """
-        return self._log_level
-
-    @log_level.setter
-    def log_level(self, log_level):
-        """Sets the log_level of this SessionEvent.
-
-        The severity of the session event.  # noqa: E501
-
-        :param log_level: The log_level of this SessionEvent.  # noqa: E501
-        :type: LogLevelTypes
-        """
-        if self.local_vars_configuration.client_side_validation and log_level is None:  # noqa: E501
-            raise ValueError("Invalid value for `log_level`, must not be `None`")  # noqa: E501
-
-        self._log_level = log_level
-
-    @property
-    def timestamp(self):
-        """Gets the timestamp of this SessionEvent.  # noqa: E501
-
-        The time at which the session event occurred.  # noqa: E501
-
-        :return: The timestamp of this SessionEvent.  # noqa: E501
-        :rtype: datetime
-        """
-        return self._timestamp
-
-    @timestamp.setter
-    def timestamp(self, timestamp):
-        """Sets the timestamp of this SessionEvent.
-
-        The time at which the session event occurred.  # noqa: E501
-
-        :param timestamp: The timestamp of this SessionEvent.  # noqa: E501
-        :type: datetime
-        """
-        if self.local_vars_configuration.client_side_validation and timestamp is None:  # noqa: E501
-            raise ValueError("Invalid value for `timestamp`, must not be `None`")  # noqa: E501
-
-        self._timestamp = timestamp
-
-    @property
-    def description(self):
-        """Gets the description of this SessionEvent.  # noqa: E501
-
-        A human readable description of the session event.  # noqa: E501
-
-        :return: The description of this SessionEvent.  # noqa: E501
-        :rtype: str
-        """
-        return self._description
-
-    @description.setter
-    def description(self, description):
-        """Sets the description of this SessionEvent.
-
-        A human readable description of the session event.  # noqa: E501
-
-        :param description: The description of this SessionEvent.  # noqa: E501
-        :type: str
-        """
-        if self.local_vars_configuration.client_side_validation and description is None:  # noqa: E501
-            raise ValueError("Invalid value for `description`, must not be `None`")  # noqa: E501
-
-        self._description = description
-
-    @property
-    def cause(self):
-        """Gets the cause of this SessionEvent.  # noqa: E501
-
-        The reason why the session event occurred. None indicates an unknown cause.  # noqa: E501
-
-        :return: The cause of this SessionEvent.  # noqa: E501
-        :rtype: SessionEventCause
-        """
-        return self._cause
-
-    @cause.setter
-    def cause(self, cause):
-        """Sets the cause of this SessionEvent.
-
-        The reason why the session event occurred. None indicates an unknown cause.  # noqa: E501
-
-        :param cause: The cause of this SessionEvent.  # noqa: E501
-        :type: SessionEventCause
-        """
-
-        self._cause = cause
-
-    @property
-    def id(self):
-        """Gets the id of this SessionEvent.  # noqa: E501
-
-        A unique identifier for the session event.  # noqa: E501
-
-        :return: The id of this SessionEvent.  # noqa: E501
-        :rtype: str
-        """
-        return self._id
-
-    @id.setter
-    def id(self, id):
-        """Sets the id of this SessionEvent.
-
-        A unique identifier for the session event.  # noqa: E501
-
-        :param id: The id of this SessionEvent.  # noqa: E501
-        :type: str
-        """
-        if self.local_vars_configuration.client_side_validation and id is None:  # noqa: E501
-            raise ValueError("Invalid value for `id`, must not be `None`")  # noqa: E501
-
-        self._id = id
-
-    def to_dict(self):
-        """Returns the model properties as a dict"""
-        result = {}
-
-        for attr, _ in six.iteritems(self.openapi_types):
-            value = getattr(self, attr)
-            if isinstance(value, list):
-                result[attr] = list(map(
-                    lambda x: x.to_dict() if hasattr(x, "to_dict") else x,
-                    value
-                ))
-            elif hasattr(value, "to_dict"):
-                result[attr] = value.to_dict()
-            elif isinstance(value, dict):
-                result[attr] = dict(map(
-                    lambda item: (item[0], item[1].to_dict())
-                    if hasattr(item[1], "to_dict") else item,
-                    value.items()
-                ))
-            else:
-                result[attr] = value
-
-        return result
-
-    def to_str(self):
-        """Returns the string representation of the model"""
-        return pprint.pformat(self.to_dict())
-
-    def __repr__(self):
-        """For `print` and `pprint`"""
-        return self.to_str()
-
-    def __eq__(self, other):
-        """Returns true if both objects are equal"""
-        if not isinstance(other, SessionEvent):
-            return False
-
-        return self.to_dict() == other.to_dict()
-
-    def __ne__(self, other):
-        """Returns true if both objects are not equal"""
-        if not isinstance(other, SessionEvent):
-            return True
-
-        return self.to_dict() != other.to_dict()

anyscale/sdk/anyscale_client/models/session_event_cause.py

@@ -1,150 +0,0 @@
-# coding: utf-8
-
-"""
-    Anyscale API
-
-    No description provided (generated by Openapi Generator https://github.com/openapitools/openapi-generator)  # noqa: E501
-
-    The version of the OpenAPI document: 0.1.0
-    Generated by: https://openapi-generator.tech
-"""
-
-
-import pprint
-import re  # noqa: F401
-
-import six
-
-from anyscale_client.configuration import Configuration
-
-
-class SessionEventCause(object):
-    """NOTE: This class is auto generated by OpenAPI Generator.
-    Ref: https://openapi-generator.tech
-
-    Do not edit the class manually.
-    """
-
-    """
-    Attributes:
-      openapi_types (dict): The key is attribute name
-                            and the value is attribute type.
-      attribute_map (dict): The key is attribute name
-                            and the value is json key in definition.
-    """
-    openapi_types = {
-        'cause_user': 'str',
-        'cause_system': 'str'
-    }
-
-    attribute_map = {
-        'cause_user': 'cause_user',
-        'cause_system': 'cause_system'
-    }
-
-    def __init__(self, cause_user=None, cause_system=None, local_vars_configuration=None):  # noqa: E501
-        """SessionEventCause - a model defined in OpenAPI"""  # noqa: E501
-        if local_vars_configuration is None:
-            local_vars_configuration = Configuration()
-        self.local_vars_configuration = local_vars_configuration
-
-        self._cause_user = None
-        self._cause_system = None
-        self.discriminator = None
-
-        if cause_user is not None:
-            self.cause_user = cause_user
-        if cause_system is not None:
-            self.cause_system = cause_system
-
-    @property
-    def cause_user(self):
-        """Gets the cause_user of this SessionEventCause.  # noqa: E501
-
-        The username of the user who caused the session event.  # noqa: E501
-
-        :return: The cause_user of this SessionEventCause.  # noqa: E501
-        :rtype: str
-        """
-        return self._cause_user
-
-    @cause_user.setter
-    def cause_user(self, cause_user):
-        """Sets the cause_user of this SessionEventCause.
-
-        The username of the user who caused the session event.  # noqa: E501
-
-        :param cause_user: The cause_user of this SessionEventCause.  # noqa: E501
-        :type: str
-        """
-
-        self._cause_user = cause_user
-
-    @property
-    def cause_system(self):
-        """Gets the cause_system of this SessionEventCause.  # noqa: E501
-
-        The name of the internal anyscale system that caused the session event.  # noqa: E501
-
-        :return: The cause_system of this SessionEventCause.  # noqa: E501
-        :rtype: str
-        """
-        return self._cause_system
-
-    @cause_system.setter
-    def cause_system(self, cause_system):
-        """Sets the cause_system of this SessionEventCause.
-
-        The name of the internal anyscale system that caused the session event.  # noqa: E501
-
-        :param cause_system: The cause_system of this SessionEventCause.  # noqa: E501
-        :type: str
-        """
-
-        self._cause_system = cause_system
-
-    def to_dict(self):
-        """Returns the model properties as a dict"""
-        result = {}
-
-        for attr, _ in six.iteritems(self.openapi_types):
-            value = getattr(self, attr)
-            if isinstance(value, list):
-                result[attr] = list(map(
-                    lambda x: x.to_dict() if hasattr(x, "to_dict") else x,
-                    value
-                ))
-            elif hasattr(value, "to_dict"):
-                result[attr] = value.to_dict()
-            elif isinstance(value, dict):
-                result[attr] = dict(map(
-                    lambda item: (item[0], item[1].to_dict())
-                    if hasattr(item[1], "to_dict") else item,
-                    value.items()
-                ))
-            else:
-                result[attr] = value
-
-        return result
-
-    def to_str(self):
-        """Returns the string representation of the model"""
-        return pprint.pformat(self.to_dict())
-
-    def __repr__(self):
-        """For `print` and `pprint`"""
-        return self.to_str()
-
-    def __eq__(self, other):
-        """Returns true if both objects are equal"""
-        if not isinstance(other, SessionEventCause):
-            return False
-
-        return self.to_dict() == other.to_dict()
-
-    def __ne__(self, other):
-        """Returns true if both objects are not equal"""
-        if not isinstance(other, SessionEventCause):
-            return True
-
-        return self.to_dict() != other.to_dict()