yandexcloud 0.348.0__py3-none-any.whl → 0.350.0__py3-none-any.whl

yandex/cloud/ytsaurus/v1/cluster_service_pb2_grpc.py

@@ -0,0 +1,367 @@
+# Generated by the gRPC Python protocol compiler plugin. DO NOT EDIT!
+"""Client and server classes corresponding to protobuf-defined services."""
+import grpc
+import warnings
+
+from yandex.cloud.operation import operation_pb2 as yandex_dot_cloud_dot_operation_dot_operation__pb2
+from yandex.cloud.ytsaurus.v1 import cluster_pb2 as yandex_dot_cloud_dot_ytsaurus_dot_v1_dot_cluster__pb2
+from yandex.cloud.ytsaurus.v1 import cluster_service_pb2 as yandex_dot_cloud_dot_ytsaurus_dot_v1_dot_cluster__service__pb2
+
+GRPC_GENERATED_VERSION = '1.70.0'
+GRPC_VERSION = grpc.__version__
+_version_not_supported = False
+
+try:
+    from grpc._utilities import first_version_is_lower
+    _version_not_supported = first_version_is_lower(GRPC_VERSION, GRPC_GENERATED_VERSION)
+except ImportError:
+    _version_not_supported = True
+
+if _version_not_supported:
+    raise RuntimeError(
+        f'The grpc package installed is at version {GRPC_VERSION},'
+        + f' but the generated code in yandex/cloud/ytsaurus/v1/cluster_service_pb2_grpc.py depends on'
+        + f' grpcio>={GRPC_GENERATED_VERSION}.'
+        + f' Please upgrade your grpc module to grpcio>={GRPC_GENERATED_VERSION}'
+        + f' or downgrade your generated code using grpcio-tools<={GRPC_VERSION}.'
+    )
+
+
+class ClusterServiceStub(object):
+    """A set of methods for managing Cluster resources.
+    """
+
+    def __init__(self, channel):
+        """Constructor.
+
+        Args:
+            channel: A grpc.Channel.
+        """
+        self.Get = channel.unary_unary(
+                '/yandex.cloud.ytsaurus.v1.ClusterService/Get',
+                request_serializer=yandex_dot_cloud_dot_ytsaurus_dot_v1_dot_cluster__service__pb2.GetClusterRequest.SerializeToString,
+                response_deserializer=yandex_dot_cloud_dot_ytsaurus_dot_v1_dot_cluster__pb2.Cluster.FromString,
+                _registered_method=True)
+        self.List = channel.unary_unary(
+                '/yandex.cloud.ytsaurus.v1.ClusterService/List',
+                request_serializer=yandex_dot_cloud_dot_ytsaurus_dot_v1_dot_cluster__service__pb2.ListClustersRequest.SerializeToString,
+                response_deserializer=yandex_dot_cloud_dot_ytsaurus_dot_v1_dot_cluster__service__pb2.ListClustersResponse.FromString,
+                _registered_method=True)
+        self.Create = channel.unary_unary(
+                '/yandex.cloud.ytsaurus.v1.ClusterService/Create',
+                request_serializer=yandex_dot_cloud_dot_ytsaurus_dot_v1_dot_cluster__service__pb2.CreateClusterRequest.SerializeToString,
+                response_deserializer=yandex_dot_cloud_dot_operation_dot_operation__pb2.Operation.FromString,
+                _registered_method=True)
+        self.Update = channel.unary_unary(
+                '/yandex.cloud.ytsaurus.v1.ClusterService/Update',
+                request_serializer=yandex_dot_cloud_dot_ytsaurus_dot_v1_dot_cluster__service__pb2.UpdateClusterRequest.SerializeToString,
+                response_deserializer=yandex_dot_cloud_dot_operation_dot_operation__pb2.Operation.FromString,
+                _registered_method=True)
+        self.Delete = channel.unary_unary(
+                '/yandex.cloud.ytsaurus.v1.ClusterService/Delete',
+                request_serializer=yandex_dot_cloud_dot_ytsaurus_dot_v1_dot_cluster__service__pb2.DeleteClusterRequest.SerializeToString,
+                response_deserializer=yandex_dot_cloud_dot_operation_dot_operation__pb2.Operation.FromString,
+                _registered_method=True)
+        self.Start = channel.unary_unary(
+                '/yandex.cloud.ytsaurus.v1.ClusterService/Start',
+                request_serializer=yandex_dot_cloud_dot_ytsaurus_dot_v1_dot_cluster__service__pb2.StartClusterRequest.SerializeToString,
+                response_deserializer=yandex_dot_cloud_dot_operation_dot_operation__pb2.Operation.FromString,
+                _registered_method=True)
+        self.Stop = channel.unary_unary(
+                '/yandex.cloud.ytsaurus.v1.ClusterService/Stop',
+                request_serializer=yandex_dot_cloud_dot_ytsaurus_dot_v1_dot_cluster__service__pb2.StopClusterRequest.SerializeToString,
+                response_deserializer=yandex_dot_cloud_dot_operation_dot_operation__pb2.Operation.FromString,
+                _registered_method=True)
+
+
+class ClusterServiceServicer(object):
+    """A set of methods for managing Cluster resources.
+    """
+
+    def Get(self, request, context):
+        """Returns the specified cluster.
+        """
+        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
+        context.set_details('Method not implemented!')
+        raise NotImplementedError('Method not implemented!')
+
+    def List(self, request, context):
+        """Retrieves the list of clusters in the specified folder.
+        """
+        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
+        context.set_details('Method not implemented!')
+        raise NotImplementedError('Method not implemented!')
+
+    def Create(self, request, context):
+        """Creates a cluster in the specified folder.
+        """
+        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
+        context.set_details('Method not implemented!')
+        raise NotImplementedError('Method not implemented!')
+
+    def Update(self, request, context):
+        """Updates the specified cluster.
+        """
+        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
+        context.set_details('Method not implemented!')
+        raise NotImplementedError('Method not implemented!')
+
+    def Delete(self, request, context):
+        """Deletes the specified cluster.
+        """
+        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
+        context.set_details('Method not implemented!')
+        raise NotImplementedError('Method not implemented!')
+
+    def Start(self, request, context):
+        """Starts the specified cluster.
+        """
+        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
+        context.set_details('Method not implemented!')
+        raise NotImplementedError('Method not implemented!')
+
+    def Stop(self, request, context):
+        """Stops the specified cluster.
+        """
+        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
+        context.set_details('Method not implemented!')
+        raise NotImplementedError('Method not implemented!')
+
+
+def add_ClusterServiceServicer_to_server(servicer, server):
+    rpc_method_handlers = {
+            'Get': grpc.unary_unary_rpc_method_handler(
+                    servicer.Get,
+                    request_deserializer=yandex_dot_cloud_dot_ytsaurus_dot_v1_dot_cluster__service__pb2.GetClusterRequest.FromString,
+                    response_serializer=yandex_dot_cloud_dot_ytsaurus_dot_v1_dot_cluster__pb2.Cluster.SerializeToString,
+            ),
+            'List': grpc.unary_unary_rpc_method_handler(
+                    servicer.List,
+                    request_deserializer=yandex_dot_cloud_dot_ytsaurus_dot_v1_dot_cluster__service__pb2.ListClustersRequest.FromString,
+                    response_serializer=yandex_dot_cloud_dot_ytsaurus_dot_v1_dot_cluster__service__pb2.ListClustersResponse.SerializeToString,
+            ),
+            'Create': grpc.unary_unary_rpc_method_handler(
+                    servicer.Create,
+                    request_deserializer=yandex_dot_cloud_dot_ytsaurus_dot_v1_dot_cluster__service__pb2.CreateClusterRequest.FromString,
+                    response_serializer=yandex_dot_cloud_dot_operation_dot_operation__pb2.Operation.SerializeToString,
+            ),
+            'Update': grpc.unary_unary_rpc_method_handler(
+                    servicer.Update,
+                    request_deserializer=yandex_dot_cloud_dot_ytsaurus_dot_v1_dot_cluster__service__pb2.UpdateClusterRequest.FromString,
+                    response_serializer=yandex_dot_cloud_dot_operation_dot_operation__pb2.Operation.SerializeToString,
+            ),
+            'Delete': grpc.unary_unary_rpc_method_handler(
+                    servicer.Delete,
+                    request_deserializer=yandex_dot_cloud_dot_ytsaurus_dot_v1_dot_cluster__service__pb2.DeleteClusterRequest.FromString,
+                    response_serializer=yandex_dot_cloud_dot_operation_dot_operation__pb2.Operation.SerializeToString,
+            ),
+            'Start': grpc.unary_unary_rpc_method_handler(
+                    servicer.Start,
+                    request_deserializer=yandex_dot_cloud_dot_ytsaurus_dot_v1_dot_cluster__service__pb2.StartClusterRequest.FromString,
+                    response_serializer=yandex_dot_cloud_dot_operation_dot_operation__pb2.Operation.SerializeToString,
+            ),
+            'Stop': grpc.unary_unary_rpc_method_handler(
+                    servicer.Stop,
+                    request_deserializer=yandex_dot_cloud_dot_ytsaurus_dot_v1_dot_cluster__service__pb2.StopClusterRequest.FromString,
+                    response_serializer=yandex_dot_cloud_dot_operation_dot_operation__pb2.Operation.SerializeToString,
+            ),
+    }
+    generic_handler = grpc.method_handlers_generic_handler(
+            'yandex.cloud.ytsaurus.v1.ClusterService', rpc_method_handlers)
+    server.add_generic_rpc_handlers((generic_handler,))
+    server.add_registered_method_handlers('yandex.cloud.ytsaurus.v1.ClusterService', rpc_method_handlers)
+
+
+ # This class is part of an EXPERIMENTAL API.
+class ClusterService(object):
+    """A set of methods for managing Cluster resources.
+    """
+
+    @staticmethod
+    def Get(request,
+            target,
+            options=(),
+            channel_credentials=None,
+            call_credentials=None,
+            insecure=False,
+            compression=None,
+            wait_for_ready=None,
+            timeout=None,
+            metadata=None):
+        return grpc.experimental.unary_unary(
+            request,
+            target,
+            '/yandex.cloud.ytsaurus.v1.ClusterService/Get',
+            yandex_dot_cloud_dot_ytsaurus_dot_v1_dot_cluster__service__pb2.GetClusterRequest.SerializeToString,
+            yandex_dot_cloud_dot_ytsaurus_dot_v1_dot_cluster__pb2.Cluster.FromString,
+            options,
+            channel_credentials,
+            insecure,
+            call_credentials,
+            compression,
+            wait_for_ready,
+            timeout,
+            metadata,
+            _registered_method=True)
+
+    @staticmethod
+    def List(request,
+            target,
+            options=(),
+            channel_credentials=None,
+            call_credentials=None,
+            insecure=False,
+            compression=None,
+            wait_for_ready=None,
+            timeout=None,
+            metadata=None):
+        return grpc.experimental.unary_unary(
+            request,
+            target,
+            '/yandex.cloud.ytsaurus.v1.ClusterService/List',
+            yandex_dot_cloud_dot_ytsaurus_dot_v1_dot_cluster__service__pb2.ListClustersRequest.SerializeToString,
+            yandex_dot_cloud_dot_ytsaurus_dot_v1_dot_cluster__service__pb2.ListClustersResponse.FromString,
+            options,
+            channel_credentials,
+            insecure,
+            call_credentials,
+            compression,
+            wait_for_ready,
+            timeout,
+            metadata,
+            _registered_method=True)
+
+    @staticmethod
+    def Create(request,
+            target,
+            options=(),
+            channel_credentials=None,
+            call_credentials=None,
+            insecure=False,
+            compression=None,
+            wait_for_ready=None,
+            timeout=None,
+            metadata=None):
+        return grpc.experimental.unary_unary(
+            request,
+            target,
+            '/yandex.cloud.ytsaurus.v1.ClusterService/Create',
+            yandex_dot_cloud_dot_ytsaurus_dot_v1_dot_cluster__service__pb2.CreateClusterRequest.SerializeToString,
+            yandex_dot_cloud_dot_operation_dot_operation__pb2.Operation.FromString,
+            options,
+            channel_credentials,
+            insecure,
+            call_credentials,
+            compression,
+            wait_for_ready,
+            timeout,
+            metadata,
+            _registered_method=True)
+
+    @staticmethod
+    def Update(request,
+            target,
+            options=(),
+            channel_credentials=None,
+            call_credentials=None,
+            insecure=False,
+            compression=None,
+            wait_for_ready=None,
+            timeout=None,
+            metadata=None):
+        return grpc.experimental.unary_unary(
+            request,
+            target,
+            '/yandex.cloud.ytsaurus.v1.ClusterService/Update',
+            yandex_dot_cloud_dot_ytsaurus_dot_v1_dot_cluster__service__pb2.UpdateClusterRequest.SerializeToString,
+            yandex_dot_cloud_dot_operation_dot_operation__pb2.Operation.FromString,
+            options,
+            channel_credentials,
+            insecure,
+            call_credentials,
+            compression,
+            wait_for_ready,
+            timeout,
+            metadata,
+            _registered_method=True)
+
+    @staticmethod
+    def Delete(request,
+            target,
+            options=(),
+            channel_credentials=None,
+            call_credentials=None,
+            insecure=False,
+            compression=None,
+            wait_for_ready=None,
+            timeout=None,
+            metadata=None):
+        return grpc.experimental.unary_unary(
+            request,
+            target,
+            '/yandex.cloud.ytsaurus.v1.ClusterService/Delete',
+            yandex_dot_cloud_dot_ytsaurus_dot_v1_dot_cluster__service__pb2.DeleteClusterRequest.SerializeToString,
+            yandex_dot_cloud_dot_operation_dot_operation__pb2.Operation.FromString,
+            options,
+            channel_credentials,
+            insecure,
+            call_credentials,
+            compression,
+            wait_for_ready,
+            timeout,
+            metadata,
+            _registered_method=True)
+
+    @staticmethod
+    def Start(request,
+            target,
+            options=(),
+            channel_credentials=None,
+            call_credentials=None,
+            insecure=False,
+            compression=None,
+            wait_for_ready=None,
+            timeout=None,
+            metadata=None):
+        return grpc.experimental.unary_unary(
+            request,
+            target,
+            '/yandex.cloud.ytsaurus.v1.ClusterService/Start',
+            yandex_dot_cloud_dot_ytsaurus_dot_v1_dot_cluster__service__pb2.StartClusterRequest.SerializeToString,
+            yandex_dot_cloud_dot_operation_dot_operation__pb2.Operation.FromString,
+            options,
+            channel_credentials,
+            insecure,
+            call_credentials,
+            compression,
+            wait_for_ready,
+            timeout,
+            metadata,
+            _registered_method=True)
+
+    @staticmethod
+    def Stop(request,
+            target,
+            options=(),
+            channel_credentials=None,
+            call_credentials=None,
+            insecure=False,
+            compression=None,
+            wait_for_ready=None,
+            timeout=None,
+            metadata=None):
+        return grpc.experimental.unary_unary(
+            request,
+            target,
+            '/yandex.cloud.ytsaurus.v1.ClusterService/Stop',
+            yandex_dot_cloud_dot_ytsaurus_dot_v1_dot_cluster__service__pb2.StopClusterRequest.SerializeToString,
+            yandex_dot_cloud_dot_operation_dot_operation__pb2.Operation.FromString,
+            options,
+            channel_credentials,
+            insecure,
+            call_credentials,
+            compression,
+            wait_for_ready,
+            timeout,
+            metadata,
+            _registered_method=True)

yandex/cloud/ytsaurus/v1/cluster_service_pb2_grpc.pyi

@@ -0,0 +1,172 @@
+"""
+@generated by mypy-protobuf.  Do not edit manually!
+isort:skip_file
+"""
+
+import abc
+import collections.abc
+import grpc
+import grpc.aio
+import typing
+import yandex.cloud.operation.operation_pb2
+import yandex.cloud.ytsaurus.v1.cluster_pb2
+import yandex.cloud.ytsaurus.v1.cluster_service_pb2
+
+_T = typing.TypeVar("_T")
+
+class _MaybeAsyncIterator(collections.abc.AsyncIterator[_T], collections.abc.Iterator[_T], metaclass=abc.ABCMeta): ...
+
+class _ServicerContext(grpc.ServicerContext, grpc.aio.ServicerContext):  # type: ignore[misc, type-arg]
+    ...
+
+class ClusterServiceStub:
+    """A set of methods for managing Cluster resources."""
+
+    def __init__(self, channel: typing.Union[grpc.Channel, grpc.aio.Channel]) -> None: ...
+    Get: grpc.UnaryUnaryMultiCallable[
+        yandex.cloud.ytsaurus.v1.cluster_service_pb2.GetClusterRequest,
+        yandex.cloud.ytsaurus.v1.cluster_pb2.Cluster,
+    ]
+    """Returns the specified cluster."""
+
+    List: grpc.UnaryUnaryMultiCallable[
+        yandex.cloud.ytsaurus.v1.cluster_service_pb2.ListClustersRequest,
+        yandex.cloud.ytsaurus.v1.cluster_service_pb2.ListClustersResponse,
+    ]
+    """Retrieves the list of clusters in the specified folder."""
+
+    Create: grpc.UnaryUnaryMultiCallable[
+        yandex.cloud.ytsaurus.v1.cluster_service_pb2.CreateClusterRequest,
+        yandex.cloud.operation.operation_pb2.Operation,
+    ]
+    """Creates a cluster in the specified folder."""
+
+    Update: grpc.UnaryUnaryMultiCallable[
+        yandex.cloud.ytsaurus.v1.cluster_service_pb2.UpdateClusterRequest,
+        yandex.cloud.operation.operation_pb2.Operation,
+    ]
+    """Updates the specified cluster."""
+
+    Delete: grpc.UnaryUnaryMultiCallable[
+        yandex.cloud.ytsaurus.v1.cluster_service_pb2.DeleteClusterRequest,
+        yandex.cloud.operation.operation_pb2.Operation,
+    ]
+    """Deletes the specified cluster."""
+
+    Start: grpc.UnaryUnaryMultiCallable[
+        yandex.cloud.ytsaurus.v1.cluster_service_pb2.StartClusterRequest,
+        yandex.cloud.operation.operation_pb2.Operation,
+    ]
+    """Starts the specified cluster."""
+
+    Stop: grpc.UnaryUnaryMultiCallable[
+        yandex.cloud.ytsaurus.v1.cluster_service_pb2.StopClusterRequest,
+        yandex.cloud.operation.operation_pb2.Operation,
+    ]
+    """Stops the specified cluster."""
+
+class ClusterServiceAsyncStub:
+    """A set of methods for managing Cluster resources."""
+
+    Get: grpc.aio.UnaryUnaryMultiCallable[
+        yandex.cloud.ytsaurus.v1.cluster_service_pb2.GetClusterRequest,
+        yandex.cloud.ytsaurus.v1.cluster_pb2.Cluster,
+    ]
+    """Returns the specified cluster."""
+
+    List: grpc.aio.UnaryUnaryMultiCallable[
+        yandex.cloud.ytsaurus.v1.cluster_service_pb2.ListClustersRequest,
+        yandex.cloud.ytsaurus.v1.cluster_service_pb2.ListClustersResponse,
+    ]
+    """Retrieves the list of clusters in the specified folder."""
+
+    Create: grpc.aio.UnaryUnaryMultiCallable[
+        yandex.cloud.ytsaurus.v1.cluster_service_pb2.CreateClusterRequest,
+        yandex.cloud.operation.operation_pb2.Operation,
+    ]
+    """Creates a cluster in the specified folder."""
+
+    Update: grpc.aio.UnaryUnaryMultiCallable[
+        yandex.cloud.ytsaurus.v1.cluster_service_pb2.UpdateClusterRequest,
+        yandex.cloud.operation.operation_pb2.Operation,
+    ]
+    """Updates the specified cluster."""
+
+    Delete: grpc.aio.UnaryUnaryMultiCallable[
+        yandex.cloud.ytsaurus.v1.cluster_service_pb2.DeleteClusterRequest,
+        yandex.cloud.operation.operation_pb2.Operation,
+    ]
+    """Deletes the specified cluster."""
+
+    Start: grpc.aio.UnaryUnaryMultiCallable[
+        yandex.cloud.ytsaurus.v1.cluster_service_pb2.StartClusterRequest,
+        yandex.cloud.operation.operation_pb2.Operation,
+    ]
+    """Starts the specified cluster."""
+
+    Stop: grpc.aio.UnaryUnaryMultiCallable[
+        yandex.cloud.ytsaurus.v1.cluster_service_pb2.StopClusterRequest,
+        yandex.cloud.operation.operation_pb2.Operation,
+    ]
+    """Stops the specified cluster."""
+
+class ClusterServiceServicer(metaclass=abc.ABCMeta):
+    """A set of methods for managing Cluster resources."""
+
+    @abc.abstractmethod
+    def Get(
+        self,
+        request: yandex.cloud.ytsaurus.v1.cluster_service_pb2.GetClusterRequest,
+        context: _ServicerContext,
+    ) -> typing.Union[yandex.cloud.ytsaurus.v1.cluster_pb2.Cluster, collections.abc.Awaitable[yandex.cloud.ytsaurus.v1.cluster_pb2.Cluster]]:
+        """Returns the specified cluster."""
+
+    @abc.abstractmethod
+    def List(
+        self,
+        request: yandex.cloud.ytsaurus.v1.cluster_service_pb2.ListClustersRequest,
+        context: _ServicerContext,
+    ) -> typing.Union[yandex.cloud.ytsaurus.v1.cluster_service_pb2.ListClustersResponse, collections.abc.Awaitable[yandex.cloud.ytsaurus.v1.cluster_service_pb2.ListClustersResponse]]:
+        """Retrieves the list of clusters in the specified folder."""
+
+    @abc.abstractmethod
+    def Create(
+        self,
+        request: yandex.cloud.ytsaurus.v1.cluster_service_pb2.CreateClusterRequest,
+        context: _ServicerContext,
+    ) -> typing.Union[yandex.cloud.operation.operation_pb2.Operation, collections.abc.Awaitable[yandex.cloud.operation.operation_pb2.Operation]]:
+        """Creates a cluster in the specified folder."""
+
+    @abc.abstractmethod
+    def Update(
+        self,
+        request: yandex.cloud.ytsaurus.v1.cluster_service_pb2.UpdateClusterRequest,
+        context: _ServicerContext,
+    ) -> typing.Union[yandex.cloud.operation.operation_pb2.Operation, collections.abc.Awaitable[yandex.cloud.operation.operation_pb2.Operation]]:
+        """Updates the specified cluster."""
+
+    @abc.abstractmethod
+    def Delete(
+        self,
+        request: yandex.cloud.ytsaurus.v1.cluster_service_pb2.DeleteClusterRequest,
+        context: _ServicerContext,
+    ) -> typing.Union[yandex.cloud.operation.operation_pb2.Operation, collections.abc.Awaitable[yandex.cloud.operation.operation_pb2.Operation]]:
+        """Deletes the specified cluster."""
+
+    @abc.abstractmethod
+    def Start(
+        self,
+        request: yandex.cloud.ytsaurus.v1.cluster_service_pb2.StartClusterRequest,
+        context: _ServicerContext,
+    ) -> typing.Union[yandex.cloud.operation.operation_pb2.Operation, collections.abc.Awaitable[yandex.cloud.operation.operation_pb2.Operation]]:
+        """Starts the specified cluster."""
+
+    @abc.abstractmethod
+    def Stop(
+        self,
+        request: yandex.cloud.ytsaurus.v1.cluster_service_pb2.StopClusterRequest,
+        context: _ServicerContext,
+    ) -> typing.Union[yandex.cloud.operation.operation_pb2.Operation, collections.abc.Awaitable[yandex.cloud.operation.operation_pb2.Operation]]:
+        """Stops the specified cluster."""
+
+def add_ClusterServiceServicer_to_server(servicer: ClusterServiceServicer, server: typing.Union[grpc.Server, grpc.aio.Server]) -> None: ...

yandexcloud/__init__.py

@@ -11,4 +11,4 @@ from yandexcloud._retry_interceptor import RetryInterceptor
 from yandexcloud._retry_policy import RetryPolicy, ThrottlingMode
 from yandexcloud._sdk import SDK
 
-__version__ = "0.348.0"
+__version__ = "0.350.0"

yandexcloud/_wrappers/dataproc/__init__.py

@@ -68,6 +68,7 @@ class Dataproc:
         services=None,
         zone="ru-central1-b",
         service_account_id=None,
+        environment=None,
         masternode_resource_preset=None,
         masternode_disk_size=None,
         masternode_disk_type=None,
@@ -92,6 +93,7 @@ class Dataproc:
         host_group_ids=None,
         security_group_ids=None,
         initialization_actions=None,
+        oslogin_enabled=False,
         labels=None,
     ):
         """
@@ -121,6 +123,8 @@ class Dataproc:
         :param service_account_id: Service account id for the cluster.
                                    Service account can be created inside the folder.
         :type service_account_id: str
+        :param environment: Environment for the cluster. Possible options: PRODUCTION, PRESTABLE.
+        :type environment: str
         :param masternode_resource_preset: Resources preset (CPU+RAM configuration)
                                            for the master node of the cluster.
         :type masternode_resource_preset: str
@@ -182,6 +186,8 @@ class Dataproc:
         :type security_group_ids: Iterable[str] | None
         :param initialization_actions: Set of init-actions to run when cluster starts
         :type initialization_actions: Iterable[InitializationAction] | None
+        :param oslogin_enabled: Enable authorization via OS Login for cluster.
+        :type oslogin_enabled: bool
         :param labels: Cluster labels as key:value pairs. No more than 64 per resource.
         :type labels: Dict[str, str]
 
@@ -208,11 +214,11 @@ class Dataproc:
         if not service_account_id:
             service_account_id = self.sdk.helpers.find_service_account_id(folder_id)
 
-        if not ssh_public_keys:
+        if not ssh_public_keys and not oslogin_enabled:
             if self.default_public_ssh_key:
                 ssh_public_keys = (self.default_public_ssh_key,)
             else:
-                raise RuntimeError("Public ssh keys must be specified.")
+                raise RuntimeError("Public ssh keys must be specified or OS Login must be enabled.")
         elif isinstance(ssh_public_keys, str):
             ssh_public_keys = [ssh_public_keys]
 
@@ -293,11 +299,13 @@ class Dataproc:
                         if initialization_actions
                         else None
                     ),
+                    oslogin_enabled=oslogin_enabled,
                 ),
                 subclusters_spec=subclusters,
             ),
             zone_id=zone,
             service_account_id=service_account_id,
+            environment=environment,
             bucket=s3_bucket,
             ui_proxy=enable_ui_proxy,
             host_group_ids=host_group_ids,

{yandexcloud-0.348.0.dist-info → yandexcloud-0.350.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: yandexcloud
-Version: 0.348.0
+Version: 0.350.0
 Summary: The Yandex Cloud official SDK
 Author-email: Yandex LLC <cloud@support.yandex.ru>
 License: MIT