yandexcloud 0.303.0__py3-none-any.whl → 0.305.0__py3-none-any.whl

yandexcloud/_operation_waiter.py

@@ -10,11 +10,14 @@ from yandex.cloud.operation.operation_service_pb2 import GetOperationRequest
 from yandex.cloud.operation.operation_service_pb2_grpc import OperationServiceStub
 from yandexcloud._backoff import backoff_exponential_jittered_min_interval
 from yandexcloud._retry_interceptor import RetryInterceptor
-from yandexcloud.operations import OperationError, OperationResult
+from yandexcloud.operations import (
+    MetaType,
+    OperationError,
+    OperationResult,
+    ResponseType,
+)
 
 if TYPE_CHECKING:
-    import google.protobuf.message
-
     from yandex.cloud.operation.operation_pb2 import Operation
     from yandexcloud._sdk import SDK
 
@@ -49,15 +52,15 @@ def wait_for_operation(sdk: "SDK", operation_id: str, timeout: Optional[float])
 def get_operation_result(
     sdk: "SDK",
     operation: "Operation",
-    response_type: Optional[Type["google.protobuf.message.Message"]] = None,
-    meta_type: Optional[Type["google.protobuf.message.Message"]] = None,
+    response_type: Optional[Type["ResponseType"]] = None,
+    meta_type: Optional[Type["MetaType"]] = None,
     timeout: Optional[float] = None,
     logger: Optional[logging.Logger] = None,
-) -> Union["OperationResult", "OperationError"]:
+) -> Union["OperationResult[ResponseType, MetaType]", "OperationError"]:
     if not logger:
         logger = logging.getLogger()
         logger.addHandler(logging.NullHandler())
-    operation_result = OperationResult(operation)
+    operation_result: OperationResult[ResponseType, MetaType] = OperationResult(operation)
     created_at = datetime.fromtimestamp(operation.created_at.seconds)
     message = (
         "Running Yandex.Cloud operation. ID: {id}. "

yandexcloud/_sdk.py

@@ -11,11 +11,15 @@ from yandexcloud._wrappers import Wrappers
 if TYPE_CHECKING:
     import logging
 
-    import google.protobuf.message
-
     from yandex.cloud.operation.operation_pb2 import Operation
     from yandexcloud._operation_waiter import OperationWaiter
-    from yandexcloud.operations import OperationError, OperationResult
+    from yandexcloud.operations import (
+        MetaType,
+        OperationError,
+        OperationResult,
+        RequestType,
+        ResponseType,
+    )
 
 
 class SDK:
@@ -96,20 +100,20 @@ class SDK:
     def wait_operation_and_get_result(
         self,
         operation: "Operation",
-        response_type: Optional[Type["google.protobuf.message.Message"]] = None,
-        meta_type: Optional[Type["google.protobuf.message.Message"]] = None,
+        response_type: Optional[Type["ResponseType"]] = None,
+        meta_type: Optional[Type["MetaType"]] = None,
         timeout: Optional[float] = None,
         logger: Optional["logging.Logger"] = None,
-    ) -> Union["OperationResult", "OperationError"]:
+    ) -> Union["OperationResult[ResponseType, MetaType]", "OperationError"]:
         return _operation_waiter.get_operation_result(self, operation, response_type, meta_type, timeout, logger)
 
     def create_operation_and_get_result(
         self,
-        request: Type["google.protobuf.message.Message"],
+        request: Type["RequestType"],
         service: Any,
         method_name: str,
-        response_type: Optional[Type["google.protobuf.message.Message"]] = None,
-        meta_type: Optional[Type["google.protobuf.message.Message"]] = None,
+        response_type: Optional[Type["ResponseType"]] = None,
+        meta_type: Optional[Type["MetaType"]] = None,
         timeout: Optional[float] = None,
         logger: Optional["logging.Logger"] = None,
     ) -> Union["OperationResult", "OperationError"]:

yandexcloud/operations.py

@@ -1,19 +1,22 @@
-from typing import TYPE_CHECKING, Optional
+from typing import TYPE_CHECKING, Generic, Optional, TypeVar
 
 if TYPE_CHECKING:
     import google.protobuf.message
 
     from yandex.cloud.operation.operation_pb2 import Operation
 
-    # from yandex.cloud.api.operation_pb2 import Operation
 
+RequestType = TypeVar("RequestType", bound="google.protobuf.message.Message")  # pylint: disable=C0103
+ResponseType = TypeVar("ResponseType", bound="google.protobuf.message.Message")  # pylint: disable=C0103
+MetaType = TypeVar("MetaType", bound="google.protobuf.message.Message")  # pylint: disable=C0103
 
-class OperationResult:
+
+class OperationResult(Generic[ResponseType, MetaType]):
     def __init__(
         self,
         operation: "Operation",
-        response: Optional["google.protobuf.message.Message"] = None,
-        meta: Optional["google.protobuf.message.Message"] = None,
+        response: Optional["ResponseType"] = None,
+        meta: Optional["MetaType"] = None,
     ):
         self.operation = operation
         self.response = response
@@ -21,7 +24,7 @@ class OperationResult:
 
 
 class OperationError(RuntimeError):
-    def __init__(self, message: str, operation_result: OperationResult):
+    def __init__(self, message: str, operation_result: OperationResult[ResponseType, MetaType]):
         super(OperationError, self).__init__(message)  # pylint: disable=super-with-arguments
         self.message = message
         self.operation_result = operation_result

{yandexcloud-0.303.0.dist-info → yandexcloud-0.305.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: yandexcloud
-Version: 0.303.0
+Version: 0.305.0
 Summary: The Yandex.Cloud official SDK
 Home-page: https://github.com/yandex-cloud/python-sdk
 Author: Yandex LLC

{yandexcloud-0.303.0.dist-info → yandexcloud-0.305.0.dist-info}/RECORD

@@ -2418,17 +2418,17 @@ yandexcloud/_auth_plugin.py,sha256=ASc_uAcfoXKtC7w7JOEQeHSvNYB3h7YepVG2LbrZoyc,3
 yandexcloud/_backoff.py,sha256=dXoKZZOFJnEjc2VH35zfHjpJ3hwN2kecy9qzDF-QSso,1246
 yandexcloud/_channels.py,sha256=p-A6JRG5V_z0tBgsFJDxslYDGsofsOCBm1PCts_ihfI,5017
 yandexcloud/_helpers.py,sha256=EeTeR_m8FvpWMSrOW4mzVlCdsTaEvg4YOEOy_VnnFpk,3054
-yandexcloud/_operation_waiter.py,sha256=5nqfltMmnb_KOvYrFSEvmJYvCnZTxRlV4s2uM37K_Vo,4882
+yandexcloud/_operation_waiter.py,sha256=RqfzO51WP6Ui6auwanJ8NZ_-97znWA1okZtBKZRjGcI,4914
 yandexcloud/_retry_interceptor.py,sha256=UdRxyLSfX6EZrvFNTfVkB_LprCSilWRtxIyRhLwpmXU,7642
-yandexcloud/_sdk.py,sha256=GccAxhCobtk0RelX2jCbdRKLORtmFTUQmS0V-zEtmRU,7844
+yandexcloud/_sdk.py,sha256=eESvsOt-W4-fyRUWaHs8KTdm_ELFyRPHtI1Z6bnZLiU,7814
 yandexcloud/auth.py,sha256=SD_IVGUhN_Ny8f-qtnNWtZQ1VPYrGbKypFAbQcaSQ58,1008
-yandexcloud/operations.py,sha256=ZSvEhe_JMBefkDLcFtSHnOAoY2g_MVVVM1qcDUtpvgs,831
+yandexcloud/operations.py,sha256=rOlhJT3Nx7u3hjRY4tnMc1_Z-yctcjcCkc_7GsDpSwY,1111
 yandexcloud/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 yandexcloud/_wrappers/__init__.py,sha256=z19vuf3vgVGWFc-0d1LFsyVXiM-SYlhvvWRQI8l1T_8,428
 yandexcloud/_wrappers/dataproc/__init__.py,sha256=QWM1jhd8dHwTEjhdiAalreNzac1g4jJ5djXoDdXRdfg,33016
-yandexcloud-0.303.0.dist-info/AUTHORS,sha256=0o7IPkgdTswBveGhbt_73xRPpCrfeEGoZE11AiLvYSQ,231
-yandexcloud-0.303.0.dist-info/LICENSE,sha256=AFcOYhNOyuBQP89lObqyipdScN2KUUS-OuWoUlVo6yE,1077
-yandexcloud-0.303.0.dist-info/METADATA,sha256=RJpiT_MLyjeSgHQ63EOVr1xY2Ue_IY84KDp0nkp77oE,10419
-yandexcloud-0.303.0.dist-info/WHEEL,sha256=Z4pYXqR_rTB7OWNDYFOm1qRk0RX6GFP2o8LgvP453Hk,91
-yandexcloud-0.303.0.dist-info/top_level.txt,sha256=p6aBMPGD526A1jM2WVnAneI2qO4kGDWeJi6uwYApDqg,19
-yandexcloud-0.303.0.dist-info/RECORD,,
+yandexcloud-0.305.0.dist-info/AUTHORS,sha256=0o7IPkgdTswBveGhbt_73xRPpCrfeEGoZE11AiLvYSQ,231
+yandexcloud-0.305.0.dist-info/LICENSE,sha256=AFcOYhNOyuBQP89lObqyipdScN2KUUS-OuWoUlVo6yE,1077
+yandexcloud-0.305.0.dist-info/METADATA,sha256=HPC5rbSo3q4rzgOQfldiqHuzaszs3UTG72l3iPSDdN4,10419
+yandexcloud-0.305.0.dist-info/WHEEL,sha256=Z4pYXqR_rTB7OWNDYFOm1qRk0RX6GFP2o8LgvP453Hk,91
+yandexcloud-0.305.0.dist-info/top_level.txt,sha256=p6aBMPGD526A1jM2WVnAneI2qO4kGDWeJi6uwYApDqg,19
+yandexcloud-0.305.0.dist-info/RECORD,,