extent public api with QueryTxContext

vgvoleg · vgvoleg · commit f601499ea963 · 2024-09-10T13:50:03.000+03:00
diff --git a/docs/apireference.rst b/docs/apireference.rst
@@ -16,7 +16,7 @@ Driver
 Driver object
 ~~~~~~~~~~~~~
 
-.. autoclass:: Driver
+.. autoclass:: ydb.Driver
     :members:
     :inherited-members:
     :undoc-members:
@@ -34,7 +34,7 @@ Driver object (AsyncIO)
 DriverConfig
 ~~~~~~~~~~~~
 
-.. autoclass:: DriverConfig
+.. autoclass:: ydb.DriverConfig
     :members:
     :inherited-members:
     :undoc-members:
@@ -48,23 +48,23 @@ Query Service
 QuerySessionPool
 ~~~~~~~~~~~~~~~~
 
-.. autoclass:: QuerySessionPool
+.. autoclass:: ydb.QuerySessionPool
     :members:
     :inherited-members:
     :undoc-members:
 
 QuerySession
 ~~~~~~~~~~~~~~~~
 
-.. autoclass:: QuerySession
+.. autoclass:: ydb.QuerySession
     :members:
     :inherited-members:
     :undoc-members:
 
 QueryTxContext
 ~~~~~~~~~~~~~~~~~~
 
-.. autoclass:: QueryTxContext
+.. autoclass:: ydb.QueryTxContext
     :members:
     :inherited-members:
     :undoc-members:
@@ -76,47 +76,47 @@ Table Service
 
 TableClient
 ~~~~~~~~~~~
-.. autoclass:: TableClient
+.. autoclass:: ydb.TableClient
     :members:
     :inherited-members:
     :undoc-members:
 
 TableClientSettings
 ~~~~~~~~~~~~~~~~~~~
 
-.. autoclass:: TableClientSettings
+.. autoclass:: ydb.TableClientSettings
     :members:
     :inherited-members:
     :undoc-members:
 
 Session Pool
 ~~~~~~~~~~~~
 
-.. autoclass:: SessionPool
+.. autoclass:: ydb.SessionPool
    :members:
    :inherited-members:
    :undoc-members:
 
 Session
 ~~~~~~~
 
-.. autoclass:: Session
+.. autoclass:: ydb.Session
    :members:
    :inherited-members:
    :undoc-members:
 
 Transaction Context
 ~~~~~~~~~~~~~~~~~~~
 
-.. autoclass:: TxContext
+.. autoclass:: ydb.TxContext
    :members:
    :inherited-members:
    :undoc-members:
 
 DataQuery
 ^^^^^^^^^
 
-.. autoclass:: DataQuery
+.. autoclass:: ydb.DataQuery
    :members:
    :inherited-members:
    :undoc-members:
@@ -129,7 +129,7 @@ Scheme
 SchemeClient
 ~~~~~~~~~~~~
 
-.. autoclass:: SchemeClient
+.. autoclass:: ydb.SchemeClient
    :members:
    :inherited-members:
    :undoc-members:
diff --git a/ydb/aio/query/__init__.py b/ydb/aio/query/__init__.py
@@ -1,7 +1,9 @@
 __all__ = [
     "QuerySessionPool",
     "QuerySession",
+    "QueryTxContext",
 ]
 
 from .pool import QuerySessionPool
 from .session import QuerySession
+from .transaction import QueryTxContext
diff --git a/ydb/query/__init__.py b/ydb/query/__init__.py
@@ -6,6 +6,7 @@
     "QuerySessionPool",
     "QueryClientSync",
     "QuerySession",
+    "QueryTxContext",
 ]
 
 import logging
@@ -15,6 +16,7 @@
 )
 
 from .session import QuerySession
+from .transaction import QueryTxContext
 
 from .._grpc.grpcwrapper import common_utils
 from .._grpc.grpcwrapper.ydb_query_public_types import (

Original file line number	Diff line number	Diff line change
`@@ -1,7 +1,9 @@`
`1`	`1`	`__all__ = [`
`2`	`2`	`"QuerySessionPool",`
`3`	`3`	`"QuerySession",`
	`4`	`+ "QueryTxContext",`
`4`	`5`	`]`
`5`	`6`
`6`	`7`	`from .pool import QuerySessionPool`
`7`	`8`	`from .session import QuerySession`
	`9`	`+from .transaction import QueryTxContext`
Original file line number	Diff line number	Diff line change
`@@ -6,6 +6,7 @@`
`6`	`6`	`"QuerySessionPool",`
`7`	`7`	`"QueryClientSync",`
`8`	`8`	`"QuerySession",`
	`9`	`+ "QueryTxContext",`
`9`	`10`	`]`
`10`	`11`
`11`	`12`	`import logging`
`@@ -15,6 +16,7 @@`
`15`	`16`	`)`
`16`	`17`
`17`	`18`	`from .session import QuerySession`
	`19`	`+from .transaction import QueryTxContext`
`18`	`20`
`19`	`21`	`from .._grpc.grpcwrapper import common_utils`
`20`	`22`	`from .._grpc.grpcwrapper.ydb_query_public_types import (`