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

invest-python-master/t_tech/invest/_errors.py

@@ -0,0 +1,137 @@
+from functools import wraps
+from typing import Any, Callable, TypeVar, cast
+
+from grpc import Call, RpcError, StatusCode
+from grpc.aio import AioRpcError
+
+from .exceptions import (
+    AioRequestError,
+    AioUnauthenticatedError,
+    RequestError,
+    UnauthenticatedError,
+)
+from .logging import get_metadata_from_aio_error, get_metadata_from_call, log_error
+
+TFunc = TypeVar("TFunc", bound=Callable[..., Any])
+
+
+def handle_request_error(name: str) -> Callable[[TFunc], TFunc]:
+    def decorator(func: TFunc) -> TFunc:
+        # noinspection DuplicatedCode
+        @wraps(func)
+        def wrapper(*args: Any, **kwargs: Any) -> Any:
+            try:
+                return func(*args, **kwargs)
+            except RpcError as e:
+                if issubclass(type(e), Call):
+                    metadata = get_metadata_from_call(e)
+                    tracking_id = metadata.tracking_id if metadata else None
+                    log_error(
+                        tracking_id,
+                        name,
+                        f"{e.code().name} {e.details()}",  # type:ignore
+                    )
+                    status_code = e.code()  # type:ignore
+                    details = e.details()  # type:ignore
+
+                    if status_code == StatusCode.UNAUTHENTICATED:
+                        raise UnauthenticatedError(
+                            status_code, details, metadata
+                        ) from e
+
+                    raise RequestError(status_code, details, metadata) from e
+                raise
+
+        return cast(TFunc, wrapper)
+
+    return decorator
+
+
+def handle_request_error_gen(name: str) -> Callable[[TFunc], TFunc]:
+    def decorator(func: TFunc) -> TFunc:
+        # noinspection DuplicatedCode
+        @wraps(func)
+        def wrapper(*args: Any, **kwargs: Any) -> Any:
+            try:
+                for message in func(*args, **kwargs):
+                    yield message
+            except RpcError as e:
+                if issubclass(type(e), Call):
+                    metadata = get_metadata_from_call(e)
+                    tracking_id = metadata.tracking_id if metadata else None
+                    log_error(
+                        tracking_id,
+                        name,
+                        f"{e.code().name} {e.details()}",  # type:ignore
+                    )
+                    status_code = e.code()  # type:ignore
+                    details = e.details()  # type:ignore
+
+                    if status_code == StatusCode.UNAUTHENTICATED:
+                        raise UnauthenticatedError(
+                            status_code, details, metadata
+                        ) from e
+
+                    raise RequestError(status_code, details, metadata) from e
+                raise
+
+        return cast(TFunc, wrapper)
+
+    return decorator
+
+
+def handle_aio_request_error(name: str) -> Callable[[TFunc], TFunc]:
+    def decorator(func: TFunc) -> TFunc:
+        # noinspection DuplicatedCode
+        @wraps(func)
+        async def wrapper(*args: Any, **kwargs: Any) -> Any:
+            try:
+                return await func(*args, **kwargs)
+            except AioRpcError as e:
+                metadata = get_metadata_from_aio_error(e)
+                tracking_id = metadata.tracking_id if metadata else None
+                log_error(
+                    tracking_id,
+                    name,
+                    f"{e.code().name} {e.details()}",  # type:ignore
+                )
+                status_code = e.code()  # type:ignore
+                details = e.details()  # type:ignore
+
+                if status_code == StatusCode.UNAUTHENTICATED:
+                    raise AioUnauthenticatedError(status_code, details, metadata) from e
+
+                raise AioRequestError(status_code, details, metadata) from e
+
+        return cast(TFunc, wrapper)
+
+    return decorator
+
+
+def handle_aio_request_error_gen(name: str) -> Callable[[TFunc], TFunc]:
+    def decorator(func: TFunc) -> TFunc:
+        # noinspection DuplicatedCode
+        @wraps(func)
+        async def wrapper(*args: Any, **kwargs: Any) -> Any:
+            try:
+                async for result in func(*args, **kwargs):
+                    yield result
+            except AioRpcError as e:
+                metadata = get_metadata_from_aio_error(e)
+                tracking_id = metadata.tracking_id if metadata else None
+                log_error(
+                    tracking_id,
+                    name,
+                    f"{e.code().name} {e.details()}",  # type:ignore
+                )
+                status_code = e.code()  # type:ignore
+                details = e.details()  # type:ignore
+
+                if status_code == StatusCode.UNAUTHENTICATED:
+                    raise AioUnauthenticatedError(status_code, details, metadata) from e
+
+                raise AioRequestError(status_code, details, metadata) from e
+
+        return cast(TFunc, wrapper)
+
+    return decorator