RAPTOR v18.4: Исправлена отчетность, активированы выходные

invest-python-master/t_tech/invest/retrying/sync/client.py

@@ -0,0 +1,21 @@
+from t_tech.invest import Client
+from t_tech.invest.retrying.settings_protocol import RetryClientSettingsProtocol
+from t_tech.invest.retrying.sync.grpc_interceptor import RetryClientInterceptor
+from t_tech.invest.retrying.sync.retry_manager import RetryManager
+
+
+class RetryingClient(Client):
+    def __init__(
+        self,
+        token: str,
+        settings: RetryClientSettingsProtocol,
+        **kwargs,
+    ):
+        self._retry_manager = RetryManager(settings=settings)
+        self._retry_interceptor = RetryClientInterceptor(
+            retry_manager=self._retry_manager
+        )
+        interceptors = kwargs.get("interceptors", [])
+        interceptors.append(self._retry_interceptor)
+        kwargs["interceptors"] = interceptors
+        super().__init__(token, **kwargs)

invest-python-master/t_tech/invest/retrying/sync/grpc_interceptor.py

@@ -0,0 +1,19 @@
+import grpc
+
+from t_tech.invest.retrying.sync.retry_manager import RetryManager
+
+
+class RetryClientInterceptor(grpc.UnaryUnaryClientInterceptor):
+    def __init__(
+        self, retry_manager: RetryManager
+    ):  # pylint: disable=super-init-not-called
+        self._retry_manager = retry_manager
+
+    def _intercept_with_retry(self, continuation, client_call_details, request):
+        def call():
+            return continuation(client_call_details, request)
+
+        return self._retry_manager.call_with_retries(call=call)
+
+    def intercept_unary_unary(self, continuation, client_call_details, request):
+        return self._intercept_with_retry(continuation, client_call_details, request)

invest-python-master/t_tech/invest/retrying/sync/retry_manager.py

@@ -0,0 +1,31 @@
+import logging
+import time
+from typing import Any
+
+from t_tech.invest.logging import get_metadata_from_call
+from t_tech.invest.retrying.base_retry_manager import BaseRetryManager
+
+logger = logging.getLogger(__name__)
+
+
+class RetryManager(BaseRetryManager):
+    def call_with_retries(self, call: Any):
+        retries_left = self.get_initial_retries()
+        while retries_left > 0:
+            logger.debug("Trying to call")
+            result = call()
+            logger.debug("Call succeeded")
+            exception = result.exception()
+            if not exception:
+                return result
+            retries_left -= 1
+            logger.debug("Retries left = %s", retries_left)
+            metadata = get_metadata_from_call(exception)
+            seconds_to_sleep = self.extract_seconds_to_sleep(metadata)
+            self._sleep(seconds_to_sleep)
+
+        logger.debug("RetryManager exhausted, no retries left")
+        return result
+
+    def _sleep(self, seconds_to_sleep):
+        time.sleep(seconds_to_sleep)