codestyle fixes

Author: vgvoleg

examples/query-service/basic_example.py

@@ -26,5 +26,5 @@ def main():
         print(f"rows: {str(result_set.rows)}")
 
 
-if __name__ == '__main__':
+if __name__ == "__main__":
     main()

tests/query/test_query_transaction.py

@@ -1,5 +1,6 @@
 import pytest
 
+
 class TestQueryTransaction:
     def test_tx_begin(self, tx):
         assert tx.tx_id is None

ydb/_grpc/grpcwrapper/ydb_query.py

@@ -18,6 +18,7 @@
     ServerStatus,
 )
 
+
 @dataclass
 class CreateSessionResponse(IFromProto):
     status: Optional[ServerStatus]
@@ -68,13 +69,13 @@ def from_public(tx_mode: public_types.BaseQueryTxMode) -> "TransactionSettings":
         return TransactionSettings(tx_mode=tx_mode)
 
     def to_proto(self) -> ydb_query_pb2.TransactionSettings:
-        if self.tx_mode.name == 'snapshot_read_only':
+        if self.tx_mode.name == "snapshot_read_only":
             return ydb_query_pb2.TransactionSettings(snapshot_read_only=self.tx_mode.to_proto())
-        if self.tx_mode.name == 'serializable_read_write':
+        if self.tx_mode.name == "serializable_read_write":
             return ydb_query_pb2.TransactionSettings(serializable_read_write=self.tx_mode.to_proto())
-        if self.tx_mode.name == 'online_read_only':
+        if self.tx_mode.name == "online_read_only":
             return ydb_query_pb2.TransactionSettings(online_read_only=self.tx_mode.to_proto())
-        if self.tx_mode.name == 'stale_read_only':
+        if self.tx_mode.name == "stale_read_only":
             return ydb_query_pb2.TransactionSettings(stale_read_only=self.tx_mode.to_proto())
 
 
@@ -87,7 +88,7 @@ def to_proto(self) -> ydb_query_pb2.BeginTransactionRequest:
         return ydb_query_pb2.BeginTransactionRequest(
             session_id=self.session_id,
             tx_settings=self.tx_settings.to_proto(),
-            )
+        )
 
 
 @dataclass
@@ -146,14 +147,8 @@ class TransactionControl(IToProto):
 
     def to_proto(self) -> ydb_query_pb2.TransactionControl:
         if self.tx_id:
-            return ydb_query_pb2.TransactionControl(
-                tx_id=self.tx_id,
-                commit_tx=self.commit_tx,
-            )
-        return ydb_query_pb2.TransactionControl(
-            begin_tx=self.begin_tx.to_proto(),
-            commit_tx=self.commit_tx
-        )
+            return ydb_query_pb2.TransactionControl(tx_id=self.tx_id,commit_tx=self.commit_tx)
+        return ydb_query_pb2.TransactionControl(begin_tx=self.begin_tx.to_proto(), commit_tx=self.commit_tx)
 
 
 @dataclass

ydb/query/base.py

@@ -11,7 +11,7 @@
 from .._grpc.grpcwrapper import ydb_query
 from .._grpc.grpcwrapper.ydb_query_public_types import (
     BaseQueryTxMode,
-    QuerySerializableReadWrite
+    QuerySerializableReadWrite,
 )
 from .. import convert
 from .. import issues
@@ -78,7 +78,13 @@ def transaction(self, tx_mode: BaseQueryTxMode) -> "IQueryTxContext":
 class IQueryTxContext(abc.ABC):
 
     @abc.abstractmethod
-    def __init__(self, driver: SupportedDriverType, session_state: IQuerySessionState, session: IQuerySession, tx_mode: BaseQueryTxMode = None):
+    def __init__(
+        self,
+        driver: SupportedDriverType,
+        session_state: IQuerySessionState,
+        session: IQuerySession,
+        tx_mode: BaseQueryTxMode = None
+    ):
         pass
 
     @abc.abstractmethod
@@ -125,7 +131,9 @@ def session(self) -> IQuerySession:
         pass
 
 
-def create_execute_query_request(query: str, session_id: str, tx_id: str = None, commit_tx: bool = False, tx_mode: BaseQueryTxMode = None):
+def create_execute_query_request(
+        query: str, session_id: str, tx_id: str = None, commit_tx: bool = False, tx_mode: BaseQueryTxMode = None
+):
     if tx_id:
         req = ydb_query.ExecuteQueryRequest(
             session_id=session_id,
@@ -134,7 +142,7 @@ def create_execute_query_request(query: str, session_id: str, tx_id: str = None,
             ),
             tx_control=ydb_query.TransactionControl(
                 tx_id=tx_id,
-                commit_tx=commit_tx
+                commit_tx=commit_tx,
             ),
         )
     else:
@@ -148,7 +156,7 @@ def create_execute_query_request(query: str, session_id: str, tx_id: str = None,
                 begin_tx=ydb_query.TransactionSettings(
                     tx_mode=tx_mode,
                 ),
-                commit_tx=commit_tx
+                commit_tx=commit_tx,
             ),
         )
 

ydb/query/session.py

@@ -32,7 +32,7 @@ class QuerySessionStateHelper(abc.ABC):
     }
 
     _READY_TO_USE = [
-        QuerySessionStateEnum.CREATED
+        QuerySessionStateEnum.CREATED,
     ]
 
     @classmethod
@@ -154,7 +154,7 @@ def _execute_call(self, query: str, commit_tx: bool):
         request = base.create_execute_query_request(
             query=query,
             session_id=self._state.session_id,
-            commit_tx=commit_tx
+            commit_tx=commit_tx,
         )
 
         return self._driver(
@@ -189,13 +189,13 @@ def _attach(self):
         ).start()
 
     def _chech_session_status_loop(self, status_stream):
-            try:
-                for status in status_stream:
-                    if status.status != issues.StatusCode.SUCCESS:
-                        self._state.reset()
-                        self._state._change_state(QuerySessionStateEnum.CLOSED)
-            except Exception as e:
-                pass
+        try:
+            for status in status_stream:
+                if status.status != issues.StatusCode.SUCCESS:
+                    self._state.reset()
+                    self._state._change_state(QuerySessionStateEnum.CLOSED)
+        except Exception as e:
+            pass
 
 
     def delete(self) -> None:

ydb/query/transaction.py

@@ -275,7 +275,7 @@ def _execute_call(self, query: str, commit_tx: bool):
         request = base.create_execute_query_request(
             query=query,
             session_id=self._session_state.session_id,
-            commit_tx=commit_tx
+            commit_tx=commit_tx,
         )
         return self._driver(
             request,