anyscale 0.24.87__py3-none-any.whl → 0.24.91__py3-none-any.whl

anyscale/user/models.py

@@ -0,0 +1,201 @@
+from dataclasses import dataclass, field
+from datetime import datetime
+from typing import Any, Dict, List, Optional
+
+from anyscale._private.models import ModelBase
+
+
+@dataclass(frozen=True)
+class AdminCreateUser(ModelBase):
+    """User to be created by an admin.
+    """
+
+    __doc_py_example__ = """\
+import anyscale
+from anyscale.user.models import AdminCreateUser
+
+admin_create_user = AdminCreateUser(
+    # First name of the user to be created.
+    name="name",
+    # Email of the user to be created.
+    email="test@anyscale.com",
+    # Password for the user account being created.
+    password="",
+    # Whether the user is an SSO user. SSO users can log in using SSO.
+    is_sso_user=False,
+    # Optional last name of the user to be created.
+    lastname="lastname",
+    # Optional title of the user to be created.
+    title="title",
+)
+"""
+    name: str = field(metadata={"docstring": "First name of the user to be created."})
+
+    def _validate_name(self, name: str):
+        if not isinstance(name, str):
+            raise TypeError("name must be a string.")
+
+    email: str = field(metadata={"docstring": "Email of the user to be created."})
+
+    def _validate_email(self, email: str):
+        if not isinstance(email, str):
+            raise TypeError("email must be a string.")
+
+    password: Optional[str] = field(
+        metadata={"docstring": "Password for the user account being created."}
+    )
+
+    def _validate_password(self, password: Optional[str]):
+        if password is not None and not isinstance(password, str):
+            raise TypeError("password must be a string.")
+
+    is_sso_user: bool = field(
+        metadata={
+            "docstring": "Whether the user is an SSO user. SSO users can log in using SSO."
+        },
+    )
+
+    def _validate_is_sso_user(self, is_sso_user: bool):
+        if not isinstance(is_sso_user, bool):
+            raise TypeError("is_sso_user must be a boolean.")
+
+    lastname: Optional[str] = field(
+        default=None,
+        metadata={"docstring": "Optional last name of the user to be created."},
+    )
+
+    def _validate_lastname(self, lastname: Optional[str]):
+        if lastname is not None and not isinstance(lastname, str):
+            raise TypeError("lastname must be a string.")
+
+    title: Optional[str] = field(
+        default=None,
+        metadata={"docstring": "Optional title of the user to be created."},
+    )
+
+    def _validate_title(self, title: Optional[str]):
+        if title is not None and not isinstance(title, str):
+            raise TypeError("title must be a string.")
+
+
+@dataclass(frozen=True)
+class AdminCreateUsers(ModelBase):
+    """Users to be created by an admin.
+    """
+
+    __doc_py_example__ = """\
+import anyscale
+from anyscale.user.models import AdminCreateUser
+
+admin_create_user = AdminCreateUser(
+    # First name of the user to be created.
+    name="name",
+    # Email of the user to be created.
+    email="test@anyscale.com",
+    # Password for the user account being created.
+    password="",
+    # Whether the user is an SSO user. SSO users can log in using SSO.
+    is_sso_user=False,
+    # Optional last name of the user to be created.
+    lastname="lastname",
+    # Optional title of the user to be created.
+    title="title",
+)
+admin_create_users = AdminCreateUsers(
+    # Users to be created by an admin.
+    create_users=[admin_create_user]
+)
+"""
+    create_users: List[Dict[str, Any]] = field(
+        metadata={"docstring": "Users to be created by an admin."}
+    )
+
+    def _validate_create_users(self, create_users: List[Dict[str, Any]]):
+        if not isinstance(create_users, list):
+            raise TypeError("create_users must be a list.")
+
+
+@dataclass(frozen=True)
+class AdminCreatedUser(ModelBase):
+    """User account created by an admin that has organization collaborator permissions.
+    """
+
+    __doc_py_example__ = """\
+import anyscale
+from anyscale.user.models import AdminCreatedUser
+
+admin_create_user = AdminCreateUser(
+    # First name of the user to be created.
+    name="name",
+    # Email of the user to be created.
+    email="test@anyscale.com",
+    # Password for the user account being created.
+    password="",
+    # Whether the user is an SSO user. SSO users can log in using SSO.
+    is_sso_user=False,
+    # Optional last name of the user to be created.
+    lastname="lastname",
+    # Optional title of the user to be created.
+    title="title",
+)
+admin_created_users: List[AdminCreatedUser] = anyscale.user.admin_batch_create([admin_create_user])
+"""
+    user_id: str = field(
+        metadata={"docstring": "ID of the user that has been created."}
+    )
+
+    def _validate_user_id(self, user_id: str):
+        if not isinstance(user_id, str):
+            raise TypeError("user_id must be a string.")
+
+    name: str = field(
+        metadata={"docstring": "First name of the user that has been created."}
+    )
+
+    def _validate_name(self, name: str):
+        if not isinstance(name, str):
+            raise TypeError("name must be a string.")
+
+    email: str = field(
+        metadata={"docstring": "Email of the user that has been created."}
+    )
+
+    def _validate_email(self, email: str):
+        if not isinstance(email, str):
+            raise TypeError("email must be a string.")
+
+    created_at: datetime = field(
+        metadata={"docstring": "The timestamp of when the user is created."}
+    )
+
+    def _validate_created_at(self, created_at: datetime):
+        if not isinstance(created_at, datetime):
+            raise TypeError("created_at must be a datetime.")
+
+    is_sso_user: bool = field(
+        metadata={
+            "docstring": "Whether the user is an SSO user. SSO users can log in using SSO."
+        },
+    )
+
+    def _validate_is_sso_user(self, is_sso_user: bool):
+        if not isinstance(is_sso_user, bool):
+            raise TypeError("is_sso_user must be a boolean.")
+
+    lastname: Optional[str] = field(
+        default=None,
+        metadata={"docstring": "Optional last name of the user that has been created."},
+    )
+
+    def _validate_lastname(self, lastname: Optional[str]):
+        if lastname is not None and not isinstance(lastname, str):
+            raise TypeError("lastname must be a string.")
+
+    title: Optional[str] = field(
+        default=None,
+        metadata={"docstring": "Optional title of the user that has been created."},
+    )
+
+    def _validate_title(self, title: Optional[str]):
+        if title is not None and not isinstance(title, str):
+            raise TypeError("title must be a string.")

anyscale/util.py

@@ -44,6 +44,7 @@ from anyscale.client.openapi_client.models.cloud_analytics_event_cloud_resource
 from anyscale.client.openapi_client.models.decorated_compute_template import (
     DecoratedComputeTemplate,
 )
+from anyscale.client.openapi_client.models.ha_job_states import HaJobStates
 from anyscale.client.openapi_client.models.user_info import UserInfo
 from anyscale.cluster_compute import get_cluster_compute_from_name
 import anyscale.conf
@@ -986,6 +987,20 @@ def validate_job_config_dict(
         )
 
 
+def validate_list_jobs_state_filter(ctx, param, value) -> List[str]:  # noqa: ARG001
+    """
+    Validate the job state filter for list jobs CLI method
+    """
+    if not value:
+        return []
+    for each_value in value:
+        if each_value.upper() not in HaJobStates.allowable_values:
+            raise click.ClickException(
+                f"{each_value} is not a valid value for {param.opts[0]}. Allowed values: {', '.join(HaJobStates.allowable_values)}"
+            )
+    return [each_value.upper() for each_value in value]
+
+
 def get_cluster_model_for_current_workspace(
     anyscale_api_client: BaseApi,
 ) -> Optional[Cluster]:

anyscale/utils/cloud_utils.py

@@ -20,7 +20,7 @@ from anyscale.client.openapi_client.models import (
     CloudProviders,
     CreateAnalyticsEvent,
 )
-from anyscale.cloud import get_cloud_id_and_name
+from anyscale.cloud_utils import get_cloud_id_and_name
 from anyscale.shared_anyscale_utils.utils.collections import flatten
 
 

anyscale/version.py

@@ -1 +1 @@
-__version__ = "0.24.87"
+__version__ = "0.24.91"

anyscale/workspace_utils.py

@@ -3,7 +3,7 @@ import os
 from click import ClickException
 import yaml
 
-from anyscale.project import ANYSCALE_PROJECT_FILE, find_project_root
+from anyscale.project_utils import ANYSCALE_PROJECT_FILE, find_project_root
 
 
 def write_workspace_id_to_disk(workspace_id: str, directory: str) -> None:

{anyscale-0.24.87.dist-info → anyscale-0.24.91.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: anyscale
-Version: 0.24.87
+Version: 0.24.91
 Summary: Command Line Interface for Anyscale
 Author: Anyscale Inc.
 License: AS License
@@ -47,7 +47,6 @@ Requires-Dist: google-cloud-filestore; extra == "backend"
 Requires-Dist: google-cloud-storage; extra == "backend"
 Requires-Dist: google-cloud-redis; extra == "backend"
 Requires-Dist: google-cloud-certificate-manager; extra == "backend"
-Requires-Dist: azure-storage-blob; extra == "backend"
 Provides-Extra: gcp
 Requires-Dist: protobuf<4; python_version < "3.11" and extra == "gcp"
 Requires-Dist: google-api-python-client; extra == "gcp"
@@ -58,8 +57,6 @@ Requires-Dist: google-cloud-filestore; extra == "gcp"
 Requires-Dist: google-cloud-storage; extra == "gcp"
 Requires-Dist: google-cloud-redis; extra == "gcp"
 Requires-Dist: google-cloud-certificate-manager; extra == "gcp"
-Provides-Extra: azure
-Requires-Dist: azure-storage-blob; extra == "azure"
 Provides-Extra: all
 Requires-Dist: ray>=1.4.0; extra == "all"
 Requires-Dist: protobuf<4; python_version < "3.11" and extra == "all"
@@ -71,7 +68,6 @@ Requires-Dist: google-cloud-filestore; extra == "all"
 Requires-Dist: google-cloud-storage; extra == "all"
 Requires-Dist: google-cloud-redis; extra == "all"
 Requires-Dist: google-cloud-certificate-manager; extra == "all"
-Requires-Dist: azure-storage-blob; extra == "all"
 
 # Anyscale