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

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

@@ -0,0 +1,21 @@
+from t_tech.invest import AsyncClient
+from t_tech.invest.retrying.aio.grpc_interceptor import AsyncRetryClientInterceptor
+from t_tech.invest.retrying.aio.retry_manager import AsyncRetryManager
+from t_tech.invest.retrying.settings_protocol import RetryClientSettingsProtocol
+
+
+class AsyncRetryingClient(AsyncClient):
+    def __init__(
+        self,
+        token: str,
+        settings: RetryClientSettingsProtocol,
+        **kwargs,
+    ):
+        self._retry_manager = AsyncRetryManager(settings=settings)
+        self._retry_interceptor = AsyncRetryClientInterceptor(
+            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/aio/grpc_interceptor.py

@@ -0,0 +1,21 @@
+import grpc
+
+from t_tech.invest.retrying.aio.retry_manager import AsyncRetryManager
+
+
+class AsyncRetryClientInterceptor(grpc.aio.UnaryUnaryClientInterceptor):
+    def __init__(
+        self, retry_manager: AsyncRetryManager
+    ):  # pylint: disable=super-init-not-called
+        self._retry_manager = retry_manager
+
+    async def _intercept_with_retry(self, continuation, client_call_details, request):
+        async def call():
+            return await continuation(client_call_details, request)
+
+        return await self._retry_manager.call_with_retries(call=call)
+
+    async def intercept_unary_unary(self, continuation, client_call_details, request):
+        return await self._intercept_with_retry(
+            continuation, client_call_details, request
+        )

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

@@ -0,0 +1,35 @@
+import asyncio
+import logging
+from typing import Any
+
+from grpc.aio import AioRpcError
+
+from t_tech.invest.logging import get_metadata_from_aio_error
+from t_tech.invest.retrying.base_retry_manager import BaseRetryManager
+
+logger = logging.getLogger(__name__)
+
+
+class AsyncRetryManager(BaseRetryManager):
+    async def call_with_retries(self, call: Any):
+        retries_left = self.get_initial_retries()
+        while retries_left > 0:
+            logger.debug("Trying to call")
+            response = await call()
+            try:
+                await response
+                logger.debug("Call succeeded")
+                return response
+            except AioRpcError as exception:
+                retries_left -= 1
+                logger.debug("Retries left = %s", retries_left)
+
+                metadata = get_metadata_from_aio_error(exception)
+                seconds_to_sleep = self.extract_seconds_to_sleep(metadata)
+                await self._sleep(seconds_to_sleep)
+
+        logger.debug("RetryManager exhausted, no retries left")
+        return response
+
+    async def _sleep(self, seconds_to_sleep):
+        await asyncio.sleep(seconds_to_sleep)