fix default truncated response

Author: rekby

ydb/_topic_common/common.py

@@ -30,7 +30,7 @@ def wrapper(rpc_state, response_pb, driver=None):
 
 
 _shared_event_loop_lock = threading.Lock()
-_shared_event_loop = None  # type: Optional[asyncio.AbstractEventLoop]
+_shared_event_loop: Optional[asyncio.AbstractEventLoop] = None
 
 
 def _get_shared_event_loop() -> asyncio.AbstractEventLoop:

ydb/aio/table.py

@@ -13,7 +13,6 @@
     _scan_query_request_factory,
     _wrap_scan_query_response,
     BaseTxContext,
-    _allow_split_transaction,
 )
 from . import _utilities
 from ydb import _apis, _session_impl
@@ -121,9 +120,7 @@ async def alter_table(
             set_read_replicas_settings,
         )
 
-    def transaction(
-        self, tx_mode=None, *, allow_split_transactions=_allow_split_transaction
-    ):
+    def transaction(self, tx_mode=None, *, allow_split_transactions=None):
         return TxContext(
             self._driver,
             self._state,

ydb/global_settings.py

@@ -5,14 +5,20 @@
 
 
 def global_allow_truncated_result(enabled: bool = True):
+    if convert._default_allow_truncated_result == enabled:
+        return
+
     if enabled:
         warnings.warn("Global allow truncated response is deprecated behaviour.")
 
     convert._default_allow_truncated_result = enabled
 
 
 def global_allow_split_transactions(enabled: bool):
+    if table._default_allow_split_transaction == enabled:
+        return
+
     if enabled:
-        warnings.warn("Global allow truncated response is deprecated behaviour.")
+        warnings.warn("Global allow split transaction is deprecated behaviour.")
 
-    table._allow_split_transaction = enabled
+    table._default_allow_split_transaction = enabled

ydb/table.py

@@ -27,7 +27,7 @@
 except ImportError:
     interceptor = None
 
-_allow_split_transaction = False
+_default_allow_split_transaction = False
 
 logger = logging.getLogger(__name__)
 
@@ -1181,9 +1181,7 @@ def execute_scheme(self, yql_text, settings=None):
         pass
 
     @abstractmethod
-    def transaction(
-        self, tx_mode=None, allow_split_transactions=_allow_split_transaction
-    ):
+    def transaction(self, tx_mode=None, allow_split_transactions=None):
         pass
 
     @abstractmethod
@@ -1687,9 +1685,7 @@ def execute_scheme(self, yql_text, settings=None):
             self._state.endpoint,
         )
 
-    def transaction(
-        self, tx_mode=None, allow_split_transactions=_allow_split_transaction
-    ):
+    def transaction(self, tx_mode=None, allow_split_transactions=None):
         return TxContext(
             self._driver,
             self._state,
@@ -2226,7 +2222,7 @@ def __init__(
         session,
         tx_mode=None,
         *,
-        allow_split_transactions=_allow_split_transaction
+        allow_split_transactions=None
     ):
         """
         An object that provides a simple transaction context manager that allows statements execution
@@ -2413,7 +2409,13 @@ def _check_split(self, allow=""):
         Deny all operaions with transaction after commit/rollback.
         Exception: double commit and double rollbacks, because it is safe
         """
-        if self._allow_split_transactions:
+        allow_split_transaction = (
+            self._allow_split_transactions
+            if self._allow_split_transactions is not None
+            else _default_allow_split_transaction
+        )
+
+        if allow_split_transaction:
             return
 
         if self._finished != "" and self._finished != allow: