make retries module public

Author: vgvoleg

ydb/__init__.py

@@ -20,6 +20,7 @@
 from .topic import *  # noqa
 from .draft import *  # noqa
 from .query import *  # noqa
+from .retries import *  # noqa
 
 try:
     import ydb.aio as aio  # noqa

ydb/_topic_reader/topic_reader.py

@@ -10,7 +10,7 @@
     Callable,
 )
 
-from .._retries import RetrySettings
+from ..retries import RetrySettings
 from .._grpc.grpcwrapper.ydb_topic import StreamReadMessage, OffsetsRange
 
 

ydb/_topic_writer/topic_writer_asyncio.py

@@ -28,7 +28,7 @@
     issues,
 )
 from .._errors import check_retriable_error
-from .._retries import RetrySettings
+from ..retries import RetrySettings
 from .._grpc.grpcwrapper.ydb_topic_public_types import PublicCodec
 from .._grpc.grpcwrapper.ydb_topic import (
     UpdateTokenRequest,

ydb/aio/table.py

@@ -15,7 +15,7 @@
     BaseTxContext,
 )
 from . import _utilities
-from ydb import _apis, _session_impl, _retries
+from ydb import _apis, _session_impl
 
 logger = logging.getLogger(__name__)
 
@@ -214,9 +214,9 @@ async def retry_operation(callee, retry_settings=None, *args, **kwargs):  # pyli
     Returns awaitable result of coroutine. If retries are not succussful exception is raised.
     """
 
-    opt_generator = _retries.retry_operation_impl(callee, retry_settings, *args, **kwargs)
+    opt_generator = ydb.retry_operation_impl(callee, retry_settings, *args, **kwargs)
     for next_opt in opt_generator:
-        if isinstance(next_opt, _retries.YdbRetryOperationSleepOpt):
+        if isinstance(next_opt, ydb.YdbRetryOperationSleepOpt):
             await asyncio.sleep(next_opt.timeout)
         else:
             try:

ydb/query/pool.py

@@ -4,7 +4,7 @@
 from .session import (
     QuerySessionSync,
 )
-from .._retries import (
+from ..retries import (
     RetrySettings,
     retry_operation_sync,
 )

ydb/_retries.py renamed to ydb/retries.py

@@ -19,7 +19,7 @@
     tracing,
 )
 
-from ._retries import (
+from .retries import (
     retry_operation_sync,
     RetrySettings,
 )

ydb/table.py

@@ -1,7 +1,7 @@
 from unittest import mock
 from . import issues
 
-from ._retries import (
+from .retries import (
     retry_operation_impl,
     YdbRetryOperationFinalResult,
     YdbRetryOperationSleepOpt,