snowflake-connector-python 5.2.0__cp315-cp315-win_amd64.whl
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- snowflake/connector/__init__.py +148 -0
- snowflake/connector/_common/__init__.py +0 -0
- snowflake/connector/_common/extras.py +134 -0
- snowflake/connector/_common/telemetry.py +221 -0
- snowflake/connector/_common/wif_util.py +92 -0
- snowflake/connector/_core/sf_core_python.cp315-win_amd64.pyd +0 -0
- snowflake/connector/_core/sf_core_python.pyi +79 -0
- snowflake/connector/_internal/__init__.py +6 -0
- snowflake/connector/_internal/_private_key_helper.py +31 -0
- snowflake/connector/_internal/api_client/__init__.py +93 -0
- snowflake/connector/_internal/api_client/bridge.py +41 -0
- snowflake/connector/_internal/api_client/client_api.py +976 -0
- snowflake/connector/_internal/arrow.py +47 -0
- snowflake/connector/_internal/arrow_context.py +240 -0
- snowflake/connector/_internal/arrow_stream_async.py +266 -0
- snowflake/connector/_internal/arrow_stream_iterator.cp311-win_amd64.pyd +0 -0
- snowflake/connector/_internal/arrow_stream_iterator.cp312-win_amd64.pyd +0 -0
- snowflake/connector/_internal/arrow_stream_iterator.cp313-win_amd64.pyd +0 -0
- snowflake/connector/_internal/arrow_stream_iterator.cp314-win_amd64.pyd +0 -0
- snowflake/connector/_internal/arrow_stream_iterator.cp314t-win_amd64.pyd +0 -0
- snowflake/connector/_internal/arrow_stream_iterator.cp315-win_amd64.pyd +0 -0
- snowflake/connector/_internal/arrow_stream_iterator.pyi +42 -0
- snowflake/connector/_internal/arrow_stream_utils.py +171 -0
- snowflake/connector/_internal/authenticator_type.py +53 -0
- snowflake/connector/_internal/backward_compatibility.py +245 -0
- snowflake/connector/_internal/binding_converters.py +741 -0
- snowflake/connector/_internal/config_utils.py +71 -0
- snowflake/connector/_internal/connection/__init__.py +39 -0
- snowflake/connector/_internal/connection/connection.py +616 -0
- snowflake/connector/_internal/connection/connection_types.py +13 -0
- snowflake/connector/_internal/connection/constants.py +25 -0
- snowflake/connector/_internal/connection/decorators.py +31 -0
- snowflake/connector/_internal/connection/freezable_proxy.py +104 -0
- snowflake/connector/_internal/connection/queries.py +10 -0
- snowflake/connector/_internal/connection_config_mixin.py +755 -0
- snowflake/connector/_internal/cursor/__init__.py +25 -0
- snowflake/connector/_internal/cursor/base.py +732 -0
- snowflake/connector/_internal/cursor/cursor_types.py +13 -0
- snowflake/connector/_internal/cursor/decorators.py +56 -0
- snowflake/connector/_internal/cursor/query_result.py +202 -0
- snowflake/connector/_internal/cursor/query_result_waiter.py +98 -0
- snowflake/connector/_internal/cursor/result_metadata.py +255 -0
- snowflake/connector/_internal/cursor/result_set_wrapper.py +66 -0
- snowflake/connector/_internal/decorators.py +539 -0
- snowflake/connector/_internal/error_kinds.py +111 -0
- snowflake/connector/_internal/errorcode.py +35 -0
- snowflake/connector/_internal/errorhandler.py +278 -0
- snowflake/connector/_internal/fips.py +28 -0
- snowflake/connector/_internal/logging/__init__.py +52 -0
- snowflake/connector/_internal/logging/config.py +134 -0
- snowflake/connector/_internal/logging/core_logger.py +218 -0
- snowflake/connector/_internal/logging/native_extension_logger.py +144 -0
- snowflake/connector/_internal/logout_config_mapping.py +187 -0
- snowflake/connector/_internal/nanoarrow_cpp/ArrowIterator/ArrayConverter.cpp +57 -0
- snowflake/connector/_internal/nanoarrow_cpp/ArrowIterator/ArrayConverter.hpp +29 -0
- snowflake/connector/_internal/nanoarrow_cpp/ArrowIterator/ArrowTableConverter.cpp +943 -0
- snowflake/connector/_internal/nanoarrow_cpp/ArrowIterator/ArrowTableConverter.hpp +63 -0
- snowflake/connector/_internal/nanoarrow_cpp/ArrowIterator/BinaryConverter.cpp +18 -0
- snowflake/connector/_internal/nanoarrow_cpp/ArrowIterator/BinaryConverter.hpp +26 -0
- snowflake/connector/_internal/nanoarrow_cpp/ArrowIterator/BooleanConverter.cpp +21 -0
- snowflake/connector/_internal/nanoarrow_cpp/ArrowIterator/BooleanConverter.hpp +23 -0
- snowflake/connector/_internal/nanoarrow_cpp/ArrowIterator/CArrowIterator.cpp +118 -0
- snowflake/connector/_internal/nanoarrow_cpp/ArrowIterator/CArrowIterator.hpp +116 -0
- snowflake/connector/_internal/nanoarrow_cpp/ArrowIterator/CArrowStreamIterator.cpp +323 -0
- snowflake/connector/_internal/nanoarrow_cpp/ArrowIterator/CArrowStreamIterator.hpp +133 -0
- snowflake/connector/_internal/nanoarrow_cpp/ArrowIterator/CArrowStreamTableIterator.cpp +275 -0
- snowflake/connector/_internal/nanoarrow_cpp/ArrowIterator/CArrowStreamTableIterator.hpp +168 -0
- snowflake/connector/_internal/nanoarrow_cpp/ArrowIterator/ConverterUtil.cpp +439 -0
- snowflake/connector/_internal/nanoarrow_cpp/ArrowIterator/ConverterUtil.hpp +30 -0
- snowflake/connector/_internal/nanoarrow_cpp/ArrowIterator/DateConverter.cpp +48 -0
- snowflake/connector/_internal/nanoarrow_cpp/ArrowIterator/DateConverter.hpp +46 -0
- snowflake/connector/_internal/nanoarrow_cpp/ArrowIterator/DecFloatConverter.cpp +76 -0
- snowflake/connector/_internal/nanoarrow_cpp/ArrowIterator/DecFloatConverter.hpp +35 -0
- snowflake/connector/_internal/nanoarrow_cpp/ArrowIterator/DecimalConverter.cpp +90 -0
- snowflake/connector/_internal/nanoarrow_cpp/ArrowIterator/DecimalConverter.hpp +63 -0
- snowflake/connector/_internal/nanoarrow_cpp/ArrowIterator/FixedSizeListConverter.cpp +62 -0
- snowflake/connector/_internal/nanoarrow_cpp/ArrowIterator/FixedSizeListConverter.hpp +28 -0
- snowflake/connector/_internal/nanoarrow_cpp/ArrowIterator/FloatConverter.cpp +29 -0
- snowflake/connector/_internal/nanoarrow_cpp/ArrowIterator/FloatConverter.hpp +35 -0
- snowflake/connector/_internal/nanoarrow_cpp/ArrowIterator/IColumnConverter.hpp +17 -0
- snowflake/connector/_internal/nanoarrow_cpp/ArrowIterator/IntConverter.cpp +23 -0
- snowflake/connector/_internal/nanoarrow_cpp/ArrowIterator/IntConverter.hpp +41 -0
- snowflake/connector/_internal/nanoarrow_cpp/ArrowIterator/IntervalConverter.cpp +65 -0
- snowflake/connector/_internal/nanoarrow_cpp/ArrowIterator/IntervalConverter.hpp +54 -0
- snowflake/connector/_internal/nanoarrow_cpp/ArrowIterator/LICENSE.txt +209 -0
- snowflake/connector/_internal/nanoarrow_cpp/ArrowIterator/MapConverter.cpp +79 -0
- snowflake/connector/_internal/nanoarrow_cpp/ArrowIterator/MapConverter.hpp +30 -0
- snowflake/connector/_internal/nanoarrow_cpp/ArrowIterator/ObjectConverter.cpp +52 -0
- snowflake/connector/_internal/nanoarrow_cpp/ArrowIterator/ObjectConverter.hpp +29 -0
- snowflake/connector/_internal/nanoarrow_cpp/ArrowIterator/Python/Common.cpp +8 -0
- snowflake/connector/_internal/nanoarrow_cpp/ArrowIterator/Python/Common.hpp +117 -0
- snowflake/connector/_internal/nanoarrow_cpp/ArrowIterator/Python/Helpers.cpp +54 -0
- snowflake/connector/_internal/nanoarrow_cpp/ArrowIterator/Python/Helpers.hpp +34 -0
- snowflake/connector/_internal/nanoarrow_cpp/ArrowIterator/SnowflakeType.cpp +33 -0
- snowflake/connector/_internal/nanoarrow_cpp/ArrowIterator/SnowflakeType.hpp +48 -0
- snowflake/connector/_internal/nanoarrow_cpp/ArrowIterator/StringConverter.cpp +18 -0
- snowflake/connector/_internal/nanoarrow_cpp/ArrowIterator/StringConverter.hpp +26 -0
- snowflake/connector/_internal/nanoarrow_cpp/ArrowIterator/TimeConverter.cpp +39 -0
- snowflake/connector/_internal/nanoarrow_cpp/ArrowIterator/TimeConverter.hpp +31 -0
- snowflake/connector/_internal/nanoarrow_cpp/ArrowIterator/TimeStampConverter.cpp +326 -0
- snowflake/connector/_internal/nanoarrow_cpp/ArrowIterator/TimeStampConverter.hpp +138 -0
- snowflake/connector/_internal/nanoarrow_cpp/ArrowIterator/Util/macros.hpp +14 -0
- snowflake/connector/_internal/nanoarrow_cpp/ArrowIterator/Util/time.cpp +65 -0
- snowflake/connector/_internal/nanoarrow_cpp/ArrowIterator/Util/time.hpp +68 -0
- snowflake/connector/_internal/nanoarrow_cpp/ArrowIterator/arrow_stream_iterator.cpp +13278 -0
- snowflake/connector/_internal/nanoarrow_cpp/ArrowIterator/arrow_stream_iterator.pyx +204 -0
- snowflake/connector/_internal/nanoarrow_cpp/ArrowIterator/flatcc/flatcc_accessors.h +109 -0
- snowflake/connector/_internal/nanoarrow_cpp/ArrowIterator/flatcc/flatcc_alloc.h +125 -0
- snowflake/connector/_internal/nanoarrow_cpp/ArrowIterator/flatcc/flatcc_assert.h +45 -0
- snowflake/connector/_internal/nanoarrow_cpp/ArrowIterator/flatcc/flatcc_builder.h +1914 -0
- snowflake/connector/_internal/nanoarrow_cpp/ArrowIterator/flatcc/flatcc_emitter.h +208 -0
- snowflake/connector/_internal/nanoarrow_cpp/ArrowIterator/flatcc/flatcc_endian.h +128 -0
- snowflake/connector/_internal/nanoarrow_cpp/ArrowIterator/flatcc/flatcc_epilogue.h +7 -0
- snowflake/connector/_internal/nanoarrow_cpp/ArrowIterator/flatcc/flatcc_flatbuffers.h +58 -0
- snowflake/connector/_internal/nanoarrow_cpp/ArrowIterator/flatcc/flatcc_identifier.h +144 -0
- snowflake/connector/_internal/nanoarrow_cpp/ArrowIterator/flatcc/flatcc_iov.h +31 -0
- snowflake/connector/_internal/nanoarrow_cpp/ArrowIterator/flatcc/flatcc_prologue.h +8 -0
- snowflake/connector/_internal/nanoarrow_cpp/ArrowIterator/flatcc/flatcc_refmap.h +144 -0
- snowflake/connector/_internal/nanoarrow_cpp/ArrowIterator/flatcc/flatcc_rtconfig.h +160 -0
- snowflake/connector/_internal/nanoarrow_cpp/ArrowIterator/flatcc/flatcc_types.h +98 -0
- snowflake/connector/_internal/nanoarrow_cpp/ArrowIterator/flatcc/flatcc_verifier.h +243 -0
- snowflake/connector/_internal/nanoarrow_cpp/ArrowIterator/flatcc/portable/flatcc_portable.h +14 -0
- snowflake/connector/_internal/nanoarrow_cpp/ArrowIterator/flatcc/portable/paligned_alloc.h +207 -0
- snowflake/connector/_internal/nanoarrow_cpp/ArrowIterator/flatcc/portable/pattributes.h +80 -0
- snowflake/connector/_internal/nanoarrow_cpp/ArrowIterator/flatcc/portable/pdiagnostic.h +83 -0
- snowflake/connector/_internal/nanoarrow_cpp/ArrowIterator/flatcc/portable/pdiagnostic_pop.h +20 -0
- snowflake/connector/_internal/nanoarrow_cpp/ArrowIterator/flatcc/portable/pdiagnostic_push.h +51 -0
- snowflake/connector/_internal/nanoarrow_cpp/ArrowIterator/flatcc/portable/pendian.h +201 -0
- snowflake/connector/_internal/nanoarrow_cpp/ArrowIterator/flatcc/portable/pendian_detect.h +108 -0
- snowflake/connector/_internal/nanoarrow_cpp/ArrowIterator/flatcc/portable/pinline.h +19 -0
- snowflake/connector/_internal/nanoarrow_cpp/ArrowIterator/flatcc/portable/pinttypes.h +52 -0
- snowflake/connector/_internal/nanoarrow_cpp/ArrowIterator/flatcc/portable/portable.h +2 -0
- snowflake/connector/_internal/nanoarrow_cpp/ArrowIterator/flatcc/portable/portable_basic.h +25 -0
- snowflake/connector/_internal/nanoarrow_cpp/ArrowIterator/flatcc/portable/pstatic_assert.h +67 -0
- snowflake/connector/_internal/nanoarrow_cpp/ArrowIterator/flatcc/portable/pstdalign.h +164 -0
- snowflake/connector/_internal/nanoarrow_cpp/ArrowIterator/flatcc/portable/pstdint.h +926 -0
- snowflake/connector/_internal/nanoarrow_cpp/ArrowIterator/flatcc/portable/punaligned.h +182 -0
- snowflake/connector/_internal/nanoarrow_cpp/ArrowIterator/flatcc/portable/pversion.h +6 -0
- snowflake/connector/_internal/nanoarrow_cpp/ArrowIterator/flatcc/portable/pwarnings.h +52 -0
- snowflake/connector/_internal/nanoarrow_cpp/ArrowIterator/flatcc.c +3052 -0
- snowflake/connector/_internal/nanoarrow_cpp/ArrowIterator/nanoarrow.c +3019 -0
- snowflake/connector/_internal/nanoarrow_cpp/ArrowIterator/nanoarrow.h +3354 -0
- snowflake/connector/_internal/nanoarrow_cpp/ArrowIterator/nanoarrow.hpp +344 -0
- snowflake/connector/_internal/nanoarrow_cpp/ArrowIterator/nanoarrow_ipc.c +29305 -0
- snowflake/connector/_internal/nanoarrow_cpp/ArrowIterator/nanoarrow_ipc.h +381 -0
- snowflake/connector/_internal/nanoarrow_cpp/Logging/logging.cpp +108 -0
- snowflake/connector/_internal/nanoarrow_cpp/Logging/logging.hpp +47 -0
- snowflake/connector/_internal/nanoarrow_cpp/scripts/.clang-format +8 -0
- snowflake/connector/_internal/nanoarrow_cpp/scripts/format.sh +8 -0
- snowflake/connector/_internal/protobuf_gen/__init__.py +0 -0
- snowflake/connector/_internal/protobuf_gen/database_driver_v1_pb2.py +408 -0
- snowflake/connector/_internal/protobuf_gen/database_driver_v1_pb2.pyi +4487 -0
- snowflake/connector/_internal/protobuf_gen/database_driver_v1_services.py +2088 -0
- snowflake/connector/_internal/protobuf_gen/proto_exception.py +17 -0
- snowflake/connector/_internal/result_batch.py +150 -0
- snowflake/connector/_internal/snowflake_restful.py +222 -0
- snowflake/connector/_internal/snowpark_backcompat_helpers.py +31 -0
- snowflake/connector/_internal/sqlstate.py +11 -0
- snowflake/connector/_internal/statement_utils.py +207 -0
- snowflake/connector/_internal/text_utils.py +112 -0
- snowflake/connector/_internal/type_codes.py +139 -0
- snowflake/connector/_internal/utils.py +65 -0
- snowflake/connector/_internal/write_pandas_operation.py +657 -0
- snowflake/connector/aio/README.md +168 -0
- snowflake/connector/aio/__init__.py +37 -0
- snowflake/connector/aio/connection/__init__.py +17 -0
- snowflake/connector/aio/connection/_connection.py +506 -0
- snowflake/connector/aio/cursor/__init__.py +35 -0
- snowflake/connector/aio/cursor/_base.py +1034 -0
- snowflake/connector/aio/cursor/_dict_cursor.py +78 -0
- snowflake/connector/aio/cursor/_result_set_wrapper.py +35 -0
- snowflake/connector/aio/cursor/_snowflake_cursor.py +72 -0
- snowflake/connector/aio/pandas_tools.py +279 -0
- snowflake/connector/aio/result_batch.py +165 -0
- snowflake/connector/auth/__init__.py +0 -0
- snowflake/connector/auth/workload_identity.py +5 -0
- snowflake/connector/compat.py +22 -0
- snowflake/connector/config_manager.py +392 -0
- snowflake/connector/connection/__init__.py +30 -0
- snowflake/connector/connection/_connection.py +641 -0
- snowflake/connector/connection_config.py +517 -0
- snowflake/connector/constants.py +52 -0
- snowflake/connector/cursor/__init__.py +37 -0
- snowflake/connector/cursor/_base.py +1169 -0
- snowflake/connector/cursor/_chunked_download_reader.py +75 -0
- snowflake/connector/cursor/_dict_cursor.py +78 -0
- snowflake/connector/cursor/_result_set_wrapper.py +35 -0
- snowflake/connector/cursor/_snowflake_cursor.py +72 -0
- snowflake/connector/description.py +7 -0
- snowflake/connector/errors.py +492 -0
- snowflake/connector/network.py +71 -0
- snowflake/connector/pandas_tools.py +191 -0
- snowflake/connector/py.typed +0 -0
- snowflake/connector/result_batch.py +216 -0
- snowflake/connector/telemetry.py +40 -0
- snowflake/connector/time_util.py +11 -0
- snowflake/connector/types.py +134 -0
- snowflake/connector/util_text.py +3 -0
- snowflake/connector/version.py +23 -0
- snowflake_connector_python-5.2.0.dist-info/DELVEWHEEL +2 -0
- snowflake_connector_python-5.2.0.dist-info/METADATA +190 -0
- snowflake_connector_python-5.2.0.dist-info/RECORD +204 -0
- snowflake_connector_python-5.2.0.dist-info/WHEEL +4 -0
- snowflake_connector_python.libs/msvcp140-a4c2229bdc2a2a630acdc095b4d86008.dll +0 -0
|
@@ -0,0 +1,2088 @@
|
|
|
1
|
+
from abc import ABC, abstractmethod
|
|
2
|
+
from typing import Optional
|
|
3
|
+
from google.protobuf import message
|
|
4
|
+
from .database_driver_v1_pb2 import *
|
|
5
|
+
from .proto_exception import *
|
|
6
|
+
|
|
7
|
+
class ProtoError(Exception):
|
|
8
|
+
def __init__(self, error_type: str, details: str):
|
|
9
|
+
self.error_type = error_type
|
|
10
|
+
self.details = details
|
|
11
|
+
super().__init__(f"{error_type}: {details}")
|
|
12
|
+
|
|
13
|
+
class DatabaseDriver(ABC):
|
|
14
|
+
@abstractmethod
|
|
15
|
+
def driver_get_fips_status(self, request: DriverGetFipsStatusRequest) -> DriverGetFipsStatusResponse:
|
|
16
|
+
pass
|
|
17
|
+
|
|
18
|
+
@abstractmethod
|
|
19
|
+
def database_new(self, request: DatabaseNewRequest) -> DatabaseNewResponse:
|
|
20
|
+
pass
|
|
21
|
+
|
|
22
|
+
@abstractmethod
|
|
23
|
+
def database_set_options(self, request: DatabaseSetOptionsRequest) -> DatabaseSetOptionsResponse:
|
|
24
|
+
pass
|
|
25
|
+
|
|
26
|
+
@abstractmethod
|
|
27
|
+
def database_init(self, request: DatabaseInitRequest) -> DatabaseInitResponse:
|
|
28
|
+
pass
|
|
29
|
+
|
|
30
|
+
@abstractmethod
|
|
31
|
+
def database_release(self, request: DatabaseReleaseRequest) -> DatabaseReleaseResponse:
|
|
32
|
+
pass
|
|
33
|
+
|
|
34
|
+
@abstractmethod
|
|
35
|
+
def database_fetch_chunk(self, request: DatabaseFetchChunkRequest) -> DatabaseFetchChunkResponse:
|
|
36
|
+
pass
|
|
37
|
+
|
|
38
|
+
@abstractmethod
|
|
39
|
+
def connection_new(self, request: ConnectionNewRequest) -> ConnectionNewResponse:
|
|
40
|
+
pass
|
|
41
|
+
|
|
42
|
+
@abstractmethod
|
|
43
|
+
def connection_set_options(self, request: ConnectionSetOptionsRequest) -> ConnectionSetOptionsResponse:
|
|
44
|
+
pass
|
|
45
|
+
|
|
46
|
+
@abstractmethod
|
|
47
|
+
def connection_init(self, request: ConnectionInitRequest) -> ConnectionInitResponse:
|
|
48
|
+
pass
|
|
49
|
+
|
|
50
|
+
@abstractmethod
|
|
51
|
+
def connection_release(self, request: ConnectionReleaseRequest) -> ConnectionReleaseResponse:
|
|
52
|
+
pass
|
|
53
|
+
|
|
54
|
+
@abstractmethod
|
|
55
|
+
def connection_close(self, request: ConnectionCloseRequest) -> ConnectionCloseResponse:
|
|
56
|
+
pass
|
|
57
|
+
|
|
58
|
+
@abstractmethod
|
|
59
|
+
def connection_is_closed(self, request: ConnectionIsClosedRequest) -> ConnectionIsClosedResponse:
|
|
60
|
+
pass
|
|
61
|
+
|
|
62
|
+
@abstractmethod
|
|
63
|
+
def connection_is_expired(self, request: ConnectionIsExpiredRequest) -> ConnectionIsExpiredResponse:
|
|
64
|
+
pass
|
|
65
|
+
|
|
66
|
+
@abstractmethod
|
|
67
|
+
def connection_get_info(self, request: ConnectionGetInfoRequest) -> ConnectionGetInfoResponse:
|
|
68
|
+
pass
|
|
69
|
+
|
|
70
|
+
@abstractmethod
|
|
71
|
+
def connection_get_objects(self, request: ConnectionGetObjectsRequest) -> ConnectionGetObjectsResponse:
|
|
72
|
+
pass
|
|
73
|
+
|
|
74
|
+
@abstractmethod
|
|
75
|
+
def connection_get_table_schema(self, request: ConnectionGetTableSchemaRequest) -> ConnectionGetTableSchemaResponse:
|
|
76
|
+
pass
|
|
77
|
+
|
|
78
|
+
@abstractmethod
|
|
79
|
+
def connection_get_table_types(self, request: ConnectionGetTableTypesRequest) -> ConnectionGetTableTypesResponse:
|
|
80
|
+
pass
|
|
81
|
+
|
|
82
|
+
@abstractmethod
|
|
83
|
+
def connection_commit(self, request: ConnectionCommitRequest) -> ConnectionCommitResponse:
|
|
84
|
+
pass
|
|
85
|
+
|
|
86
|
+
@abstractmethod
|
|
87
|
+
def connection_rollback(self, request: ConnectionRollbackRequest) -> ConnectionRollbackResponse:
|
|
88
|
+
pass
|
|
89
|
+
|
|
90
|
+
@abstractmethod
|
|
91
|
+
def connection_set_autocommit(self, request: ConnectionSetAutocommitRequest) -> ConnectionSetAutocommitResponse:
|
|
92
|
+
pass
|
|
93
|
+
|
|
94
|
+
@abstractmethod
|
|
95
|
+
def connection_use_database(self, request: ConnectionUseDatabaseRequest) -> ConnectionUseDatabaseResponse:
|
|
96
|
+
pass
|
|
97
|
+
|
|
98
|
+
@abstractmethod
|
|
99
|
+
def connection_use_schema(self, request: ConnectionUseSchemaRequest) -> ConnectionUseSchemaResponse:
|
|
100
|
+
pass
|
|
101
|
+
|
|
102
|
+
@abstractmethod
|
|
103
|
+
def connection_set_session_parameters(self, request: ConnectionSetSessionParametersRequest) -> ConnectionSetSessionParametersResponse:
|
|
104
|
+
pass
|
|
105
|
+
|
|
106
|
+
@abstractmethod
|
|
107
|
+
def connection_get_parameter(self, request: ConnectionGetParameterRequest) -> ConnectionGetParameterResponse:
|
|
108
|
+
pass
|
|
109
|
+
|
|
110
|
+
@abstractmethod
|
|
111
|
+
def connection_get_all_parameters(self, request: ConnectionGetAllParametersRequest) -> ConnectionGetAllParametersResponse:
|
|
112
|
+
pass
|
|
113
|
+
|
|
114
|
+
@abstractmethod
|
|
115
|
+
def connection_get_server_version(self, request: ConnectionGetServerVersionRequest) -> ConnectionGetServerVersionResponse:
|
|
116
|
+
pass
|
|
117
|
+
|
|
118
|
+
@abstractmethod
|
|
119
|
+
def connection_validate_options(self, request: ConnectionValidateOptionsRequest) -> ConnectionValidateOptionsResponse:
|
|
120
|
+
pass
|
|
121
|
+
|
|
122
|
+
@abstractmethod
|
|
123
|
+
def connection_get_query_result(self, request: ConnectionGetQueryResultRequest) -> ExecuteQueryResponse:
|
|
124
|
+
pass
|
|
125
|
+
|
|
126
|
+
@abstractmethod
|
|
127
|
+
def connection_get_result_set(self, request: ConnectionGetResultSetRequest) -> ResultSetResponse:
|
|
128
|
+
pass
|
|
129
|
+
|
|
130
|
+
@abstractmethod
|
|
131
|
+
def connection_get_query_status(self, request: ConnectionGetQueryStatusRequest) -> ConnectionGetQueryStatusResponse:
|
|
132
|
+
pass
|
|
133
|
+
|
|
134
|
+
@abstractmethod
|
|
135
|
+
def connection_abort_query(self, request: ConnectionAbortQueryRequest) -> ConnectionAbortQueryResponse:
|
|
136
|
+
pass
|
|
137
|
+
|
|
138
|
+
@abstractmethod
|
|
139
|
+
def connection_upload_stream_begin(self, request: ConnectionUploadStreamBeginRequest) -> ConnectionUploadStreamBeginResponse:
|
|
140
|
+
pass
|
|
141
|
+
|
|
142
|
+
@abstractmethod
|
|
143
|
+
def connection_upload_stream_chunk(self, request: ConnectionUploadStreamChunkRequest) -> ConnectionUploadStreamChunkResponse:
|
|
144
|
+
pass
|
|
145
|
+
|
|
146
|
+
@abstractmethod
|
|
147
|
+
def connection_upload_stream_finish(self, request: ConnectionUploadStreamFinishRequest) -> ConnectionUploadStreamFinishResponse:
|
|
148
|
+
pass
|
|
149
|
+
|
|
150
|
+
@abstractmethod
|
|
151
|
+
def connection_upload_stream_abort(self, request: ConnectionUploadStreamAbortRequest) -> ConnectionUploadStreamAbortResponse:
|
|
152
|
+
pass
|
|
153
|
+
|
|
154
|
+
@abstractmethod
|
|
155
|
+
def connection_download_stream_begin(self, request: ConnectionDownloadStreamBeginRequest) -> ConnectionDownloadStreamBeginResponse:
|
|
156
|
+
pass
|
|
157
|
+
|
|
158
|
+
@abstractmethod
|
|
159
|
+
def connection_download_stream_chunk(self, request: ConnectionDownloadStreamChunkRequest) -> ConnectionDownloadStreamChunkResponse:
|
|
160
|
+
pass
|
|
161
|
+
|
|
162
|
+
@abstractmethod
|
|
163
|
+
def connection_download_stream_close(self, request: ConnectionDownloadStreamCloseRequest) -> ConnectionDownloadStreamCloseResponse:
|
|
164
|
+
pass
|
|
165
|
+
|
|
166
|
+
@abstractmethod
|
|
167
|
+
def connection_send_http(self, request: ConnectionSendHttpRequest) -> ConnectionSendHttpResponse:
|
|
168
|
+
pass
|
|
169
|
+
|
|
170
|
+
@abstractmethod
|
|
171
|
+
def connection_request_token(self, request: ConnectionTokenRequest) -> ConnectionTokenResponse:
|
|
172
|
+
pass
|
|
173
|
+
|
|
174
|
+
@abstractmethod
|
|
175
|
+
def connection_heartbeat(self, request: ConnectionHeartbeatRequest) -> ConnectionHeartbeatResponse:
|
|
176
|
+
pass
|
|
177
|
+
|
|
178
|
+
@abstractmethod
|
|
179
|
+
def statement_new(self, request: StatementNewRequest) -> StatementNewResponse:
|
|
180
|
+
pass
|
|
181
|
+
|
|
182
|
+
@abstractmethod
|
|
183
|
+
def statement_release(self, request: StatementReleaseRequest) -> StatementReleaseResponse:
|
|
184
|
+
pass
|
|
185
|
+
|
|
186
|
+
@abstractmethod
|
|
187
|
+
def statement_set_sql_query(self, request: StatementSetSqlQueryRequest) -> StatementSetSqlQueryResponse:
|
|
188
|
+
pass
|
|
189
|
+
|
|
190
|
+
@abstractmethod
|
|
191
|
+
def statement_set_substrait_plan(self, request: StatementSetSubstraitPlanRequest) -> StatementSetSubstraitPlanResponse:
|
|
192
|
+
pass
|
|
193
|
+
|
|
194
|
+
@abstractmethod
|
|
195
|
+
def statement_prepare(self, request: StatementPrepareRequest) -> StatementPrepareResponse:
|
|
196
|
+
pass
|
|
197
|
+
|
|
198
|
+
@abstractmethod
|
|
199
|
+
def statement_set_options(self, request: StatementSetOptionsRequest) -> StatementSetOptionsResponse:
|
|
200
|
+
pass
|
|
201
|
+
|
|
202
|
+
@abstractmethod
|
|
203
|
+
def statement_get_parameter_schema(self, request: StatementGetParameterSchemaRequest) -> StatementGetParameterSchemaResponse:
|
|
204
|
+
pass
|
|
205
|
+
|
|
206
|
+
@abstractmethod
|
|
207
|
+
def statement_execute_query(self, request: StatementExecuteQueryRequest) -> ExecuteQueryResponse:
|
|
208
|
+
pass
|
|
209
|
+
|
|
210
|
+
@abstractmethod
|
|
211
|
+
def statement_execute_partitions(self, request: StatementExecutePartitionsRequest) -> StatementExecutePartitionsResponse:
|
|
212
|
+
pass
|
|
213
|
+
|
|
214
|
+
@abstractmethod
|
|
215
|
+
def statement_read_partition(self, request: StatementReadPartitionRequest) -> StatementReadPartitionResponse:
|
|
216
|
+
pass
|
|
217
|
+
|
|
218
|
+
@abstractmethod
|
|
219
|
+
def statement_execute_async(self, request: StatementExecuteAsyncRequest) -> StatementExecuteAsyncResponse:
|
|
220
|
+
pass
|
|
221
|
+
|
|
222
|
+
@abstractmethod
|
|
223
|
+
def result_set_get_stream(self, request: ResultSetGetStreamRequest) -> ResultSetGetStreamResponse:
|
|
224
|
+
pass
|
|
225
|
+
|
|
226
|
+
@abstractmethod
|
|
227
|
+
def result_set_get_chunks(self, request: ResultSetGetChunksRequest) -> ResultSetGetChunksResponse:
|
|
228
|
+
pass
|
|
229
|
+
|
|
230
|
+
@abstractmethod
|
|
231
|
+
def result_set_release(self, request: ResultSetReleaseRequest) -> ResultSetReleaseResponse:
|
|
232
|
+
pass
|
|
233
|
+
|
|
234
|
+
@abstractmethod
|
|
235
|
+
def config_load_all_sections(self, request: ConfigLoadAllSectionsRequest) -> ConfigLoadAllSectionsResponse:
|
|
236
|
+
pass
|
|
237
|
+
|
|
238
|
+
@abstractmethod
|
|
239
|
+
def config_get_paths(self, request: ConfigGetPathsRequest) -> ConfigGetPathsResponse:
|
|
240
|
+
pass
|
|
241
|
+
|
|
242
|
+
@abstractmethod
|
|
243
|
+
def wif_create_attestation(self, request: WifCreateAttestationRequest) -> WifCreateAttestationResponse:
|
|
244
|
+
pass
|
|
245
|
+
|
|
246
|
+
@abstractmethod
|
|
247
|
+
def telemetry_send_api_usage(self, request: TelemetrySendApiUsageRequest) -> TelemetrySendResponse:
|
|
248
|
+
pass
|
|
249
|
+
|
|
250
|
+
@abstractmethod
|
|
251
|
+
def telemetry_send_wrapper_error(self, request: TelemetrySendWrapperErrorRequest) -> TelemetrySendResponse:
|
|
252
|
+
pass
|
|
253
|
+
|
|
254
|
+
@abstractmethod
|
|
255
|
+
def telemetry_send_log(self, request: TelemetrySendLogRequest) -> TelemetrySendResponse:
|
|
256
|
+
pass
|
|
257
|
+
|
|
258
|
+
|
|
259
|
+
|
|
260
|
+
class DatabaseDriverServer(DatabaseDriver):
|
|
261
|
+
def handle_message(self, method: str, message_bytes: bytes) -> bytes:
|
|
262
|
+
try:
|
|
263
|
+
# Dispatch to appropriate method
|
|
264
|
+
method_map = {
|
|
265
|
+
'driver_get_fips_status': (self.driver_get_fips_status, DriverGetFipsStatusRequest),
|
|
266
|
+
'database_new': (self.database_new, DatabaseNewRequest),
|
|
267
|
+
'database_set_options': (self.database_set_options, DatabaseSetOptionsRequest),
|
|
268
|
+
'database_init': (self.database_init, DatabaseInitRequest),
|
|
269
|
+
'database_release': (self.database_release, DatabaseReleaseRequest),
|
|
270
|
+
'database_fetch_chunk': (self.database_fetch_chunk, DatabaseFetchChunkRequest),
|
|
271
|
+
'connection_new': (self.connection_new, ConnectionNewRequest),
|
|
272
|
+
'connection_set_options': (self.connection_set_options, ConnectionSetOptionsRequest),
|
|
273
|
+
'connection_init': (self.connection_init, ConnectionInitRequest),
|
|
274
|
+
'connection_release': (self.connection_release, ConnectionReleaseRequest),
|
|
275
|
+
'connection_close': (self.connection_close, ConnectionCloseRequest),
|
|
276
|
+
'connection_is_closed': (self.connection_is_closed, ConnectionIsClosedRequest),
|
|
277
|
+
'connection_is_expired': (self.connection_is_expired, ConnectionIsExpiredRequest),
|
|
278
|
+
'connection_get_info': (self.connection_get_info, ConnectionGetInfoRequest),
|
|
279
|
+
'connection_get_objects': (self.connection_get_objects, ConnectionGetObjectsRequest),
|
|
280
|
+
'connection_get_table_schema': (self.connection_get_table_schema, ConnectionGetTableSchemaRequest),
|
|
281
|
+
'connection_get_table_types': (self.connection_get_table_types, ConnectionGetTableTypesRequest),
|
|
282
|
+
'connection_commit': (self.connection_commit, ConnectionCommitRequest),
|
|
283
|
+
'connection_rollback': (self.connection_rollback, ConnectionRollbackRequest),
|
|
284
|
+
'connection_set_autocommit': (self.connection_set_autocommit, ConnectionSetAutocommitRequest),
|
|
285
|
+
'connection_use_database': (self.connection_use_database, ConnectionUseDatabaseRequest),
|
|
286
|
+
'connection_use_schema': (self.connection_use_schema, ConnectionUseSchemaRequest),
|
|
287
|
+
'connection_set_session_parameters': (self.connection_set_session_parameters, ConnectionSetSessionParametersRequest),
|
|
288
|
+
'connection_get_parameter': (self.connection_get_parameter, ConnectionGetParameterRequest),
|
|
289
|
+
'connection_get_all_parameters': (self.connection_get_all_parameters, ConnectionGetAllParametersRequest),
|
|
290
|
+
'connection_get_server_version': (self.connection_get_server_version, ConnectionGetServerVersionRequest),
|
|
291
|
+
'connection_validate_options': (self.connection_validate_options, ConnectionValidateOptionsRequest),
|
|
292
|
+
'connection_get_query_result': (self.connection_get_query_result, ConnectionGetQueryResultRequest),
|
|
293
|
+
'connection_get_result_set': (self.connection_get_result_set, ConnectionGetResultSetRequest),
|
|
294
|
+
'connection_get_query_status': (self.connection_get_query_status, ConnectionGetQueryStatusRequest),
|
|
295
|
+
'connection_abort_query': (self.connection_abort_query, ConnectionAbortQueryRequest),
|
|
296
|
+
'connection_upload_stream_begin': (self.connection_upload_stream_begin, ConnectionUploadStreamBeginRequest),
|
|
297
|
+
'connection_upload_stream_chunk': (self.connection_upload_stream_chunk, ConnectionUploadStreamChunkRequest),
|
|
298
|
+
'connection_upload_stream_finish': (self.connection_upload_stream_finish, ConnectionUploadStreamFinishRequest),
|
|
299
|
+
'connection_upload_stream_abort': (self.connection_upload_stream_abort, ConnectionUploadStreamAbortRequest),
|
|
300
|
+
'connection_download_stream_begin': (self.connection_download_stream_begin, ConnectionDownloadStreamBeginRequest),
|
|
301
|
+
'connection_download_stream_chunk': (self.connection_download_stream_chunk, ConnectionDownloadStreamChunkRequest),
|
|
302
|
+
'connection_download_stream_close': (self.connection_download_stream_close, ConnectionDownloadStreamCloseRequest),
|
|
303
|
+
'connection_send_http': (self.connection_send_http, ConnectionSendHttpRequest),
|
|
304
|
+
'connection_request_token': (self.connection_request_token, ConnectionTokenRequest),
|
|
305
|
+
'connection_heartbeat': (self.connection_heartbeat, ConnectionHeartbeatRequest),
|
|
306
|
+
'statement_new': (self.statement_new, StatementNewRequest),
|
|
307
|
+
'statement_release': (self.statement_release, StatementReleaseRequest),
|
|
308
|
+
'statement_set_sql_query': (self.statement_set_sql_query, StatementSetSqlQueryRequest),
|
|
309
|
+
'statement_set_substrait_plan': (self.statement_set_substrait_plan, StatementSetSubstraitPlanRequest),
|
|
310
|
+
'statement_prepare': (self.statement_prepare, StatementPrepareRequest),
|
|
311
|
+
'statement_set_options': (self.statement_set_options, StatementSetOptionsRequest),
|
|
312
|
+
'statement_get_parameter_schema': (self.statement_get_parameter_schema, StatementGetParameterSchemaRequest),
|
|
313
|
+
'statement_execute_query': (self.statement_execute_query, StatementExecuteQueryRequest),
|
|
314
|
+
'statement_execute_partitions': (self.statement_execute_partitions, StatementExecutePartitionsRequest),
|
|
315
|
+
'statement_read_partition': (self.statement_read_partition, StatementReadPartitionRequest),
|
|
316
|
+
'statement_execute_async': (self.statement_execute_async, StatementExecuteAsyncRequest),
|
|
317
|
+
'result_set_get_stream': (self.result_set_get_stream, ResultSetGetStreamRequest),
|
|
318
|
+
'result_set_get_chunks': (self.result_set_get_chunks, ResultSetGetChunksRequest),
|
|
319
|
+
'result_set_release': (self.result_set_release, ResultSetReleaseRequest),
|
|
320
|
+
'config_load_all_sections': (self.config_load_all_sections, ConfigLoadAllSectionsRequest),
|
|
321
|
+
'config_get_paths': (self.config_get_paths, ConfigGetPathsRequest),
|
|
322
|
+
'wif_create_attestation': (self.wif_create_attestation, WifCreateAttestationRequest),
|
|
323
|
+
'telemetry_send_api_usage': (self.telemetry_send_api_usage, TelemetrySendApiUsageRequest),
|
|
324
|
+
'telemetry_send_wrapper_error': (self.telemetry_send_wrapper_error, TelemetrySendWrapperErrorRequest),
|
|
325
|
+
'telemetry_send_log': (self.telemetry_send_log, TelemetrySendLogRequest)
|
|
326
|
+
}
|
|
327
|
+
|
|
328
|
+
if method not in method_map:
|
|
329
|
+
raise ProtoError('Transport', f'Unknown method: {method}')
|
|
330
|
+
|
|
331
|
+
handler, request_class = method_map[method]
|
|
332
|
+
request = request_class()
|
|
333
|
+
request.ParseFromString(message_bytes)
|
|
334
|
+
response = handler(request)
|
|
335
|
+
return response.SerializeToString()
|
|
336
|
+
|
|
337
|
+
except Exception as e:
|
|
338
|
+
raise ProtoError('Transport', str(e))
|
|
339
|
+
|
|
340
|
+
class DatabaseDriverClient:
|
|
341
|
+
def __init__(self, transport, error_handler=None):
|
|
342
|
+
"""
|
|
343
|
+
Args:
|
|
344
|
+
transport: Transport layer that handles message serialization.
|
|
345
|
+
error_handler: Optional callable that converts a proto-layer exception
|
|
346
|
+
into a public exception. Must **return** the converted exception
|
|
347
|
+
(not raise it); ``_raise_error`` takes care of raising.
|
|
348
|
+
"""
|
|
349
|
+
self._transport = transport
|
|
350
|
+
self._error_handler = error_handler
|
|
351
|
+
|
|
352
|
+
def _raise_error(self, exc):
|
|
353
|
+
if self._error_handler is not None:
|
|
354
|
+
raise self._error_handler(exc) from None
|
|
355
|
+
raise exc
|
|
356
|
+
|
|
357
|
+
def driver_get_fips_status(self, request: DriverGetFipsStatusRequest) -> DriverGetFipsStatusResponse:
|
|
358
|
+
(code, response_bytes) = self._transport.handle_message('DatabaseDriver', 'driver_get_fips_status', request.SerializeToString())
|
|
359
|
+
if code == 0:
|
|
360
|
+
response = DriverGetFipsStatusResponse()
|
|
361
|
+
response.ParseFromString(response_bytes)
|
|
362
|
+
return response
|
|
363
|
+
elif code == 1:
|
|
364
|
+
error = DriverException()
|
|
365
|
+
error.ParseFromString(response_bytes)
|
|
366
|
+
self._raise_error(ProtoApplicationException(error))
|
|
367
|
+
elif code == 2:
|
|
368
|
+
self._raise_error(ProtoTransportException(str(response_bytes)))
|
|
369
|
+
else:
|
|
370
|
+
self._raise_error(ProtoTransportException(f"Unknown error code: %s", code))
|
|
371
|
+
def database_new(self, request: DatabaseNewRequest) -> DatabaseNewResponse:
|
|
372
|
+
(code, response_bytes) = self._transport.handle_message('DatabaseDriver', 'database_new', request.SerializeToString())
|
|
373
|
+
if code == 0:
|
|
374
|
+
response = DatabaseNewResponse()
|
|
375
|
+
response.ParseFromString(response_bytes)
|
|
376
|
+
return response
|
|
377
|
+
elif code == 1:
|
|
378
|
+
error = DriverException()
|
|
379
|
+
error.ParseFromString(response_bytes)
|
|
380
|
+
self._raise_error(ProtoApplicationException(error))
|
|
381
|
+
elif code == 2:
|
|
382
|
+
self._raise_error(ProtoTransportException(str(response_bytes)))
|
|
383
|
+
else:
|
|
384
|
+
self._raise_error(ProtoTransportException(f"Unknown error code: %s", code))
|
|
385
|
+
def database_set_options(self, request: DatabaseSetOptionsRequest) -> DatabaseSetOptionsResponse:
|
|
386
|
+
(code, response_bytes) = self._transport.handle_message('DatabaseDriver', 'database_set_options', request.SerializeToString())
|
|
387
|
+
if code == 0:
|
|
388
|
+
response = DatabaseSetOptionsResponse()
|
|
389
|
+
response.ParseFromString(response_bytes)
|
|
390
|
+
return response
|
|
391
|
+
elif code == 1:
|
|
392
|
+
error = DriverException()
|
|
393
|
+
error.ParseFromString(response_bytes)
|
|
394
|
+
self._raise_error(ProtoApplicationException(error))
|
|
395
|
+
elif code == 2:
|
|
396
|
+
self._raise_error(ProtoTransportException(str(response_bytes)))
|
|
397
|
+
else:
|
|
398
|
+
self._raise_error(ProtoTransportException(f"Unknown error code: %s", code))
|
|
399
|
+
def database_init(self, request: DatabaseInitRequest) -> DatabaseInitResponse:
|
|
400
|
+
(code, response_bytes) = self._transport.handle_message('DatabaseDriver', 'database_init', request.SerializeToString())
|
|
401
|
+
if code == 0:
|
|
402
|
+
response = DatabaseInitResponse()
|
|
403
|
+
response.ParseFromString(response_bytes)
|
|
404
|
+
return response
|
|
405
|
+
elif code == 1:
|
|
406
|
+
error = DriverException()
|
|
407
|
+
error.ParseFromString(response_bytes)
|
|
408
|
+
self._raise_error(ProtoApplicationException(error))
|
|
409
|
+
elif code == 2:
|
|
410
|
+
self._raise_error(ProtoTransportException(str(response_bytes)))
|
|
411
|
+
else:
|
|
412
|
+
self._raise_error(ProtoTransportException(f"Unknown error code: %s", code))
|
|
413
|
+
def database_release(self, request: DatabaseReleaseRequest) -> DatabaseReleaseResponse:
|
|
414
|
+
(code, response_bytes) = self._transport.handle_message('DatabaseDriver', 'database_release', request.SerializeToString())
|
|
415
|
+
if code == 0:
|
|
416
|
+
response = DatabaseReleaseResponse()
|
|
417
|
+
response.ParseFromString(response_bytes)
|
|
418
|
+
return response
|
|
419
|
+
elif code == 1:
|
|
420
|
+
error = DriverException()
|
|
421
|
+
error.ParseFromString(response_bytes)
|
|
422
|
+
self._raise_error(ProtoApplicationException(error))
|
|
423
|
+
elif code == 2:
|
|
424
|
+
self._raise_error(ProtoTransportException(str(response_bytes)))
|
|
425
|
+
else:
|
|
426
|
+
self._raise_error(ProtoTransportException(f"Unknown error code: %s", code))
|
|
427
|
+
def database_fetch_chunk(self, request: DatabaseFetchChunkRequest) -> DatabaseFetchChunkResponse:
|
|
428
|
+
(code, response_bytes) = self._transport.handle_message('DatabaseDriver', 'database_fetch_chunk', request.SerializeToString())
|
|
429
|
+
if code == 0:
|
|
430
|
+
response = DatabaseFetchChunkResponse()
|
|
431
|
+
response.ParseFromString(response_bytes)
|
|
432
|
+
return response
|
|
433
|
+
elif code == 1:
|
|
434
|
+
error = DriverException()
|
|
435
|
+
error.ParseFromString(response_bytes)
|
|
436
|
+
self._raise_error(ProtoApplicationException(error))
|
|
437
|
+
elif code == 2:
|
|
438
|
+
self._raise_error(ProtoTransportException(str(response_bytes)))
|
|
439
|
+
else:
|
|
440
|
+
self._raise_error(ProtoTransportException(f"Unknown error code: %s", code))
|
|
441
|
+
def connection_new(self, request: ConnectionNewRequest) -> ConnectionNewResponse:
|
|
442
|
+
(code, response_bytes) = self._transport.handle_message('DatabaseDriver', 'connection_new', request.SerializeToString())
|
|
443
|
+
if code == 0:
|
|
444
|
+
response = ConnectionNewResponse()
|
|
445
|
+
response.ParseFromString(response_bytes)
|
|
446
|
+
return response
|
|
447
|
+
elif code == 1:
|
|
448
|
+
error = DriverException()
|
|
449
|
+
error.ParseFromString(response_bytes)
|
|
450
|
+
self._raise_error(ProtoApplicationException(error))
|
|
451
|
+
elif code == 2:
|
|
452
|
+
self._raise_error(ProtoTransportException(str(response_bytes)))
|
|
453
|
+
else:
|
|
454
|
+
self._raise_error(ProtoTransportException(f"Unknown error code: %s", code))
|
|
455
|
+
def connection_set_options(self, request: ConnectionSetOptionsRequest) -> ConnectionSetOptionsResponse:
|
|
456
|
+
(code, response_bytes) = self._transport.handle_message('DatabaseDriver', 'connection_set_options', request.SerializeToString())
|
|
457
|
+
if code == 0:
|
|
458
|
+
response = ConnectionSetOptionsResponse()
|
|
459
|
+
response.ParseFromString(response_bytes)
|
|
460
|
+
return response
|
|
461
|
+
elif code == 1:
|
|
462
|
+
error = DriverException()
|
|
463
|
+
error.ParseFromString(response_bytes)
|
|
464
|
+
self._raise_error(ProtoApplicationException(error))
|
|
465
|
+
elif code == 2:
|
|
466
|
+
self._raise_error(ProtoTransportException(str(response_bytes)))
|
|
467
|
+
else:
|
|
468
|
+
self._raise_error(ProtoTransportException(f"Unknown error code: %s", code))
|
|
469
|
+
def connection_init(self, request: ConnectionInitRequest) -> ConnectionInitResponse:
|
|
470
|
+
(code, response_bytes) = self._transport.handle_message('DatabaseDriver', 'connection_init', request.SerializeToString())
|
|
471
|
+
if code == 0:
|
|
472
|
+
response = ConnectionInitResponse()
|
|
473
|
+
response.ParseFromString(response_bytes)
|
|
474
|
+
return response
|
|
475
|
+
elif code == 1:
|
|
476
|
+
error = DriverException()
|
|
477
|
+
error.ParseFromString(response_bytes)
|
|
478
|
+
self._raise_error(ProtoApplicationException(error))
|
|
479
|
+
elif code == 2:
|
|
480
|
+
self._raise_error(ProtoTransportException(str(response_bytes)))
|
|
481
|
+
else:
|
|
482
|
+
self._raise_error(ProtoTransportException(f"Unknown error code: %s", code))
|
|
483
|
+
def connection_release(self, request: ConnectionReleaseRequest) -> ConnectionReleaseResponse:
|
|
484
|
+
(code, response_bytes) = self._transport.handle_message('DatabaseDriver', 'connection_release', request.SerializeToString())
|
|
485
|
+
if code == 0:
|
|
486
|
+
response = ConnectionReleaseResponse()
|
|
487
|
+
response.ParseFromString(response_bytes)
|
|
488
|
+
return response
|
|
489
|
+
elif code == 1:
|
|
490
|
+
error = DriverException()
|
|
491
|
+
error.ParseFromString(response_bytes)
|
|
492
|
+
self._raise_error(ProtoApplicationException(error))
|
|
493
|
+
elif code == 2:
|
|
494
|
+
self._raise_error(ProtoTransportException(str(response_bytes)))
|
|
495
|
+
else:
|
|
496
|
+
self._raise_error(ProtoTransportException(f"Unknown error code: %s", code))
|
|
497
|
+
def connection_close(self, request: ConnectionCloseRequest) -> ConnectionCloseResponse:
|
|
498
|
+
(code, response_bytes) = self._transport.handle_message('DatabaseDriver', 'connection_close', request.SerializeToString())
|
|
499
|
+
if code == 0:
|
|
500
|
+
response = ConnectionCloseResponse()
|
|
501
|
+
response.ParseFromString(response_bytes)
|
|
502
|
+
return response
|
|
503
|
+
elif code == 1:
|
|
504
|
+
error = DriverException()
|
|
505
|
+
error.ParseFromString(response_bytes)
|
|
506
|
+
self._raise_error(ProtoApplicationException(error))
|
|
507
|
+
elif code == 2:
|
|
508
|
+
self._raise_error(ProtoTransportException(str(response_bytes)))
|
|
509
|
+
else:
|
|
510
|
+
self._raise_error(ProtoTransportException(f"Unknown error code: %s", code))
|
|
511
|
+
def connection_is_closed(self, request: ConnectionIsClosedRequest) -> ConnectionIsClosedResponse:
|
|
512
|
+
(code, response_bytes) = self._transport.handle_message('DatabaseDriver', 'connection_is_closed', request.SerializeToString())
|
|
513
|
+
if code == 0:
|
|
514
|
+
response = ConnectionIsClosedResponse()
|
|
515
|
+
response.ParseFromString(response_bytes)
|
|
516
|
+
return response
|
|
517
|
+
elif code == 1:
|
|
518
|
+
error = DriverException()
|
|
519
|
+
error.ParseFromString(response_bytes)
|
|
520
|
+
self._raise_error(ProtoApplicationException(error))
|
|
521
|
+
elif code == 2:
|
|
522
|
+
self._raise_error(ProtoTransportException(str(response_bytes)))
|
|
523
|
+
else:
|
|
524
|
+
self._raise_error(ProtoTransportException(f"Unknown error code: %s", code))
|
|
525
|
+
def connection_is_expired(self, request: ConnectionIsExpiredRequest) -> ConnectionIsExpiredResponse:
|
|
526
|
+
(code, response_bytes) = self._transport.handle_message('DatabaseDriver', 'connection_is_expired', request.SerializeToString())
|
|
527
|
+
if code == 0:
|
|
528
|
+
response = ConnectionIsExpiredResponse()
|
|
529
|
+
response.ParseFromString(response_bytes)
|
|
530
|
+
return response
|
|
531
|
+
elif code == 1:
|
|
532
|
+
error = DriverException()
|
|
533
|
+
error.ParseFromString(response_bytes)
|
|
534
|
+
self._raise_error(ProtoApplicationException(error))
|
|
535
|
+
elif code == 2:
|
|
536
|
+
self._raise_error(ProtoTransportException(str(response_bytes)))
|
|
537
|
+
else:
|
|
538
|
+
self._raise_error(ProtoTransportException(f"Unknown error code: %s", code))
|
|
539
|
+
def connection_get_info(self, request: ConnectionGetInfoRequest) -> ConnectionGetInfoResponse:
|
|
540
|
+
(code, response_bytes) = self._transport.handle_message('DatabaseDriver', 'connection_get_info', request.SerializeToString())
|
|
541
|
+
if code == 0:
|
|
542
|
+
response = ConnectionGetInfoResponse()
|
|
543
|
+
response.ParseFromString(response_bytes)
|
|
544
|
+
return response
|
|
545
|
+
elif code == 1:
|
|
546
|
+
error = DriverException()
|
|
547
|
+
error.ParseFromString(response_bytes)
|
|
548
|
+
self._raise_error(ProtoApplicationException(error))
|
|
549
|
+
elif code == 2:
|
|
550
|
+
self._raise_error(ProtoTransportException(str(response_bytes)))
|
|
551
|
+
else:
|
|
552
|
+
self._raise_error(ProtoTransportException(f"Unknown error code: %s", code))
|
|
553
|
+
def connection_get_objects(self, request: ConnectionGetObjectsRequest) -> ConnectionGetObjectsResponse:
|
|
554
|
+
(code, response_bytes) = self._transport.handle_message('DatabaseDriver', 'connection_get_objects', request.SerializeToString())
|
|
555
|
+
if code == 0:
|
|
556
|
+
response = ConnectionGetObjectsResponse()
|
|
557
|
+
response.ParseFromString(response_bytes)
|
|
558
|
+
return response
|
|
559
|
+
elif code == 1:
|
|
560
|
+
error = DriverException()
|
|
561
|
+
error.ParseFromString(response_bytes)
|
|
562
|
+
self._raise_error(ProtoApplicationException(error))
|
|
563
|
+
elif code == 2:
|
|
564
|
+
self._raise_error(ProtoTransportException(str(response_bytes)))
|
|
565
|
+
else:
|
|
566
|
+
self._raise_error(ProtoTransportException(f"Unknown error code: %s", code))
|
|
567
|
+
def connection_get_table_schema(self, request: ConnectionGetTableSchemaRequest) -> ConnectionGetTableSchemaResponse:
|
|
568
|
+
(code, response_bytes) = self._transport.handle_message('DatabaseDriver', 'connection_get_table_schema', request.SerializeToString())
|
|
569
|
+
if code == 0:
|
|
570
|
+
response = ConnectionGetTableSchemaResponse()
|
|
571
|
+
response.ParseFromString(response_bytes)
|
|
572
|
+
return response
|
|
573
|
+
elif code == 1:
|
|
574
|
+
error = DriverException()
|
|
575
|
+
error.ParseFromString(response_bytes)
|
|
576
|
+
self._raise_error(ProtoApplicationException(error))
|
|
577
|
+
elif code == 2:
|
|
578
|
+
self._raise_error(ProtoTransportException(str(response_bytes)))
|
|
579
|
+
else:
|
|
580
|
+
self._raise_error(ProtoTransportException(f"Unknown error code: %s", code))
|
|
581
|
+
def connection_get_table_types(self, request: ConnectionGetTableTypesRequest) -> ConnectionGetTableTypesResponse:
|
|
582
|
+
(code, response_bytes) = self._transport.handle_message('DatabaseDriver', 'connection_get_table_types', request.SerializeToString())
|
|
583
|
+
if code == 0:
|
|
584
|
+
response = ConnectionGetTableTypesResponse()
|
|
585
|
+
response.ParseFromString(response_bytes)
|
|
586
|
+
return response
|
|
587
|
+
elif code == 1:
|
|
588
|
+
error = DriverException()
|
|
589
|
+
error.ParseFromString(response_bytes)
|
|
590
|
+
self._raise_error(ProtoApplicationException(error))
|
|
591
|
+
elif code == 2:
|
|
592
|
+
self._raise_error(ProtoTransportException(str(response_bytes)))
|
|
593
|
+
else:
|
|
594
|
+
self._raise_error(ProtoTransportException(f"Unknown error code: %s", code))
|
|
595
|
+
def connection_commit(self, request: ConnectionCommitRequest) -> ConnectionCommitResponse:
|
|
596
|
+
(code, response_bytes) = self._transport.handle_message('DatabaseDriver', 'connection_commit', request.SerializeToString())
|
|
597
|
+
if code == 0:
|
|
598
|
+
response = ConnectionCommitResponse()
|
|
599
|
+
response.ParseFromString(response_bytes)
|
|
600
|
+
return response
|
|
601
|
+
elif code == 1:
|
|
602
|
+
error = DriverException()
|
|
603
|
+
error.ParseFromString(response_bytes)
|
|
604
|
+
self._raise_error(ProtoApplicationException(error))
|
|
605
|
+
elif code == 2:
|
|
606
|
+
self._raise_error(ProtoTransportException(str(response_bytes)))
|
|
607
|
+
else:
|
|
608
|
+
self._raise_error(ProtoTransportException(f"Unknown error code: %s", code))
|
|
609
|
+
def connection_rollback(self, request: ConnectionRollbackRequest) -> ConnectionRollbackResponse:
|
|
610
|
+
(code, response_bytes) = self._transport.handle_message('DatabaseDriver', 'connection_rollback', request.SerializeToString())
|
|
611
|
+
if code == 0:
|
|
612
|
+
response = ConnectionRollbackResponse()
|
|
613
|
+
response.ParseFromString(response_bytes)
|
|
614
|
+
return response
|
|
615
|
+
elif code == 1:
|
|
616
|
+
error = DriverException()
|
|
617
|
+
error.ParseFromString(response_bytes)
|
|
618
|
+
self._raise_error(ProtoApplicationException(error))
|
|
619
|
+
elif code == 2:
|
|
620
|
+
self._raise_error(ProtoTransportException(str(response_bytes)))
|
|
621
|
+
else:
|
|
622
|
+
self._raise_error(ProtoTransportException(f"Unknown error code: %s", code))
|
|
623
|
+
def connection_set_autocommit(self, request: ConnectionSetAutocommitRequest) -> ConnectionSetAutocommitResponse:
|
|
624
|
+
(code, response_bytes) = self._transport.handle_message('DatabaseDriver', 'connection_set_autocommit', request.SerializeToString())
|
|
625
|
+
if code == 0:
|
|
626
|
+
response = ConnectionSetAutocommitResponse()
|
|
627
|
+
response.ParseFromString(response_bytes)
|
|
628
|
+
return response
|
|
629
|
+
elif code == 1:
|
|
630
|
+
error = DriverException()
|
|
631
|
+
error.ParseFromString(response_bytes)
|
|
632
|
+
self._raise_error(ProtoApplicationException(error))
|
|
633
|
+
elif code == 2:
|
|
634
|
+
self._raise_error(ProtoTransportException(str(response_bytes)))
|
|
635
|
+
else:
|
|
636
|
+
self._raise_error(ProtoTransportException(f"Unknown error code: %s", code))
|
|
637
|
+
def connection_use_database(self, request: ConnectionUseDatabaseRequest) -> ConnectionUseDatabaseResponse:
|
|
638
|
+
(code, response_bytes) = self._transport.handle_message('DatabaseDriver', 'connection_use_database', request.SerializeToString())
|
|
639
|
+
if code == 0:
|
|
640
|
+
response = ConnectionUseDatabaseResponse()
|
|
641
|
+
response.ParseFromString(response_bytes)
|
|
642
|
+
return response
|
|
643
|
+
elif code == 1:
|
|
644
|
+
error = DriverException()
|
|
645
|
+
error.ParseFromString(response_bytes)
|
|
646
|
+
self._raise_error(ProtoApplicationException(error))
|
|
647
|
+
elif code == 2:
|
|
648
|
+
self._raise_error(ProtoTransportException(str(response_bytes)))
|
|
649
|
+
else:
|
|
650
|
+
self._raise_error(ProtoTransportException(f"Unknown error code: %s", code))
|
|
651
|
+
def connection_use_schema(self, request: ConnectionUseSchemaRequest) -> ConnectionUseSchemaResponse:
|
|
652
|
+
(code, response_bytes) = self._transport.handle_message('DatabaseDriver', 'connection_use_schema', request.SerializeToString())
|
|
653
|
+
if code == 0:
|
|
654
|
+
response = ConnectionUseSchemaResponse()
|
|
655
|
+
response.ParseFromString(response_bytes)
|
|
656
|
+
return response
|
|
657
|
+
elif code == 1:
|
|
658
|
+
error = DriverException()
|
|
659
|
+
error.ParseFromString(response_bytes)
|
|
660
|
+
self._raise_error(ProtoApplicationException(error))
|
|
661
|
+
elif code == 2:
|
|
662
|
+
self._raise_error(ProtoTransportException(str(response_bytes)))
|
|
663
|
+
else:
|
|
664
|
+
self._raise_error(ProtoTransportException(f"Unknown error code: %s", code))
|
|
665
|
+
def connection_set_session_parameters(self, request: ConnectionSetSessionParametersRequest) -> ConnectionSetSessionParametersResponse:
|
|
666
|
+
(code, response_bytes) = self._transport.handle_message('DatabaseDriver', 'connection_set_session_parameters', request.SerializeToString())
|
|
667
|
+
if code == 0:
|
|
668
|
+
response = ConnectionSetSessionParametersResponse()
|
|
669
|
+
response.ParseFromString(response_bytes)
|
|
670
|
+
return response
|
|
671
|
+
elif code == 1:
|
|
672
|
+
error = DriverException()
|
|
673
|
+
error.ParseFromString(response_bytes)
|
|
674
|
+
self._raise_error(ProtoApplicationException(error))
|
|
675
|
+
elif code == 2:
|
|
676
|
+
self._raise_error(ProtoTransportException(str(response_bytes)))
|
|
677
|
+
else:
|
|
678
|
+
self._raise_error(ProtoTransportException(f"Unknown error code: %s", code))
|
|
679
|
+
def connection_get_parameter(self, request: ConnectionGetParameterRequest) -> ConnectionGetParameterResponse:
|
|
680
|
+
(code, response_bytes) = self._transport.handle_message('DatabaseDriver', 'connection_get_parameter', request.SerializeToString())
|
|
681
|
+
if code == 0:
|
|
682
|
+
response = ConnectionGetParameterResponse()
|
|
683
|
+
response.ParseFromString(response_bytes)
|
|
684
|
+
return response
|
|
685
|
+
elif code == 1:
|
|
686
|
+
error = DriverException()
|
|
687
|
+
error.ParseFromString(response_bytes)
|
|
688
|
+
self._raise_error(ProtoApplicationException(error))
|
|
689
|
+
elif code == 2:
|
|
690
|
+
self._raise_error(ProtoTransportException(str(response_bytes)))
|
|
691
|
+
else:
|
|
692
|
+
self._raise_error(ProtoTransportException(f"Unknown error code: %s", code))
|
|
693
|
+
def connection_get_all_parameters(self, request: ConnectionGetAllParametersRequest) -> ConnectionGetAllParametersResponse:
|
|
694
|
+
(code, response_bytes) = self._transport.handle_message('DatabaseDriver', 'connection_get_all_parameters', request.SerializeToString())
|
|
695
|
+
if code == 0:
|
|
696
|
+
response = ConnectionGetAllParametersResponse()
|
|
697
|
+
response.ParseFromString(response_bytes)
|
|
698
|
+
return response
|
|
699
|
+
elif code == 1:
|
|
700
|
+
error = DriverException()
|
|
701
|
+
error.ParseFromString(response_bytes)
|
|
702
|
+
self._raise_error(ProtoApplicationException(error))
|
|
703
|
+
elif code == 2:
|
|
704
|
+
self._raise_error(ProtoTransportException(str(response_bytes)))
|
|
705
|
+
else:
|
|
706
|
+
self._raise_error(ProtoTransportException(f"Unknown error code: %s", code))
|
|
707
|
+
def connection_get_server_version(self, request: ConnectionGetServerVersionRequest) -> ConnectionGetServerVersionResponse:
|
|
708
|
+
(code, response_bytes) = self._transport.handle_message('DatabaseDriver', 'connection_get_server_version', request.SerializeToString())
|
|
709
|
+
if code == 0:
|
|
710
|
+
response = ConnectionGetServerVersionResponse()
|
|
711
|
+
response.ParseFromString(response_bytes)
|
|
712
|
+
return response
|
|
713
|
+
elif code == 1:
|
|
714
|
+
error = DriverException()
|
|
715
|
+
error.ParseFromString(response_bytes)
|
|
716
|
+
self._raise_error(ProtoApplicationException(error))
|
|
717
|
+
elif code == 2:
|
|
718
|
+
self._raise_error(ProtoTransportException(str(response_bytes)))
|
|
719
|
+
else:
|
|
720
|
+
self._raise_error(ProtoTransportException(f"Unknown error code: %s", code))
|
|
721
|
+
def connection_validate_options(self, request: ConnectionValidateOptionsRequest) -> ConnectionValidateOptionsResponse:
|
|
722
|
+
(code, response_bytes) = self._transport.handle_message('DatabaseDriver', 'connection_validate_options', request.SerializeToString())
|
|
723
|
+
if code == 0:
|
|
724
|
+
response = ConnectionValidateOptionsResponse()
|
|
725
|
+
response.ParseFromString(response_bytes)
|
|
726
|
+
return response
|
|
727
|
+
elif code == 1:
|
|
728
|
+
error = DriverException()
|
|
729
|
+
error.ParseFromString(response_bytes)
|
|
730
|
+
self._raise_error(ProtoApplicationException(error))
|
|
731
|
+
elif code == 2:
|
|
732
|
+
self._raise_error(ProtoTransportException(str(response_bytes)))
|
|
733
|
+
else:
|
|
734
|
+
self._raise_error(ProtoTransportException(f"Unknown error code: %s", code))
|
|
735
|
+
def connection_get_query_result(self, request: ConnectionGetQueryResultRequest) -> ExecuteQueryResponse:
|
|
736
|
+
(code, response_bytes) = self._transport.handle_message('DatabaseDriver', 'connection_get_query_result', request.SerializeToString())
|
|
737
|
+
if code == 0:
|
|
738
|
+
response = ExecuteQueryResponse()
|
|
739
|
+
response.ParseFromString(response_bytes)
|
|
740
|
+
return response
|
|
741
|
+
elif code == 1:
|
|
742
|
+
error = DriverException()
|
|
743
|
+
error.ParseFromString(response_bytes)
|
|
744
|
+
self._raise_error(ProtoApplicationException(error))
|
|
745
|
+
elif code == 2:
|
|
746
|
+
self._raise_error(ProtoTransportException(str(response_bytes)))
|
|
747
|
+
else:
|
|
748
|
+
self._raise_error(ProtoTransportException(f"Unknown error code: %s", code))
|
|
749
|
+
def connection_get_result_set(self, request: ConnectionGetResultSetRequest) -> ResultSetResponse:
|
|
750
|
+
(code, response_bytes) = self._transport.handle_message('DatabaseDriver', 'connection_get_result_set', request.SerializeToString())
|
|
751
|
+
if code == 0:
|
|
752
|
+
response = ResultSetResponse()
|
|
753
|
+
response.ParseFromString(response_bytes)
|
|
754
|
+
return response
|
|
755
|
+
elif code == 1:
|
|
756
|
+
error = DriverException()
|
|
757
|
+
error.ParseFromString(response_bytes)
|
|
758
|
+
self._raise_error(ProtoApplicationException(error))
|
|
759
|
+
elif code == 2:
|
|
760
|
+
self._raise_error(ProtoTransportException(str(response_bytes)))
|
|
761
|
+
else:
|
|
762
|
+
self._raise_error(ProtoTransportException(f"Unknown error code: %s", code))
|
|
763
|
+
def connection_get_query_status(self, request: ConnectionGetQueryStatusRequest) -> ConnectionGetQueryStatusResponse:
|
|
764
|
+
(code, response_bytes) = self._transport.handle_message('DatabaseDriver', 'connection_get_query_status', request.SerializeToString())
|
|
765
|
+
if code == 0:
|
|
766
|
+
response = ConnectionGetQueryStatusResponse()
|
|
767
|
+
response.ParseFromString(response_bytes)
|
|
768
|
+
return response
|
|
769
|
+
elif code == 1:
|
|
770
|
+
error = DriverException()
|
|
771
|
+
error.ParseFromString(response_bytes)
|
|
772
|
+
self._raise_error(ProtoApplicationException(error))
|
|
773
|
+
elif code == 2:
|
|
774
|
+
self._raise_error(ProtoTransportException(str(response_bytes)))
|
|
775
|
+
else:
|
|
776
|
+
self._raise_error(ProtoTransportException(f"Unknown error code: %s", code))
|
|
777
|
+
def connection_abort_query(self, request: ConnectionAbortQueryRequest) -> ConnectionAbortQueryResponse:
|
|
778
|
+
(code, response_bytes) = self._transport.handle_message('DatabaseDriver', 'connection_abort_query', request.SerializeToString())
|
|
779
|
+
if code == 0:
|
|
780
|
+
response = ConnectionAbortQueryResponse()
|
|
781
|
+
response.ParseFromString(response_bytes)
|
|
782
|
+
return response
|
|
783
|
+
elif code == 1:
|
|
784
|
+
error = DriverException()
|
|
785
|
+
error.ParseFromString(response_bytes)
|
|
786
|
+
self._raise_error(ProtoApplicationException(error))
|
|
787
|
+
elif code == 2:
|
|
788
|
+
self._raise_error(ProtoTransportException(str(response_bytes)))
|
|
789
|
+
else:
|
|
790
|
+
self._raise_error(ProtoTransportException(f"Unknown error code: %s", code))
|
|
791
|
+
def connection_upload_stream_begin(self, request: ConnectionUploadStreamBeginRequest) -> ConnectionUploadStreamBeginResponse:
|
|
792
|
+
(code, response_bytes) = self._transport.handle_message('DatabaseDriver', 'connection_upload_stream_begin', request.SerializeToString())
|
|
793
|
+
if code == 0:
|
|
794
|
+
response = ConnectionUploadStreamBeginResponse()
|
|
795
|
+
response.ParseFromString(response_bytes)
|
|
796
|
+
return response
|
|
797
|
+
elif code == 1:
|
|
798
|
+
error = DriverException()
|
|
799
|
+
error.ParseFromString(response_bytes)
|
|
800
|
+
self._raise_error(ProtoApplicationException(error))
|
|
801
|
+
elif code == 2:
|
|
802
|
+
self._raise_error(ProtoTransportException(str(response_bytes)))
|
|
803
|
+
else:
|
|
804
|
+
self._raise_error(ProtoTransportException(f"Unknown error code: %s", code))
|
|
805
|
+
def connection_upload_stream_chunk(self, request: ConnectionUploadStreamChunkRequest) -> ConnectionUploadStreamChunkResponse:
|
|
806
|
+
(code, response_bytes) = self._transport.handle_message('DatabaseDriver', 'connection_upload_stream_chunk', request.SerializeToString())
|
|
807
|
+
if code == 0:
|
|
808
|
+
response = ConnectionUploadStreamChunkResponse()
|
|
809
|
+
response.ParseFromString(response_bytes)
|
|
810
|
+
return response
|
|
811
|
+
elif code == 1:
|
|
812
|
+
error = DriverException()
|
|
813
|
+
error.ParseFromString(response_bytes)
|
|
814
|
+
self._raise_error(ProtoApplicationException(error))
|
|
815
|
+
elif code == 2:
|
|
816
|
+
self._raise_error(ProtoTransportException(str(response_bytes)))
|
|
817
|
+
else:
|
|
818
|
+
self._raise_error(ProtoTransportException(f"Unknown error code: %s", code))
|
|
819
|
+
def connection_upload_stream_finish(self, request: ConnectionUploadStreamFinishRequest) -> ConnectionUploadStreamFinishResponse:
|
|
820
|
+
(code, response_bytes) = self._transport.handle_message('DatabaseDriver', 'connection_upload_stream_finish', request.SerializeToString())
|
|
821
|
+
if code == 0:
|
|
822
|
+
response = ConnectionUploadStreamFinishResponse()
|
|
823
|
+
response.ParseFromString(response_bytes)
|
|
824
|
+
return response
|
|
825
|
+
elif code == 1:
|
|
826
|
+
error = DriverException()
|
|
827
|
+
error.ParseFromString(response_bytes)
|
|
828
|
+
self._raise_error(ProtoApplicationException(error))
|
|
829
|
+
elif code == 2:
|
|
830
|
+
self._raise_error(ProtoTransportException(str(response_bytes)))
|
|
831
|
+
else:
|
|
832
|
+
self._raise_error(ProtoTransportException(f"Unknown error code: %s", code))
|
|
833
|
+
def connection_upload_stream_abort(self, request: ConnectionUploadStreamAbortRequest) -> ConnectionUploadStreamAbortResponse:
|
|
834
|
+
(code, response_bytes) = self._transport.handle_message('DatabaseDriver', 'connection_upload_stream_abort', request.SerializeToString())
|
|
835
|
+
if code == 0:
|
|
836
|
+
response = ConnectionUploadStreamAbortResponse()
|
|
837
|
+
response.ParseFromString(response_bytes)
|
|
838
|
+
return response
|
|
839
|
+
elif code == 1:
|
|
840
|
+
error = DriverException()
|
|
841
|
+
error.ParseFromString(response_bytes)
|
|
842
|
+
self._raise_error(ProtoApplicationException(error))
|
|
843
|
+
elif code == 2:
|
|
844
|
+
self._raise_error(ProtoTransportException(str(response_bytes)))
|
|
845
|
+
else:
|
|
846
|
+
self._raise_error(ProtoTransportException(f"Unknown error code: %s", code))
|
|
847
|
+
def connection_download_stream_begin(self, request: ConnectionDownloadStreamBeginRequest) -> ConnectionDownloadStreamBeginResponse:
|
|
848
|
+
(code, response_bytes) = self._transport.handle_message('DatabaseDriver', 'connection_download_stream_begin', request.SerializeToString())
|
|
849
|
+
if code == 0:
|
|
850
|
+
response = ConnectionDownloadStreamBeginResponse()
|
|
851
|
+
response.ParseFromString(response_bytes)
|
|
852
|
+
return response
|
|
853
|
+
elif code == 1:
|
|
854
|
+
error = DriverException()
|
|
855
|
+
error.ParseFromString(response_bytes)
|
|
856
|
+
self._raise_error(ProtoApplicationException(error))
|
|
857
|
+
elif code == 2:
|
|
858
|
+
self._raise_error(ProtoTransportException(str(response_bytes)))
|
|
859
|
+
else:
|
|
860
|
+
self._raise_error(ProtoTransportException(f"Unknown error code: %s", code))
|
|
861
|
+
def connection_download_stream_chunk(self, request: ConnectionDownloadStreamChunkRequest) -> ConnectionDownloadStreamChunkResponse:
|
|
862
|
+
(code, response_bytes) = self._transport.handle_message('DatabaseDriver', 'connection_download_stream_chunk', request.SerializeToString())
|
|
863
|
+
if code == 0:
|
|
864
|
+
response = ConnectionDownloadStreamChunkResponse()
|
|
865
|
+
response.ParseFromString(response_bytes)
|
|
866
|
+
return response
|
|
867
|
+
elif code == 1:
|
|
868
|
+
error = DriverException()
|
|
869
|
+
error.ParseFromString(response_bytes)
|
|
870
|
+
self._raise_error(ProtoApplicationException(error))
|
|
871
|
+
elif code == 2:
|
|
872
|
+
self._raise_error(ProtoTransportException(str(response_bytes)))
|
|
873
|
+
else:
|
|
874
|
+
self._raise_error(ProtoTransportException(f"Unknown error code: %s", code))
|
|
875
|
+
def connection_download_stream_close(self, request: ConnectionDownloadStreamCloseRequest) -> ConnectionDownloadStreamCloseResponse:
|
|
876
|
+
(code, response_bytes) = self._transport.handle_message('DatabaseDriver', 'connection_download_stream_close', request.SerializeToString())
|
|
877
|
+
if code == 0:
|
|
878
|
+
response = ConnectionDownloadStreamCloseResponse()
|
|
879
|
+
response.ParseFromString(response_bytes)
|
|
880
|
+
return response
|
|
881
|
+
elif code == 1:
|
|
882
|
+
error = DriverException()
|
|
883
|
+
error.ParseFromString(response_bytes)
|
|
884
|
+
self._raise_error(ProtoApplicationException(error))
|
|
885
|
+
elif code == 2:
|
|
886
|
+
self._raise_error(ProtoTransportException(str(response_bytes)))
|
|
887
|
+
else:
|
|
888
|
+
self._raise_error(ProtoTransportException(f"Unknown error code: %s", code))
|
|
889
|
+
def connection_send_http(self, request: ConnectionSendHttpRequest) -> ConnectionSendHttpResponse:
|
|
890
|
+
(code, response_bytes) = self._transport.handle_message('DatabaseDriver', 'connection_send_http', request.SerializeToString())
|
|
891
|
+
if code == 0:
|
|
892
|
+
response = ConnectionSendHttpResponse()
|
|
893
|
+
response.ParseFromString(response_bytes)
|
|
894
|
+
return response
|
|
895
|
+
elif code == 1:
|
|
896
|
+
error = DriverException()
|
|
897
|
+
error.ParseFromString(response_bytes)
|
|
898
|
+
self._raise_error(ProtoApplicationException(error))
|
|
899
|
+
elif code == 2:
|
|
900
|
+
self._raise_error(ProtoTransportException(str(response_bytes)))
|
|
901
|
+
else:
|
|
902
|
+
self._raise_error(ProtoTransportException(f"Unknown error code: %s", code))
|
|
903
|
+
def connection_request_token(self, request: ConnectionTokenRequest) -> ConnectionTokenResponse:
|
|
904
|
+
(code, response_bytes) = self._transport.handle_message('DatabaseDriver', 'connection_request_token', request.SerializeToString())
|
|
905
|
+
if code == 0:
|
|
906
|
+
response = ConnectionTokenResponse()
|
|
907
|
+
response.ParseFromString(response_bytes)
|
|
908
|
+
return response
|
|
909
|
+
elif code == 1:
|
|
910
|
+
error = DriverException()
|
|
911
|
+
error.ParseFromString(response_bytes)
|
|
912
|
+
self._raise_error(ProtoApplicationException(error))
|
|
913
|
+
elif code == 2:
|
|
914
|
+
self._raise_error(ProtoTransportException(str(response_bytes)))
|
|
915
|
+
else:
|
|
916
|
+
self._raise_error(ProtoTransportException(f"Unknown error code: %s", code))
|
|
917
|
+
def connection_heartbeat(self, request: ConnectionHeartbeatRequest) -> ConnectionHeartbeatResponse:
|
|
918
|
+
(code, response_bytes) = self._transport.handle_message('DatabaseDriver', 'connection_heartbeat', request.SerializeToString())
|
|
919
|
+
if code == 0:
|
|
920
|
+
response = ConnectionHeartbeatResponse()
|
|
921
|
+
response.ParseFromString(response_bytes)
|
|
922
|
+
return response
|
|
923
|
+
elif code == 1:
|
|
924
|
+
error = DriverException()
|
|
925
|
+
error.ParseFromString(response_bytes)
|
|
926
|
+
self._raise_error(ProtoApplicationException(error))
|
|
927
|
+
elif code == 2:
|
|
928
|
+
self._raise_error(ProtoTransportException(str(response_bytes)))
|
|
929
|
+
else:
|
|
930
|
+
self._raise_error(ProtoTransportException(f"Unknown error code: %s", code))
|
|
931
|
+
def statement_new(self, request: StatementNewRequest) -> StatementNewResponse:
|
|
932
|
+
(code, response_bytes) = self._transport.handle_message('DatabaseDriver', 'statement_new', request.SerializeToString())
|
|
933
|
+
if code == 0:
|
|
934
|
+
response = StatementNewResponse()
|
|
935
|
+
response.ParseFromString(response_bytes)
|
|
936
|
+
return response
|
|
937
|
+
elif code == 1:
|
|
938
|
+
error = DriverException()
|
|
939
|
+
error.ParseFromString(response_bytes)
|
|
940
|
+
self._raise_error(ProtoApplicationException(error))
|
|
941
|
+
elif code == 2:
|
|
942
|
+
self._raise_error(ProtoTransportException(str(response_bytes)))
|
|
943
|
+
else:
|
|
944
|
+
self._raise_error(ProtoTransportException(f"Unknown error code: %s", code))
|
|
945
|
+
def statement_release(self, request: StatementReleaseRequest) -> StatementReleaseResponse:
|
|
946
|
+
(code, response_bytes) = self._transport.handle_message('DatabaseDriver', 'statement_release', request.SerializeToString())
|
|
947
|
+
if code == 0:
|
|
948
|
+
response = StatementReleaseResponse()
|
|
949
|
+
response.ParseFromString(response_bytes)
|
|
950
|
+
return response
|
|
951
|
+
elif code == 1:
|
|
952
|
+
error = DriverException()
|
|
953
|
+
error.ParseFromString(response_bytes)
|
|
954
|
+
self._raise_error(ProtoApplicationException(error))
|
|
955
|
+
elif code == 2:
|
|
956
|
+
self._raise_error(ProtoTransportException(str(response_bytes)))
|
|
957
|
+
else:
|
|
958
|
+
self._raise_error(ProtoTransportException(f"Unknown error code: %s", code))
|
|
959
|
+
def statement_set_sql_query(self, request: StatementSetSqlQueryRequest) -> StatementSetSqlQueryResponse:
|
|
960
|
+
(code, response_bytes) = self._transport.handle_message('DatabaseDriver', 'statement_set_sql_query', request.SerializeToString())
|
|
961
|
+
if code == 0:
|
|
962
|
+
response = StatementSetSqlQueryResponse()
|
|
963
|
+
response.ParseFromString(response_bytes)
|
|
964
|
+
return response
|
|
965
|
+
elif code == 1:
|
|
966
|
+
error = DriverException()
|
|
967
|
+
error.ParseFromString(response_bytes)
|
|
968
|
+
self._raise_error(ProtoApplicationException(error))
|
|
969
|
+
elif code == 2:
|
|
970
|
+
self._raise_error(ProtoTransportException(str(response_bytes)))
|
|
971
|
+
else:
|
|
972
|
+
self._raise_error(ProtoTransportException(f"Unknown error code: %s", code))
|
|
973
|
+
def statement_set_substrait_plan(self, request: StatementSetSubstraitPlanRequest) -> StatementSetSubstraitPlanResponse:
|
|
974
|
+
(code, response_bytes) = self._transport.handle_message('DatabaseDriver', 'statement_set_substrait_plan', request.SerializeToString())
|
|
975
|
+
if code == 0:
|
|
976
|
+
response = StatementSetSubstraitPlanResponse()
|
|
977
|
+
response.ParseFromString(response_bytes)
|
|
978
|
+
return response
|
|
979
|
+
elif code == 1:
|
|
980
|
+
error = DriverException()
|
|
981
|
+
error.ParseFromString(response_bytes)
|
|
982
|
+
self._raise_error(ProtoApplicationException(error))
|
|
983
|
+
elif code == 2:
|
|
984
|
+
self._raise_error(ProtoTransportException(str(response_bytes)))
|
|
985
|
+
else:
|
|
986
|
+
self._raise_error(ProtoTransportException(f"Unknown error code: %s", code))
|
|
987
|
+
def statement_prepare(self, request: StatementPrepareRequest) -> StatementPrepareResponse:
|
|
988
|
+
(code, response_bytes) = self._transport.handle_message('DatabaseDriver', 'statement_prepare', request.SerializeToString())
|
|
989
|
+
if code == 0:
|
|
990
|
+
response = StatementPrepareResponse()
|
|
991
|
+
response.ParseFromString(response_bytes)
|
|
992
|
+
return response
|
|
993
|
+
elif code == 1:
|
|
994
|
+
error = DriverException()
|
|
995
|
+
error.ParseFromString(response_bytes)
|
|
996
|
+
self._raise_error(ProtoApplicationException(error))
|
|
997
|
+
elif code == 2:
|
|
998
|
+
self._raise_error(ProtoTransportException(str(response_bytes)))
|
|
999
|
+
else:
|
|
1000
|
+
self._raise_error(ProtoTransportException(f"Unknown error code: %s", code))
|
|
1001
|
+
def statement_set_options(self, request: StatementSetOptionsRequest) -> StatementSetOptionsResponse:
|
|
1002
|
+
(code, response_bytes) = self._transport.handle_message('DatabaseDriver', 'statement_set_options', request.SerializeToString())
|
|
1003
|
+
if code == 0:
|
|
1004
|
+
response = StatementSetOptionsResponse()
|
|
1005
|
+
response.ParseFromString(response_bytes)
|
|
1006
|
+
return response
|
|
1007
|
+
elif code == 1:
|
|
1008
|
+
error = DriverException()
|
|
1009
|
+
error.ParseFromString(response_bytes)
|
|
1010
|
+
self._raise_error(ProtoApplicationException(error))
|
|
1011
|
+
elif code == 2:
|
|
1012
|
+
self._raise_error(ProtoTransportException(str(response_bytes)))
|
|
1013
|
+
else:
|
|
1014
|
+
self._raise_error(ProtoTransportException(f"Unknown error code: %s", code))
|
|
1015
|
+
def statement_get_parameter_schema(self, request: StatementGetParameterSchemaRequest) -> StatementGetParameterSchemaResponse:
|
|
1016
|
+
(code, response_bytes) = self._transport.handle_message('DatabaseDriver', 'statement_get_parameter_schema', request.SerializeToString())
|
|
1017
|
+
if code == 0:
|
|
1018
|
+
response = StatementGetParameterSchemaResponse()
|
|
1019
|
+
response.ParseFromString(response_bytes)
|
|
1020
|
+
return response
|
|
1021
|
+
elif code == 1:
|
|
1022
|
+
error = DriverException()
|
|
1023
|
+
error.ParseFromString(response_bytes)
|
|
1024
|
+
self._raise_error(ProtoApplicationException(error))
|
|
1025
|
+
elif code == 2:
|
|
1026
|
+
self._raise_error(ProtoTransportException(str(response_bytes)))
|
|
1027
|
+
else:
|
|
1028
|
+
self._raise_error(ProtoTransportException(f"Unknown error code: %s", code))
|
|
1029
|
+
def statement_execute_query(self, request: StatementExecuteQueryRequest) -> ExecuteQueryResponse:
|
|
1030
|
+
(code, response_bytes) = self._transport.handle_message('DatabaseDriver', 'statement_execute_query', request.SerializeToString())
|
|
1031
|
+
if code == 0:
|
|
1032
|
+
response = ExecuteQueryResponse()
|
|
1033
|
+
response.ParseFromString(response_bytes)
|
|
1034
|
+
return response
|
|
1035
|
+
elif code == 1:
|
|
1036
|
+
error = DriverException()
|
|
1037
|
+
error.ParseFromString(response_bytes)
|
|
1038
|
+
self._raise_error(ProtoApplicationException(error))
|
|
1039
|
+
elif code == 2:
|
|
1040
|
+
self._raise_error(ProtoTransportException(str(response_bytes)))
|
|
1041
|
+
else:
|
|
1042
|
+
self._raise_error(ProtoTransportException(f"Unknown error code: %s", code))
|
|
1043
|
+
def statement_execute_partitions(self, request: StatementExecutePartitionsRequest) -> StatementExecutePartitionsResponse:
|
|
1044
|
+
(code, response_bytes) = self._transport.handle_message('DatabaseDriver', 'statement_execute_partitions', request.SerializeToString())
|
|
1045
|
+
if code == 0:
|
|
1046
|
+
response = StatementExecutePartitionsResponse()
|
|
1047
|
+
response.ParseFromString(response_bytes)
|
|
1048
|
+
return response
|
|
1049
|
+
elif code == 1:
|
|
1050
|
+
error = DriverException()
|
|
1051
|
+
error.ParseFromString(response_bytes)
|
|
1052
|
+
self._raise_error(ProtoApplicationException(error))
|
|
1053
|
+
elif code == 2:
|
|
1054
|
+
self._raise_error(ProtoTransportException(str(response_bytes)))
|
|
1055
|
+
else:
|
|
1056
|
+
self._raise_error(ProtoTransportException(f"Unknown error code: %s", code))
|
|
1057
|
+
def statement_read_partition(self, request: StatementReadPartitionRequest) -> StatementReadPartitionResponse:
|
|
1058
|
+
(code, response_bytes) = self._transport.handle_message('DatabaseDriver', 'statement_read_partition', request.SerializeToString())
|
|
1059
|
+
if code == 0:
|
|
1060
|
+
response = StatementReadPartitionResponse()
|
|
1061
|
+
response.ParseFromString(response_bytes)
|
|
1062
|
+
return response
|
|
1063
|
+
elif code == 1:
|
|
1064
|
+
error = DriverException()
|
|
1065
|
+
error.ParseFromString(response_bytes)
|
|
1066
|
+
self._raise_error(ProtoApplicationException(error))
|
|
1067
|
+
elif code == 2:
|
|
1068
|
+
self._raise_error(ProtoTransportException(str(response_bytes)))
|
|
1069
|
+
else:
|
|
1070
|
+
self._raise_error(ProtoTransportException(f"Unknown error code: %s", code))
|
|
1071
|
+
def statement_execute_async(self, request: StatementExecuteAsyncRequest) -> StatementExecuteAsyncResponse:
|
|
1072
|
+
(code, response_bytes) = self._transport.handle_message('DatabaseDriver', 'statement_execute_async', request.SerializeToString())
|
|
1073
|
+
if code == 0:
|
|
1074
|
+
response = StatementExecuteAsyncResponse()
|
|
1075
|
+
response.ParseFromString(response_bytes)
|
|
1076
|
+
return response
|
|
1077
|
+
elif code == 1:
|
|
1078
|
+
error = DriverException()
|
|
1079
|
+
error.ParseFromString(response_bytes)
|
|
1080
|
+
self._raise_error(ProtoApplicationException(error))
|
|
1081
|
+
elif code == 2:
|
|
1082
|
+
self._raise_error(ProtoTransportException(str(response_bytes)))
|
|
1083
|
+
else:
|
|
1084
|
+
self._raise_error(ProtoTransportException(f"Unknown error code: %s", code))
|
|
1085
|
+
def result_set_get_stream(self, request: ResultSetGetStreamRequest) -> ResultSetGetStreamResponse:
|
|
1086
|
+
(code, response_bytes) = self._transport.handle_message('DatabaseDriver', 'result_set_get_stream', request.SerializeToString())
|
|
1087
|
+
if code == 0:
|
|
1088
|
+
response = ResultSetGetStreamResponse()
|
|
1089
|
+
response.ParseFromString(response_bytes)
|
|
1090
|
+
return response
|
|
1091
|
+
elif code == 1:
|
|
1092
|
+
error = DriverException()
|
|
1093
|
+
error.ParseFromString(response_bytes)
|
|
1094
|
+
self._raise_error(ProtoApplicationException(error))
|
|
1095
|
+
elif code == 2:
|
|
1096
|
+
self._raise_error(ProtoTransportException(str(response_bytes)))
|
|
1097
|
+
else:
|
|
1098
|
+
self._raise_error(ProtoTransportException(f"Unknown error code: %s", code))
|
|
1099
|
+
def result_set_get_chunks(self, request: ResultSetGetChunksRequest) -> ResultSetGetChunksResponse:
|
|
1100
|
+
(code, response_bytes) = self._transport.handle_message('DatabaseDriver', 'result_set_get_chunks', request.SerializeToString())
|
|
1101
|
+
if code == 0:
|
|
1102
|
+
response = ResultSetGetChunksResponse()
|
|
1103
|
+
response.ParseFromString(response_bytes)
|
|
1104
|
+
return response
|
|
1105
|
+
elif code == 1:
|
|
1106
|
+
error = DriverException()
|
|
1107
|
+
error.ParseFromString(response_bytes)
|
|
1108
|
+
self._raise_error(ProtoApplicationException(error))
|
|
1109
|
+
elif code == 2:
|
|
1110
|
+
self._raise_error(ProtoTransportException(str(response_bytes)))
|
|
1111
|
+
else:
|
|
1112
|
+
self._raise_error(ProtoTransportException(f"Unknown error code: %s", code))
|
|
1113
|
+
def result_set_release(self, request: ResultSetReleaseRequest) -> ResultSetReleaseResponse:
|
|
1114
|
+
(code, response_bytes) = self._transport.handle_message('DatabaseDriver', 'result_set_release', request.SerializeToString())
|
|
1115
|
+
if code == 0:
|
|
1116
|
+
response = ResultSetReleaseResponse()
|
|
1117
|
+
response.ParseFromString(response_bytes)
|
|
1118
|
+
return response
|
|
1119
|
+
elif code == 1:
|
|
1120
|
+
error = DriverException()
|
|
1121
|
+
error.ParseFromString(response_bytes)
|
|
1122
|
+
self._raise_error(ProtoApplicationException(error))
|
|
1123
|
+
elif code == 2:
|
|
1124
|
+
self._raise_error(ProtoTransportException(str(response_bytes)))
|
|
1125
|
+
else:
|
|
1126
|
+
self._raise_error(ProtoTransportException(f"Unknown error code: %s", code))
|
|
1127
|
+
def config_load_all_sections(self, request: ConfigLoadAllSectionsRequest) -> ConfigLoadAllSectionsResponse:
|
|
1128
|
+
(code, response_bytes) = self._transport.handle_message('DatabaseDriver', 'config_load_all_sections', request.SerializeToString())
|
|
1129
|
+
if code == 0:
|
|
1130
|
+
response = ConfigLoadAllSectionsResponse()
|
|
1131
|
+
response.ParseFromString(response_bytes)
|
|
1132
|
+
return response
|
|
1133
|
+
elif code == 1:
|
|
1134
|
+
error = DriverException()
|
|
1135
|
+
error.ParseFromString(response_bytes)
|
|
1136
|
+
self._raise_error(ProtoApplicationException(error))
|
|
1137
|
+
elif code == 2:
|
|
1138
|
+
self._raise_error(ProtoTransportException(str(response_bytes)))
|
|
1139
|
+
else:
|
|
1140
|
+
self._raise_error(ProtoTransportException(f"Unknown error code: %s", code))
|
|
1141
|
+
def config_get_paths(self, request: ConfigGetPathsRequest) -> ConfigGetPathsResponse:
|
|
1142
|
+
(code, response_bytes) = self._transport.handle_message('DatabaseDriver', 'config_get_paths', request.SerializeToString())
|
|
1143
|
+
if code == 0:
|
|
1144
|
+
response = ConfigGetPathsResponse()
|
|
1145
|
+
response.ParseFromString(response_bytes)
|
|
1146
|
+
return response
|
|
1147
|
+
elif code == 1:
|
|
1148
|
+
error = DriverException()
|
|
1149
|
+
error.ParseFromString(response_bytes)
|
|
1150
|
+
self._raise_error(ProtoApplicationException(error))
|
|
1151
|
+
elif code == 2:
|
|
1152
|
+
self._raise_error(ProtoTransportException(str(response_bytes)))
|
|
1153
|
+
else:
|
|
1154
|
+
self._raise_error(ProtoTransportException(f"Unknown error code: %s", code))
|
|
1155
|
+
def wif_create_attestation(self, request: WifCreateAttestationRequest) -> WifCreateAttestationResponse:
|
|
1156
|
+
(code, response_bytes) = self._transport.handle_message('DatabaseDriver', 'wif_create_attestation', request.SerializeToString())
|
|
1157
|
+
if code == 0:
|
|
1158
|
+
response = WifCreateAttestationResponse()
|
|
1159
|
+
response.ParseFromString(response_bytes)
|
|
1160
|
+
return response
|
|
1161
|
+
elif code == 1:
|
|
1162
|
+
error = DriverException()
|
|
1163
|
+
error.ParseFromString(response_bytes)
|
|
1164
|
+
self._raise_error(ProtoApplicationException(error))
|
|
1165
|
+
elif code == 2:
|
|
1166
|
+
self._raise_error(ProtoTransportException(str(response_bytes)))
|
|
1167
|
+
else:
|
|
1168
|
+
self._raise_error(ProtoTransportException(f"Unknown error code: %s", code))
|
|
1169
|
+
def telemetry_send_api_usage(self, request: TelemetrySendApiUsageRequest) -> TelemetrySendResponse:
|
|
1170
|
+
(code, response_bytes) = self._transport.handle_message('DatabaseDriver', 'telemetry_send_api_usage', request.SerializeToString())
|
|
1171
|
+
if code == 0:
|
|
1172
|
+
response = TelemetrySendResponse()
|
|
1173
|
+
response.ParseFromString(response_bytes)
|
|
1174
|
+
return response
|
|
1175
|
+
elif code == 1:
|
|
1176
|
+
error = DriverException()
|
|
1177
|
+
error.ParseFromString(response_bytes)
|
|
1178
|
+
self._raise_error(ProtoApplicationException(error))
|
|
1179
|
+
elif code == 2:
|
|
1180
|
+
self._raise_error(ProtoTransportException(str(response_bytes)))
|
|
1181
|
+
else:
|
|
1182
|
+
self._raise_error(ProtoTransportException(f"Unknown error code: %s", code))
|
|
1183
|
+
def telemetry_send_wrapper_error(self, request: TelemetrySendWrapperErrorRequest) -> TelemetrySendResponse:
|
|
1184
|
+
(code, response_bytes) = self._transport.handle_message('DatabaseDriver', 'telemetry_send_wrapper_error', request.SerializeToString())
|
|
1185
|
+
if code == 0:
|
|
1186
|
+
response = TelemetrySendResponse()
|
|
1187
|
+
response.ParseFromString(response_bytes)
|
|
1188
|
+
return response
|
|
1189
|
+
elif code == 1:
|
|
1190
|
+
error = DriverException()
|
|
1191
|
+
error.ParseFromString(response_bytes)
|
|
1192
|
+
self._raise_error(ProtoApplicationException(error))
|
|
1193
|
+
elif code == 2:
|
|
1194
|
+
self._raise_error(ProtoTransportException(str(response_bytes)))
|
|
1195
|
+
else:
|
|
1196
|
+
self._raise_error(ProtoTransportException(f"Unknown error code: %s", code))
|
|
1197
|
+
def telemetry_send_log(self, request: TelemetrySendLogRequest) -> TelemetrySendResponse:
|
|
1198
|
+
(code, response_bytes) = self._transport.handle_message('DatabaseDriver', 'telemetry_send_log', request.SerializeToString())
|
|
1199
|
+
if code == 0:
|
|
1200
|
+
response = TelemetrySendResponse()
|
|
1201
|
+
response.ParseFromString(response_bytes)
|
|
1202
|
+
return response
|
|
1203
|
+
elif code == 1:
|
|
1204
|
+
error = DriverException()
|
|
1205
|
+
error.ParseFromString(response_bytes)
|
|
1206
|
+
self._raise_error(ProtoApplicationException(error))
|
|
1207
|
+
elif code == 2:
|
|
1208
|
+
self._raise_error(ProtoTransportException(str(response_bytes)))
|
|
1209
|
+
else:
|
|
1210
|
+
self._raise_error(ProtoTransportException(f"Unknown error code: %s", code))
|
|
1211
|
+
class AsyncDatabaseDriverClient:
|
|
1212
|
+
"""Asynchronous DatabaseDriver client.
|
|
1213
|
+
|
|
1214
|
+
Each RPC awaits ``transport.handle_message_async``, which runs the
|
|
1215
|
+
blocking FFI call on a worker thread so the event loop is not blocked.
|
|
1216
|
+
This is a separate code path from the synchronous ``DatabaseDriverClient``.
|
|
1217
|
+
"""
|
|
1218
|
+
|
|
1219
|
+
def __init__(self, transport, error_handler=None):
|
|
1220
|
+
"""
|
|
1221
|
+
Args:
|
|
1222
|
+
transport: Transport layer that handles message serialization.
|
|
1223
|
+
error_handler: Optional callable that converts a proto-layer exception
|
|
1224
|
+
into a public exception. Must **return** the converted exception
|
|
1225
|
+
(not raise it); ``_raise_error`` takes care of raising.
|
|
1226
|
+
"""
|
|
1227
|
+
self._transport = transport
|
|
1228
|
+
self._error_handler = error_handler
|
|
1229
|
+
|
|
1230
|
+
def _raise_error(self, exc):
|
|
1231
|
+
if self._error_handler is not None:
|
|
1232
|
+
raise self._error_handler(exc) from None
|
|
1233
|
+
raise exc
|
|
1234
|
+
|
|
1235
|
+
async def driver_get_fips_status(self, request: DriverGetFipsStatusRequest) -> DriverGetFipsStatusResponse:
|
|
1236
|
+
(code, response_bytes) = await self._transport.handle_message_async('DatabaseDriver', 'driver_get_fips_status', request.SerializeToString())
|
|
1237
|
+
if code == 0:
|
|
1238
|
+
response = DriverGetFipsStatusResponse()
|
|
1239
|
+
response.ParseFromString(response_bytes)
|
|
1240
|
+
return response
|
|
1241
|
+
elif code == 1:
|
|
1242
|
+
error = DriverException()
|
|
1243
|
+
error.ParseFromString(response_bytes)
|
|
1244
|
+
self._raise_error(ProtoApplicationException(error))
|
|
1245
|
+
elif code == 2:
|
|
1246
|
+
self._raise_error(ProtoTransportException(str(response_bytes)))
|
|
1247
|
+
else:
|
|
1248
|
+
self._raise_error(ProtoTransportException(f"Unknown error code: %s", code))
|
|
1249
|
+
async def database_new(self, request: DatabaseNewRequest) -> DatabaseNewResponse:
|
|
1250
|
+
(code, response_bytes) = await self._transport.handle_message_async('DatabaseDriver', 'database_new', request.SerializeToString())
|
|
1251
|
+
if code == 0:
|
|
1252
|
+
response = DatabaseNewResponse()
|
|
1253
|
+
response.ParseFromString(response_bytes)
|
|
1254
|
+
return response
|
|
1255
|
+
elif code == 1:
|
|
1256
|
+
error = DriverException()
|
|
1257
|
+
error.ParseFromString(response_bytes)
|
|
1258
|
+
self._raise_error(ProtoApplicationException(error))
|
|
1259
|
+
elif code == 2:
|
|
1260
|
+
self._raise_error(ProtoTransportException(str(response_bytes)))
|
|
1261
|
+
else:
|
|
1262
|
+
self._raise_error(ProtoTransportException(f"Unknown error code: %s", code))
|
|
1263
|
+
async def database_set_options(self, request: DatabaseSetOptionsRequest) -> DatabaseSetOptionsResponse:
|
|
1264
|
+
(code, response_bytes) = await self._transport.handle_message_async('DatabaseDriver', 'database_set_options', request.SerializeToString())
|
|
1265
|
+
if code == 0:
|
|
1266
|
+
response = DatabaseSetOptionsResponse()
|
|
1267
|
+
response.ParseFromString(response_bytes)
|
|
1268
|
+
return response
|
|
1269
|
+
elif code == 1:
|
|
1270
|
+
error = DriverException()
|
|
1271
|
+
error.ParseFromString(response_bytes)
|
|
1272
|
+
self._raise_error(ProtoApplicationException(error))
|
|
1273
|
+
elif code == 2:
|
|
1274
|
+
self._raise_error(ProtoTransportException(str(response_bytes)))
|
|
1275
|
+
else:
|
|
1276
|
+
self._raise_error(ProtoTransportException(f"Unknown error code: %s", code))
|
|
1277
|
+
async def database_init(self, request: DatabaseInitRequest) -> DatabaseInitResponse:
|
|
1278
|
+
(code, response_bytes) = await self._transport.handle_message_async('DatabaseDriver', 'database_init', request.SerializeToString())
|
|
1279
|
+
if code == 0:
|
|
1280
|
+
response = DatabaseInitResponse()
|
|
1281
|
+
response.ParseFromString(response_bytes)
|
|
1282
|
+
return response
|
|
1283
|
+
elif code == 1:
|
|
1284
|
+
error = DriverException()
|
|
1285
|
+
error.ParseFromString(response_bytes)
|
|
1286
|
+
self._raise_error(ProtoApplicationException(error))
|
|
1287
|
+
elif code == 2:
|
|
1288
|
+
self._raise_error(ProtoTransportException(str(response_bytes)))
|
|
1289
|
+
else:
|
|
1290
|
+
self._raise_error(ProtoTransportException(f"Unknown error code: %s", code))
|
|
1291
|
+
async def database_release(self, request: DatabaseReleaseRequest) -> DatabaseReleaseResponse:
|
|
1292
|
+
(code, response_bytes) = await self._transport.handle_message_async('DatabaseDriver', 'database_release', request.SerializeToString())
|
|
1293
|
+
if code == 0:
|
|
1294
|
+
response = DatabaseReleaseResponse()
|
|
1295
|
+
response.ParseFromString(response_bytes)
|
|
1296
|
+
return response
|
|
1297
|
+
elif code == 1:
|
|
1298
|
+
error = DriverException()
|
|
1299
|
+
error.ParseFromString(response_bytes)
|
|
1300
|
+
self._raise_error(ProtoApplicationException(error))
|
|
1301
|
+
elif code == 2:
|
|
1302
|
+
self._raise_error(ProtoTransportException(str(response_bytes)))
|
|
1303
|
+
else:
|
|
1304
|
+
self._raise_error(ProtoTransportException(f"Unknown error code: %s", code))
|
|
1305
|
+
async def database_fetch_chunk(self, request: DatabaseFetchChunkRequest) -> DatabaseFetchChunkResponse:
|
|
1306
|
+
(code, response_bytes) = await self._transport.handle_message_async('DatabaseDriver', 'database_fetch_chunk', request.SerializeToString())
|
|
1307
|
+
if code == 0:
|
|
1308
|
+
response = DatabaseFetchChunkResponse()
|
|
1309
|
+
response.ParseFromString(response_bytes)
|
|
1310
|
+
return response
|
|
1311
|
+
elif code == 1:
|
|
1312
|
+
error = DriverException()
|
|
1313
|
+
error.ParseFromString(response_bytes)
|
|
1314
|
+
self._raise_error(ProtoApplicationException(error))
|
|
1315
|
+
elif code == 2:
|
|
1316
|
+
self._raise_error(ProtoTransportException(str(response_bytes)))
|
|
1317
|
+
else:
|
|
1318
|
+
self._raise_error(ProtoTransportException(f"Unknown error code: %s", code))
|
|
1319
|
+
async def connection_new(self, request: ConnectionNewRequest) -> ConnectionNewResponse:
|
|
1320
|
+
(code, response_bytes) = await self._transport.handle_message_async('DatabaseDriver', 'connection_new', request.SerializeToString())
|
|
1321
|
+
if code == 0:
|
|
1322
|
+
response = ConnectionNewResponse()
|
|
1323
|
+
response.ParseFromString(response_bytes)
|
|
1324
|
+
return response
|
|
1325
|
+
elif code == 1:
|
|
1326
|
+
error = DriverException()
|
|
1327
|
+
error.ParseFromString(response_bytes)
|
|
1328
|
+
self._raise_error(ProtoApplicationException(error))
|
|
1329
|
+
elif code == 2:
|
|
1330
|
+
self._raise_error(ProtoTransportException(str(response_bytes)))
|
|
1331
|
+
else:
|
|
1332
|
+
self._raise_error(ProtoTransportException(f"Unknown error code: %s", code))
|
|
1333
|
+
async def connection_set_options(self, request: ConnectionSetOptionsRequest) -> ConnectionSetOptionsResponse:
|
|
1334
|
+
(code, response_bytes) = await self._transport.handle_message_async('DatabaseDriver', 'connection_set_options', request.SerializeToString())
|
|
1335
|
+
if code == 0:
|
|
1336
|
+
response = ConnectionSetOptionsResponse()
|
|
1337
|
+
response.ParseFromString(response_bytes)
|
|
1338
|
+
return response
|
|
1339
|
+
elif code == 1:
|
|
1340
|
+
error = DriverException()
|
|
1341
|
+
error.ParseFromString(response_bytes)
|
|
1342
|
+
self._raise_error(ProtoApplicationException(error))
|
|
1343
|
+
elif code == 2:
|
|
1344
|
+
self._raise_error(ProtoTransportException(str(response_bytes)))
|
|
1345
|
+
else:
|
|
1346
|
+
self._raise_error(ProtoTransportException(f"Unknown error code: %s", code))
|
|
1347
|
+
async def connection_init(self, request: ConnectionInitRequest) -> ConnectionInitResponse:
|
|
1348
|
+
(code, response_bytes) = await self._transport.handle_message_async('DatabaseDriver', 'connection_init', request.SerializeToString())
|
|
1349
|
+
if code == 0:
|
|
1350
|
+
response = ConnectionInitResponse()
|
|
1351
|
+
response.ParseFromString(response_bytes)
|
|
1352
|
+
return response
|
|
1353
|
+
elif code == 1:
|
|
1354
|
+
error = DriverException()
|
|
1355
|
+
error.ParseFromString(response_bytes)
|
|
1356
|
+
self._raise_error(ProtoApplicationException(error))
|
|
1357
|
+
elif code == 2:
|
|
1358
|
+
self._raise_error(ProtoTransportException(str(response_bytes)))
|
|
1359
|
+
else:
|
|
1360
|
+
self._raise_error(ProtoTransportException(f"Unknown error code: %s", code))
|
|
1361
|
+
async def connection_release(self, request: ConnectionReleaseRequest) -> ConnectionReleaseResponse:
|
|
1362
|
+
(code, response_bytes) = await self._transport.handle_message_async('DatabaseDriver', 'connection_release', request.SerializeToString())
|
|
1363
|
+
if code == 0:
|
|
1364
|
+
response = ConnectionReleaseResponse()
|
|
1365
|
+
response.ParseFromString(response_bytes)
|
|
1366
|
+
return response
|
|
1367
|
+
elif code == 1:
|
|
1368
|
+
error = DriverException()
|
|
1369
|
+
error.ParseFromString(response_bytes)
|
|
1370
|
+
self._raise_error(ProtoApplicationException(error))
|
|
1371
|
+
elif code == 2:
|
|
1372
|
+
self._raise_error(ProtoTransportException(str(response_bytes)))
|
|
1373
|
+
else:
|
|
1374
|
+
self._raise_error(ProtoTransportException(f"Unknown error code: %s", code))
|
|
1375
|
+
async def connection_close(self, request: ConnectionCloseRequest) -> ConnectionCloseResponse:
|
|
1376
|
+
(code, response_bytes) = await self._transport.handle_message_async('DatabaseDriver', 'connection_close', request.SerializeToString())
|
|
1377
|
+
if code == 0:
|
|
1378
|
+
response = ConnectionCloseResponse()
|
|
1379
|
+
response.ParseFromString(response_bytes)
|
|
1380
|
+
return response
|
|
1381
|
+
elif code == 1:
|
|
1382
|
+
error = DriverException()
|
|
1383
|
+
error.ParseFromString(response_bytes)
|
|
1384
|
+
self._raise_error(ProtoApplicationException(error))
|
|
1385
|
+
elif code == 2:
|
|
1386
|
+
self._raise_error(ProtoTransportException(str(response_bytes)))
|
|
1387
|
+
else:
|
|
1388
|
+
self._raise_error(ProtoTransportException(f"Unknown error code: %s", code))
|
|
1389
|
+
async def connection_is_closed(self, request: ConnectionIsClosedRequest) -> ConnectionIsClosedResponse:
|
|
1390
|
+
(code, response_bytes) = await self._transport.handle_message_async('DatabaseDriver', 'connection_is_closed', request.SerializeToString())
|
|
1391
|
+
if code == 0:
|
|
1392
|
+
response = ConnectionIsClosedResponse()
|
|
1393
|
+
response.ParseFromString(response_bytes)
|
|
1394
|
+
return response
|
|
1395
|
+
elif code == 1:
|
|
1396
|
+
error = DriverException()
|
|
1397
|
+
error.ParseFromString(response_bytes)
|
|
1398
|
+
self._raise_error(ProtoApplicationException(error))
|
|
1399
|
+
elif code == 2:
|
|
1400
|
+
self._raise_error(ProtoTransportException(str(response_bytes)))
|
|
1401
|
+
else:
|
|
1402
|
+
self._raise_error(ProtoTransportException(f"Unknown error code: %s", code))
|
|
1403
|
+
async def connection_is_expired(self, request: ConnectionIsExpiredRequest) -> ConnectionIsExpiredResponse:
|
|
1404
|
+
(code, response_bytes) = await self._transport.handle_message_async('DatabaseDriver', 'connection_is_expired', request.SerializeToString())
|
|
1405
|
+
if code == 0:
|
|
1406
|
+
response = ConnectionIsExpiredResponse()
|
|
1407
|
+
response.ParseFromString(response_bytes)
|
|
1408
|
+
return response
|
|
1409
|
+
elif code == 1:
|
|
1410
|
+
error = DriverException()
|
|
1411
|
+
error.ParseFromString(response_bytes)
|
|
1412
|
+
self._raise_error(ProtoApplicationException(error))
|
|
1413
|
+
elif code == 2:
|
|
1414
|
+
self._raise_error(ProtoTransportException(str(response_bytes)))
|
|
1415
|
+
else:
|
|
1416
|
+
self._raise_error(ProtoTransportException(f"Unknown error code: %s", code))
|
|
1417
|
+
async def connection_get_info(self, request: ConnectionGetInfoRequest) -> ConnectionGetInfoResponse:
|
|
1418
|
+
(code, response_bytes) = await self._transport.handle_message_async('DatabaseDriver', 'connection_get_info', request.SerializeToString())
|
|
1419
|
+
if code == 0:
|
|
1420
|
+
response = ConnectionGetInfoResponse()
|
|
1421
|
+
response.ParseFromString(response_bytes)
|
|
1422
|
+
return response
|
|
1423
|
+
elif code == 1:
|
|
1424
|
+
error = DriverException()
|
|
1425
|
+
error.ParseFromString(response_bytes)
|
|
1426
|
+
self._raise_error(ProtoApplicationException(error))
|
|
1427
|
+
elif code == 2:
|
|
1428
|
+
self._raise_error(ProtoTransportException(str(response_bytes)))
|
|
1429
|
+
else:
|
|
1430
|
+
self._raise_error(ProtoTransportException(f"Unknown error code: %s", code))
|
|
1431
|
+
async def connection_get_objects(self, request: ConnectionGetObjectsRequest) -> ConnectionGetObjectsResponse:
|
|
1432
|
+
(code, response_bytes) = await self._transport.handle_message_async('DatabaseDriver', 'connection_get_objects', request.SerializeToString())
|
|
1433
|
+
if code == 0:
|
|
1434
|
+
response = ConnectionGetObjectsResponse()
|
|
1435
|
+
response.ParseFromString(response_bytes)
|
|
1436
|
+
return response
|
|
1437
|
+
elif code == 1:
|
|
1438
|
+
error = DriverException()
|
|
1439
|
+
error.ParseFromString(response_bytes)
|
|
1440
|
+
self._raise_error(ProtoApplicationException(error))
|
|
1441
|
+
elif code == 2:
|
|
1442
|
+
self._raise_error(ProtoTransportException(str(response_bytes)))
|
|
1443
|
+
else:
|
|
1444
|
+
self._raise_error(ProtoTransportException(f"Unknown error code: %s", code))
|
|
1445
|
+
async def connection_get_table_schema(self, request: ConnectionGetTableSchemaRequest) -> ConnectionGetTableSchemaResponse:
|
|
1446
|
+
(code, response_bytes) = await self._transport.handle_message_async('DatabaseDriver', 'connection_get_table_schema', request.SerializeToString())
|
|
1447
|
+
if code == 0:
|
|
1448
|
+
response = ConnectionGetTableSchemaResponse()
|
|
1449
|
+
response.ParseFromString(response_bytes)
|
|
1450
|
+
return response
|
|
1451
|
+
elif code == 1:
|
|
1452
|
+
error = DriverException()
|
|
1453
|
+
error.ParseFromString(response_bytes)
|
|
1454
|
+
self._raise_error(ProtoApplicationException(error))
|
|
1455
|
+
elif code == 2:
|
|
1456
|
+
self._raise_error(ProtoTransportException(str(response_bytes)))
|
|
1457
|
+
else:
|
|
1458
|
+
self._raise_error(ProtoTransportException(f"Unknown error code: %s", code))
|
|
1459
|
+
async def connection_get_table_types(self, request: ConnectionGetTableTypesRequest) -> ConnectionGetTableTypesResponse:
|
|
1460
|
+
(code, response_bytes) = await self._transport.handle_message_async('DatabaseDriver', 'connection_get_table_types', request.SerializeToString())
|
|
1461
|
+
if code == 0:
|
|
1462
|
+
response = ConnectionGetTableTypesResponse()
|
|
1463
|
+
response.ParseFromString(response_bytes)
|
|
1464
|
+
return response
|
|
1465
|
+
elif code == 1:
|
|
1466
|
+
error = DriverException()
|
|
1467
|
+
error.ParseFromString(response_bytes)
|
|
1468
|
+
self._raise_error(ProtoApplicationException(error))
|
|
1469
|
+
elif code == 2:
|
|
1470
|
+
self._raise_error(ProtoTransportException(str(response_bytes)))
|
|
1471
|
+
else:
|
|
1472
|
+
self._raise_error(ProtoTransportException(f"Unknown error code: %s", code))
|
|
1473
|
+
async def connection_commit(self, request: ConnectionCommitRequest) -> ConnectionCommitResponse:
|
|
1474
|
+
(code, response_bytes) = await self._transport.handle_message_async('DatabaseDriver', 'connection_commit', request.SerializeToString())
|
|
1475
|
+
if code == 0:
|
|
1476
|
+
response = ConnectionCommitResponse()
|
|
1477
|
+
response.ParseFromString(response_bytes)
|
|
1478
|
+
return response
|
|
1479
|
+
elif code == 1:
|
|
1480
|
+
error = DriverException()
|
|
1481
|
+
error.ParseFromString(response_bytes)
|
|
1482
|
+
self._raise_error(ProtoApplicationException(error))
|
|
1483
|
+
elif code == 2:
|
|
1484
|
+
self._raise_error(ProtoTransportException(str(response_bytes)))
|
|
1485
|
+
else:
|
|
1486
|
+
self._raise_error(ProtoTransportException(f"Unknown error code: %s", code))
|
|
1487
|
+
async def connection_rollback(self, request: ConnectionRollbackRequest) -> ConnectionRollbackResponse:
|
|
1488
|
+
(code, response_bytes) = await self._transport.handle_message_async('DatabaseDriver', 'connection_rollback', request.SerializeToString())
|
|
1489
|
+
if code == 0:
|
|
1490
|
+
response = ConnectionRollbackResponse()
|
|
1491
|
+
response.ParseFromString(response_bytes)
|
|
1492
|
+
return response
|
|
1493
|
+
elif code == 1:
|
|
1494
|
+
error = DriverException()
|
|
1495
|
+
error.ParseFromString(response_bytes)
|
|
1496
|
+
self._raise_error(ProtoApplicationException(error))
|
|
1497
|
+
elif code == 2:
|
|
1498
|
+
self._raise_error(ProtoTransportException(str(response_bytes)))
|
|
1499
|
+
else:
|
|
1500
|
+
self._raise_error(ProtoTransportException(f"Unknown error code: %s", code))
|
|
1501
|
+
async def connection_set_autocommit(self, request: ConnectionSetAutocommitRequest) -> ConnectionSetAutocommitResponse:
|
|
1502
|
+
(code, response_bytes) = await self._transport.handle_message_async('DatabaseDriver', 'connection_set_autocommit', request.SerializeToString())
|
|
1503
|
+
if code == 0:
|
|
1504
|
+
response = ConnectionSetAutocommitResponse()
|
|
1505
|
+
response.ParseFromString(response_bytes)
|
|
1506
|
+
return response
|
|
1507
|
+
elif code == 1:
|
|
1508
|
+
error = DriverException()
|
|
1509
|
+
error.ParseFromString(response_bytes)
|
|
1510
|
+
self._raise_error(ProtoApplicationException(error))
|
|
1511
|
+
elif code == 2:
|
|
1512
|
+
self._raise_error(ProtoTransportException(str(response_bytes)))
|
|
1513
|
+
else:
|
|
1514
|
+
self._raise_error(ProtoTransportException(f"Unknown error code: %s", code))
|
|
1515
|
+
async def connection_use_database(self, request: ConnectionUseDatabaseRequest) -> ConnectionUseDatabaseResponse:
|
|
1516
|
+
(code, response_bytes) = await self._transport.handle_message_async('DatabaseDriver', 'connection_use_database', request.SerializeToString())
|
|
1517
|
+
if code == 0:
|
|
1518
|
+
response = ConnectionUseDatabaseResponse()
|
|
1519
|
+
response.ParseFromString(response_bytes)
|
|
1520
|
+
return response
|
|
1521
|
+
elif code == 1:
|
|
1522
|
+
error = DriverException()
|
|
1523
|
+
error.ParseFromString(response_bytes)
|
|
1524
|
+
self._raise_error(ProtoApplicationException(error))
|
|
1525
|
+
elif code == 2:
|
|
1526
|
+
self._raise_error(ProtoTransportException(str(response_bytes)))
|
|
1527
|
+
else:
|
|
1528
|
+
self._raise_error(ProtoTransportException(f"Unknown error code: %s", code))
|
|
1529
|
+
async def connection_use_schema(self, request: ConnectionUseSchemaRequest) -> ConnectionUseSchemaResponse:
|
|
1530
|
+
(code, response_bytes) = await self._transport.handle_message_async('DatabaseDriver', 'connection_use_schema', request.SerializeToString())
|
|
1531
|
+
if code == 0:
|
|
1532
|
+
response = ConnectionUseSchemaResponse()
|
|
1533
|
+
response.ParseFromString(response_bytes)
|
|
1534
|
+
return response
|
|
1535
|
+
elif code == 1:
|
|
1536
|
+
error = DriverException()
|
|
1537
|
+
error.ParseFromString(response_bytes)
|
|
1538
|
+
self._raise_error(ProtoApplicationException(error))
|
|
1539
|
+
elif code == 2:
|
|
1540
|
+
self._raise_error(ProtoTransportException(str(response_bytes)))
|
|
1541
|
+
else:
|
|
1542
|
+
self._raise_error(ProtoTransportException(f"Unknown error code: %s", code))
|
|
1543
|
+
async def connection_set_session_parameters(self, request: ConnectionSetSessionParametersRequest) -> ConnectionSetSessionParametersResponse:
|
|
1544
|
+
(code, response_bytes) = await self._transport.handle_message_async('DatabaseDriver', 'connection_set_session_parameters', request.SerializeToString())
|
|
1545
|
+
if code == 0:
|
|
1546
|
+
response = ConnectionSetSessionParametersResponse()
|
|
1547
|
+
response.ParseFromString(response_bytes)
|
|
1548
|
+
return response
|
|
1549
|
+
elif code == 1:
|
|
1550
|
+
error = DriverException()
|
|
1551
|
+
error.ParseFromString(response_bytes)
|
|
1552
|
+
self._raise_error(ProtoApplicationException(error))
|
|
1553
|
+
elif code == 2:
|
|
1554
|
+
self._raise_error(ProtoTransportException(str(response_bytes)))
|
|
1555
|
+
else:
|
|
1556
|
+
self._raise_error(ProtoTransportException(f"Unknown error code: %s", code))
|
|
1557
|
+
async def connection_get_parameter(self, request: ConnectionGetParameterRequest) -> ConnectionGetParameterResponse:
|
|
1558
|
+
(code, response_bytes) = await self._transport.handle_message_async('DatabaseDriver', 'connection_get_parameter', request.SerializeToString())
|
|
1559
|
+
if code == 0:
|
|
1560
|
+
response = ConnectionGetParameterResponse()
|
|
1561
|
+
response.ParseFromString(response_bytes)
|
|
1562
|
+
return response
|
|
1563
|
+
elif code == 1:
|
|
1564
|
+
error = DriverException()
|
|
1565
|
+
error.ParseFromString(response_bytes)
|
|
1566
|
+
self._raise_error(ProtoApplicationException(error))
|
|
1567
|
+
elif code == 2:
|
|
1568
|
+
self._raise_error(ProtoTransportException(str(response_bytes)))
|
|
1569
|
+
else:
|
|
1570
|
+
self._raise_error(ProtoTransportException(f"Unknown error code: %s", code))
|
|
1571
|
+
async def connection_get_all_parameters(self, request: ConnectionGetAllParametersRequest) -> ConnectionGetAllParametersResponse:
|
|
1572
|
+
(code, response_bytes) = await self._transport.handle_message_async('DatabaseDriver', 'connection_get_all_parameters', request.SerializeToString())
|
|
1573
|
+
if code == 0:
|
|
1574
|
+
response = ConnectionGetAllParametersResponse()
|
|
1575
|
+
response.ParseFromString(response_bytes)
|
|
1576
|
+
return response
|
|
1577
|
+
elif code == 1:
|
|
1578
|
+
error = DriverException()
|
|
1579
|
+
error.ParseFromString(response_bytes)
|
|
1580
|
+
self._raise_error(ProtoApplicationException(error))
|
|
1581
|
+
elif code == 2:
|
|
1582
|
+
self._raise_error(ProtoTransportException(str(response_bytes)))
|
|
1583
|
+
else:
|
|
1584
|
+
self._raise_error(ProtoTransportException(f"Unknown error code: %s", code))
|
|
1585
|
+
async def connection_get_server_version(self, request: ConnectionGetServerVersionRequest) -> ConnectionGetServerVersionResponse:
|
|
1586
|
+
(code, response_bytes) = await self._transport.handle_message_async('DatabaseDriver', 'connection_get_server_version', request.SerializeToString())
|
|
1587
|
+
if code == 0:
|
|
1588
|
+
response = ConnectionGetServerVersionResponse()
|
|
1589
|
+
response.ParseFromString(response_bytes)
|
|
1590
|
+
return response
|
|
1591
|
+
elif code == 1:
|
|
1592
|
+
error = DriverException()
|
|
1593
|
+
error.ParseFromString(response_bytes)
|
|
1594
|
+
self._raise_error(ProtoApplicationException(error))
|
|
1595
|
+
elif code == 2:
|
|
1596
|
+
self._raise_error(ProtoTransportException(str(response_bytes)))
|
|
1597
|
+
else:
|
|
1598
|
+
self._raise_error(ProtoTransportException(f"Unknown error code: %s", code))
|
|
1599
|
+
async def connection_validate_options(self, request: ConnectionValidateOptionsRequest) -> ConnectionValidateOptionsResponse:
|
|
1600
|
+
(code, response_bytes) = await self._transport.handle_message_async('DatabaseDriver', 'connection_validate_options', request.SerializeToString())
|
|
1601
|
+
if code == 0:
|
|
1602
|
+
response = ConnectionValidateOptionsResponse()
|
|
1603
|
+
response.ParseFromString(response_bytes)
|
|
1604
|
+
return response
|
|
1605
|
+
elif code == 1:
|
|
1606
|
+
error = DriverException()
|
|
1607
|
+
error.ParseFromString(response_bytes)
|
|
1608
|
+
self._raise_error(ProtoApplicationException(error))
|
|
1609
|
+
elif code == 2:
|
|
1610
|
+
self._raise_error(ProtoTransportException(str(response_bytes)))
|
|
1611
|
+
else:
|
|
1612
|
+
self._raise_error(ProtoTransportException(f"Unknown error code: %s", code))
|
|
1613
|
+
async def connection_get_query_result(self, request: ConnectionGetQueryResultRequest) -> ExecuteQueryResponse:
|
|
1614
|
+
(code, response_bytes) = await self._transport.handle_message_async('DatabaseDriver', 'connection_get_query_result', request.SerializeToString())
|
|
1615
|
+
if code == 0:
|
|
1616
|
+
response = ExecuteQueryResponse()
|
|
1617
|
+
response.ParseFromString(response_bytes)
|
|
1618
|
+
return response
|
|
1619
|
+
elif code == 1:
|
|
1620
|
+
error = DriverException()
|
|
1621
|
+
error.ParseFromString(response_bytes)
|
|
1622
|
+
self._raise_error(ProtoApplicationException(error))
|
|
1623
|
+
elif code == 2:
|
|
1624
|
+
self._raise_error(ProtoTransportException(str(response_bytes)))
|
|
1625
|
+
else:
|
|
1626
|
+
self._raise_error(ProtoTransportException(f"Unknown error code: %s", code))
|
|
1627
|
+
async def connection_get_result_set(self, request: ConnectionGetResultSetRequest) -> ResultSetResponse:
|
|
1628
|
+
(code, response_bytes) = await self._transport.handle_message_async('DatabaseDriver', 'connection_get_result_set', request.SerializeToString())
|
|
1629
|
+
if code == 0:
|
|
1630
|
+
response = ResultSetResponse()
|
|
1631
|
+
response.ParseFromString(response_bytes)
|
|
1632
|
+
return response
|
|
1633
|
+
elif code == 1:
|
|
1634
|
+
error = DriverException()
|
|
1635
|
+
error.ParseFromString(response_bytes)
|
|
1636
|
+
self._raise_error(ProtoApplicationException(error))
|
|
1637
|
+
elif code == 2:
|
|
1638
|
+
self._raise_error(ProtoTransportException(str(response_bytes)))
|
|
1639
|
+
else:
|
|
1640
|
+
self._raise_error(ProtoTransportException(f"Unknown error code: %s", code))
|
|
1641
|
+
async def connection_get_query_status(self, request: ConnectionGetQueryStatusRequest) -> ConnectionGetQueryStatusResponse:
|
|
1642
|
+
(code, response_bytes) = await self._transport.handle_message_async('DatabaseDriver', 'connection_get_query_status', request.SerializeToString())
|
|
1643
|
+
if code == 0:
|
|
1644
|
+
response = ConnectionGetQueryStatusResponse()
|
|
1645
|
+
response.ParseFromString(response_bytes)
|
|
1646
|
+
return response
|
|
1647
|
+
elif code == 1:
|
|
1648
|
+
error = DriverException()
|
|
1649
|
+
error.ParseFromString(response_bytes)
|
|
1650
|
+
self._raise_error(ProtoApplicationException(error))
|
|
1651
|
+
elif code == 2:
|
|
1652
|
+
self._raise_error(ProtoTransportException(str(response_bytes)))
|
|
1653
|
+
else:
|
|
1654
|
+
self._raise_error(ProtoTransportException(f"Unknown error code: %s", code))
|
|
1655
|
+
async def connection_abort_query(self, request: ConnectionAbortQueryRequest) -> ConnectionAbortQueryResponse:
|
|
1656
|
+
(code, response_bytes) = await self._transport.handle_message_async('DatabaseDriver', 'connection_abort_query', request.SerializeToString())
|
|
1657
|
+
if code == 0:
|
|
1658
|
+
response = ConnectionAbortQueryResponse()
|
|
1659
|
+
response.ParseFromString(response_bytes)
|
|
1660
|
+
return response
|
|
1661
|
+
elif code == 1:
|
|
1662
|
+
error = DriverException()
|
|
1663
|
+
error.ParseFromString(response_bytes)
|
|
1664
|
+
self._raise_error(ProtoApplicationException(error))
|
|
1665
|
+
elif code == 2:
|
|
1666
|
+
self._raise_error(ProtoTransportException(str(response_bytes)))
|
|
1667
|
+
else:
|
|
1668
|
+
self._raise_error(ProtoTransportException(f"Unknown error code: %s", code))
|
|
1669
|
+
async def connection_upload_stream_begin(self, request: ConnectionUploadStreamBeginRequest) -> ConnectionUploadStreamBeginResponse:
|
|
1670
|
+
(code, response_bytes) = await self._transport.handle_message_async('DatabaseDriver', 'connection_upload_stream_begin', request.SerializeToString())
|
|
1671
|
+
if code == 0:
|
|
1672
|
+
response = ConnectionUploadStreamBeginResponse()
|
|
1673
|
+
response.ParseFromString(response_bytes)
|
|
1674
|
+
return response
|
|
1675
|
+
elif code == 1:
|
|
1676
|
+
error = DriverException()
|
|
1677
|
+
error.ParseFromString(response_bytes)
|
|
1678
|
+
self._raise_error(ProtoApplicationException(error))
|
|
1679
|
+
elif code == 2:
|
|
1680
|
+
self._raise_error(ProtoTransportException(str(response_bytes)))
|
|
1681
|
+
else:
|
|
1682
|
+
self._raise_error(ProtoTransportException(f"Unknown error code: %s", code))
|
|
1683
|
+
async def connection_upload_stream_chunk(self, request: ConnectionUploadStreamChunkRequest) -> ConnectionUploadStreamChunkResponse:
|
|
1684
|
+
(code, response_bytes) = await self._transport.handle_message_async('DatabaseDriver', 'connection_upload_stream_chunk', request.SerializeToString())
|
|
1685
|
+
if code == 0:
|
|
1686
|
+
response = ConnectionUploadStreamChunkResponse()
|
|
1687
|
+
response.ParseFromString(response_bytes)
|
|
1688
|
+
return response
|
|
1689
|
+
elif code == 1:
|
|
1690
|
+
error = DriverException()
|
|
1691
|
+
error.ParseFromString(response_bytes)
|
|
1692
|
+
self._raise_error(ProtoApplicationException(error))
|
|
1693
|
+
elif code == 2:
|
|
1694
|
+
self._raise_error(ProtoTransportException(str(response_bytes)))
|
|
1695
|
+
else:
|
|
1696
|
+
self._raise_error(ProtoTransportException(f"Unknown error code: %s", code))
|
|
1697
|
+
async def connection_upload_stream_finish(self, request: ConnectionUploadStreamFinishRequest) -> ConnectionUploadStreamFinishResponse:
|
|
1698
|
+
(code, response_bytes) = await self._transport.handle_message_async('DatabaseDriver', 'connection_upload_stream_finish', request.SerializeToString())
|
|
1699
|
+
if code == 0:
|
|
1700
|
+
response = ConnectionUploadStreamFinishResponse()
|
|
1701
|
+
response.ParseFromString(response_bytes)
|
|
1702
|
+
return response
|
|
1703
|
+
elif code == 1:
|
|
1704
|
+
error = DriverException()
|
|
1705
|
+
error.ParseFromString(response_bytes)
|
|
1706
|
+
self._raise_error(ProtoApplicationException(error))
|
|
1707
|
+
elif code == 2:
|
|
1708
|
+
self._raise_error(ProtoTransportException(str(response_bytes)))
|
|
1709
|
+
else:
|
|
1710
|
+
self._raise_error(ProtoTransportException(f"Unknown error code: %s", code))
|
|
1711
|
+
async def connection_upload_stream_abort(self, request: ConnectionUploadStreamAbortRequest) -> ConnectionUploadStreamAbortResponse:
|
|
1712
|
+
(code, response_bytes) = await self._transport.handle_message_async('DatabaseDriver', 'connection_upload_stream_abort', request.SerializeToString())
|
|
1713
|
+
if code == 0:
|
|
1714
|
+
response = ConnectionUploadStreamAbortResponse()
|
|
1715
|
+
response.ParseFromString(response_bytes)
|
|
1716
|
+
return response
|
|
1717
|
+
elif code == 1:
|
|
1718
|
+
error = DriverException()
|
|
1719
|
+
error.ParseFromString(response_bytes)
|
|
1720
|
+
self._raise_error(ProtoApplicationException(error))
|
|
1721
|
+
elif code == 2:
|
|
1722
|
+
self._raise_error(ProtoTransportException(str(response_bytes)))
|
|
1723
|
+
else:
|
|
1724
|
+
self._raise_error(ProtoTransportException(f"Unknown error code: %s", code))
|
|
1725
|
+
async def connection_download_stream_begin(self, request: ConnectionDownloadStreamBeginRequest) -> ConnectionDownloadStreamBeginResponse:
|
|
1726
|
+
(code, response_bytes) = await self._transport.handle_message_async('DatabaseDriver', 'connection_download_stream_begin', request.SerializeToString())
|
|
1727
|
+
if code == 0:
|
|
1728
|
+
response = ConnectionDownloadStreamBeginResponse()
|
|
1729
|
+
response.ParseFromString(response_bytes)
|
|
1730
|
+
return response
|
|
1731
|
+
elif code == 1:
|
|
1732
|
+
error = DriverException()
|
|
1733
|
+
error.ParseFromString(response_bytes)
|
|
1734
|
+
self._raise_error(ProtoApplicationException(error))
|
|
1735
|
+
elif code == 2:
|
|
1736
|
+
self._raise_error(ProtoTransportException(str(response_bytes)))
|
|
1737
|
+
else:
|
|
1738
|
+
self._raise_error(ProtoTransportException(f"Unknown error code: %s", code))
|
|
1739
|
+
async def connection_download_stream_chunk(self, request: ConnectionDownloadStreamChunkRequest) -> ConnectionDownloadStreamChunkResponse:
|
|
1740
|
+
(code, response_bytes) = await self._transport.handle_message_async('DatabaseDriver', 'connection_download_stream_chunk', request.SerializeToString())
|
|
1741
|
+
if code == 0:
|
|
1742
|
+
response = ConnectionDownloadStreamChunkResponse()
|
|
1743
|
+
response.ParseFromString(response_bytes)
|
|
1744
|
+
return response
|
|
1745
|
+
elif code == 1:
|
|
1746
|
+
error = DriverException()
|
|
1747
|
+
error.ParseFromString(response_bytes)
|
|
1748
|
+
self._raise_error(ProtoApplicationException(error))
|
|
1749
|
+
elif code == 2:
|
|
1750
|
+
self._raise_error(ProtoTransportException(str(response_bytes)))
|
|
1751
|
+
else:
|
|
1752
|
+
self._raise_error(ProtoTransportException(f"Unknown error code: %s", code))
|
|
1753
|
+
async def connection_download_stream_close(self, request: ConnectionDownloadStreamCloseRequest) -> ConnectionDownloadStreamCloseResponse:
|
|
1754
|
+
(code, response_bytes) = await self._transport.handle_message_async('DatabaseDriver', 'connection_download_stream_close', request.SerializeToString())
|
|
1755
|
+
if code == 0:
|
|
1756
|
+
response = ConnectionDownloadStreamCloseResponse()
|
|
1757
|
+
response.ParseFromString(response_bytes)
|
|
1758
|
+
return response
|
|
1759
|
+
elif code == 1:
|
|
1760
|
+
error = DriverException()
|
|
1761
|
+
error.ParseFromString(response_bytes)
|
|
1762
|
+
self._raise_error(ProtoApplicationException(error))
|
|
1763
|
+
elif code == 2:
|
|
1764
|
+
self._raise_error(ProtoTransportException(str(response_bytes)))
|
|
1765
|
+
else:
|
|
1766
|
+
self._raise_error(ProtoTransportException(f"Unknown error code: %s", code))
|
|
1767
|
+
async def connection_send_http(self, request: ConnectionSendHttpRequest) -> ConnectionSendHttpResponse:
|
|
1768
|
+
(code, response_bytes) = await self._transport.handle_message_async('DatabaseDriver', 'connection_send_http', request.SerializeToString())
|
|
1769
|
+
if code == 0:
|
|
1770
|
+
response = ConnectionSendHttpResponse()
|
|
1771
|
+
response.ParseFromString(response_bytes)
|
|
1772
|
+
return response
|
|
1773
|
+
elif code == 1:
|
|
1774
|
+
error = DriverException()
|
|
1775
|
+
error.ParseFromString(response_bytes)
|
|
1776
|
+
self._raise_error(ProtoApplicationException(error))
|
|
1777
|
+
elif code == 2:
|
|
1778
|
+
self._raise_error(ProtoTransportException(str(response_bytes)))
|
|
1779
|
+
else:
|
|
1780
|
+
self._raise_error(ProtoTransportException(f"Unknown error code: %s", code))
|
|
1781
|
+
async def connection_request_token(self, request: ConnectionTokenRequest) -> ConnectionTokenResponse:
|
|
1782
|
+
(code, response_bytes) = await self._transport.handle_message_async('DatabaseDriver', 'connection_request_token', request.SerializeToString())
|
|
1783
|
+
if code == 0:
|
|
1784
|
+
response = ConnectionTokenResponse()
|
|
1785
|
+
response.ParseFromString(response_bytes)
|
|
1786
|
+
return response
|
|
1787
|
+
elif code == 1:
|
|
1788
|
+
error = DriverException()
|
|
1789
|
+
error.ParseFromString(response_bytes)
|
|
1790
|
+
self._raise_error(ProtoApplicationException(error))
|
|
1791
|
+
elif code == 2:
|
|
1792
|
+
self._raise_error(ProtoTransportException(str(response_bytes)))
|
|
1793
|
+
else:
|
|
1794
|
+
self._raise_error(ProtoTransportException(f"Unknown error code: %s", code))
|
|
1795
|
+
async def connection_heartbeat(self, request: ConnectionHeartbeatRequest) -> ConnectionHeartbeatResponse:
|
|
1796
|
+
(code, response_bytes) = await self._transport.handle_message_async('DatabaseDriver', 'connection_heartbeat', request.SerializeToString())
|
|
1797
|
+
if code == 0:
|
|
1798
|
+
response = ConnectionHeartbeatResponse()
|
|
1799
|
+
response.ParseFromString(response_bytes)
|
|
1800
|
+
return response
|
|
1801
|
+
elif code == 1:
|
|
1802
|
+
error = DriverException()
|
|
1803
|
+
error.ParseFromString(response_bytes)
|
|
1804
|
+
self._raise_error(ProtoApplicationException(error))
|
|
1805
|
+
elif code == 2:
|
|
1806
|
+
self._raise_error(ProtoTransportException(str(response_bytes)))
|
|
1807
|
+
else:
|
|
1808
|
+
self._raise_error(ProtoTransportException(f"Unknown error code: %s", code))
|
|
1809
|
+
async def statement_new(self, request: StatementNewRequest) -> StatementNewResponse:
|
|
1810
|
+
(code, response_bytes) = await self._transport.handle_message_async('DatabaseDriver', 'statement_new', request.SerializeToString())
|
|
1811
|
+
if code == 0:
|
|
1812
|
+
response = StatementNewResponse()
|
|
1813
|
+
response.ParseFromString(response_bytes)
|
|
1814
|
+
return response
|
|
1815
|
+
elif code == 1:
|
|
1816
|
+
error = DriverException()
|
|
1817
|
+
error.ParseFromString(response_bytes)
|
|
1818
|
+
self._raise_error(ProtoApplicationException(error))
|
|
1819
|
+
elif code == 2:
|
|
1820
|
+
self._raise_error(ProtoTransportException(str(response_bytes)))
|
|
1821
|
+
else:
|
|
1822
|
+
self._raise_error(ProtoTransportException(f"Unknown error code: %s", code))
|
|
1823
|
+
async def statement_release(self, request: StatementReleaseRequest) -> StatementReleaseResponse:
|
|
1824
|
+
(code, response_bytes) = await self._transport.handle_message_async('DatabaseDriver', 'statement_release', request.SerializeToString())
|
|
1825
|
+
if code == 0:
|
|
1826
|
+
response = StatementReleaseResponse()
|
|
1827
|
+
response.ParseFromString(response_bytes)
|
|
1828
|
+
return response
|
|
1829
|
+
elif code == 1:
|
|
1830
|
+
error = DriverException()
|
|
1831
|
+
error.ParseFromString(response_bytes)
|
|
1832
|
+
self._raise_error(ProtoApplicationException(error))
|
|
1833
|
+
elif code == 2:
|
|
1834
|
+
self._raise_error(ProtoTransportException(str(response_bytes)))
|
|
1835
|
+
else:
|
|
1836
|
+
self._raise_error(ProtoTransportException(f"Unknown error code: %s", code))
|
|
1837
|
+
async def statement_set_sql_query(self, request: StatementSetSqlQueryRequest) -> StatementSetSqlQueryResponse:
|
|
1838
|
+
(code, response_bytes) = await self._transport.handle_message_async('DatabaseDriver', 'statement_set_sql_query', request.SerializeToString())
|
|
1839
|
+
if code == 0:
|
|
1840
|
+
response = StatementSetSqlQueryResponse()
|
|
1841
|
+
response.ParseFromString(response_bytes)
|
|
1842
|
+
return response
|
|
1843
|
+
elif code == 1:
|
|
1844
|
+
error = DriverException()
|
|
1845
|
+
error.ParseFromString(response_bytes)
|
|
1846
|
+
self._raise_error(ProtoApplicationException(error))
|
|
1847
|
+
elif code == 2:
|
|
1848
|
+
self._raise_error(ProtoTransportException(str(response_bytes)))
|
|
1849
|
+
else:
|
|
1850
|
+
self._raise_error(ProtoTransportException(f"Unknown error code: %s", code))
|
|
1851
|
+
async def statement_set_substrait_plan(self, request: StatementSetSubstraitPlanRequest) -> StatementSetSubstraitPlanResponse:
|
|
1852
|
+
(code, response_bytes) = await self._transport.handle_message_async('DatabaseDriver', 'statement_set_substrait_plan', request.SerializeToString())
|
|
1853
|
+
if code == 0:
|
|
1854
|
+
response = StatementSetSubstraitPlanResponse()
|
|
1855
|
+
response.ParseFromString(response_bytes)
|
|
1856
|
+
return response
|
|
1857
|
+
elif code == 1:
|
|
1858
|
+
error = DriverException()
|
|
1859
|
+
error.ParseFromString(response_bytes)
|
|
1860
|
+
self._raise_error(ProtoApplicationException(error))
|
|
1861
|
+
elif code == 2:
|
|
1862
|
+
self._raise_error(ProtoTransportException(str(response_bytes)))
|
|
1863
|
+
else:
|
|
1864
|
+
self._raise_error(ProtoTransportException(f"Unknown error code: %s", code))
|
|
1865
|
+
async def statement_prepare(self, request: StatementPrepareRequest) -> StatementPrepareResponse:
|
|
1866
|
+
(code, response_bytes) = await self._transport.handle_message_async('DatabaseDriver', 'statement_prepare', request.SerializeToString())
|
|
1867
|
+
if code == 0:
|
|
1868
|
+
response = StatementPrepareResponse()
|
|
1869
|
+
response.ParseFromString(response_bytes)
|
|
1870
|
+
return response
|
|
1871
|
+
elif code == 1:
|
|
1872
|
+
error = DriverException()
|
|
1873
|
+
error.ParseFromString(response_bytes)
|
|
1874
|
+
self._raise_error(ProtoApplicationException(error))
|
|
1875
|
+
elif code == 2:
|
|
1876
|
+
self._raise_error(ProtoTransportException(str(response_bytes)))
|
|
1877
|
+
else:
|
|
1878
|
+
self._raise_error(ProtoTransportException(f"Unknown error code: %s", code))
|
|
1879
|
+
async def statement_set_options(self, request: StatementSetOptionsRequest) -> StatementSetOptionsResponse:
|
|
1880
|
+
(code, response_bytes) = await self._transport.handle_message_async('DatabaseDriver', 'statement_set_options', request.SerializeToString())
|
|
1881
|
+
if code == 0:
|
|
1882
|
+
response = StatementSetOptionsResponse()
|
|
1883
|
+
response.ParseFromString(response_bytes)
|
|
1884
|
+
return response
|
|
1885
|
+
elif code == 1:
|
|
1886
|
+
error = DriverException()
|
|
1887
|
+
error.ParseFromString(response_bytes)
|
|
1888
|
+
self._raise_error(ProtoApplicationException(error))
|
|
1889
|
+
elif code == 2:
|
|
1890
|
+
self._raise_error(ProtoTransportException(str(response_bytes)))
|
|
1891
|
+
else:
|
|
1892
|
+
self._raise_error(ProtoTransportException(f"Unknown error code: %s", code))
|
|
1893
|
+
async def statement_get_parameter_schema(self, request: StatementGetParameterSchemaRequest) -> StatementGetParameterSchemaResponse:
|
|
1894
|
+
(code, response_bytes) = await self._transport.handle_message_async('DatabaseDriver', 'statement_get_parameter_schema', request.SerializeToString())
|
|
1895
|
+
if code == 0:
|
|
1896
|
+
response = StatementGetParameterSchemaResponse()
|
|
1897
|
+
response.ParseFromString(response_bytes)
|
|
1898
|
+
return response
|
|
1899
|
+
elif code == 1:
|
|
1900
|
+
error = DriverException()
|
|
1901
|
+
error.ParseFromString(response_bytes)
|
|
1902
|
+
self._raise_error(ProtoApplicationException(error))
|
|
1903
|
+
elif code == 2:
|
|
1904
|
+
self._raise_error(ProtoTransportException(str(response_bytes)))
|
|
1905
|
+
else:
|
|
1906
|
+
self._raise_error(ProtoTransportException(f"Unknown error code: %s", code))
|
|
1907
|
+
async def statement_execute_query(self, request: StatementExecuteQueryRequest) -> ExecuteQueryResponse:
|
|
1908
|
+
(code, response_bytes) = await self._transport.handle_message_async('DatabaseDriver', 'statement_execute_query', request.SerializeToString())
|
|
1909
|
+
if code == 0:
|
|
1910
|
+
response = ExecuteQueryResponse()
|
|
1911
|
+
response.ParseFromString(response_bytes)
|
|
1912
|
+
return response
|
|
1913
|
+
elif code == 1:
|
|
1914
|
+
error = DriverException()
|
|
1915
|
+
error.ParseFromString(response_bytes)
|
|
1916
|
+
self._raise_error(ProtoApplicationException(error))
|
|
1917
|
+
elif code == 2:
|
|
1918
|
+
self._raise_error(ProtoTransportException(str(response_bytes)))
|
|
1919
|
+
else:
|
|
1920
|
+
self._raise_error(ProtoTransportException(f"Unknown error code: %s", code))
|
|
1921
|
+
async def statement_execute_partitions(self, request: StatementExecutePartitionsRequest) -> StatementExecutePartitionsResponse:
|
|
1922
|
+
(code, response_bytes) = await self._transport.handle_message_async('DatabaseDriver', 'statement_execute_partitions', request.SerializeToString())
|
|
1923
|
+
if code == 0:
|
|
1924
|
+
response = StatementExecutePartitionsResponse()
|
|
1925
|
+
response.ParseFromString(response_bytes)
|
|
1926
|
+
return response
|
|
1927
|
+
elif code == 1:
|
|
1928
|
+
error = DriverException()
|
|
1929
|
+
error.ParseFromString(response_bytes)
|
|
1930
|
+
self._raise_error(ProtoApplicationException(error))
|
|
1931
|
+
elif code == 2:
|
|
1932
|
+
self._raise_error(ProtoTransportException(str(response_bytes)))
|
|
1933
|
+
else:
|
|
1934
|
+
self._raise_error(ProtoTransportException(f"Unknown error code: %s", code))
|
|
1935
|
+
async def statement_read_partition(self, request: StatementReadPartitionRequest) -> StatementReadPartitionResponse:
|
|
1936
|
+
(code, response_bytes) = await self._transport.handle_message_async('DatabaseDriver', 'statement_read_partition', request.SerializeToString())
|
|
1937
|
+
if code == 0:
|
|
1938
|
+
response = StatementReadPartitionResponse()
|
|
1939
|
+
response.ParseFromString(response_bytes)
|
|
1940
|
+
return response
|
|
1941
|
+
elif code == 1:
|
|
1942
|
+
error = DriverException()
|
|
1943
|
+
error.ParseFromString(response_bytes)
|
|
1944
|
+
self._raise_error(ProtoApplicationException(error))
|
|
1945
|
+
elif code == 2:
|
|
1946
|
+
self._raise_error(ProtoTransportException(str(response_bytes)))
|
|
1947
|
+
else:
|
|
1948
|
+
self._raise_error(ProtoTransportException(f"Unknown error code: %s", code))
|
|
1949
|
+
async def statement_execute_async(self, request: StatementExecuteAsyncRequest) -> StatementExecuteAsyncResponse:
|
|
1950
|
+
(code, response_bytes) = await self._transport.handle_message_async('DatabaseDriver', 'statement_execute_async', request.SerializeToString())
|
|
1951
|
+
if code == 0:
|
|
1952
|
+
response = StatementExecuteAsyncResponse()
|
|
1953
|
+
response.ParseFromString(response_bytes)
|
|
1954
|
+
return response
|
|
1955
|
+
elif code == 1:
|
|
1956
|
+
error = DriverException()
|
|
1957
|
+
error.ParseFromString(response_bytes)
|
|
1958
|
+
self._raise_error(ProtoApplicationException(error))
|
|
1959
|
+
elif code == 2:
|
|
1960
|
+
self._raise_error(ProtoTransportException(str(response_bytes)))
|
|
1961
|
+
else:
|
|
1962
|
+
self._raise_error(ProtoTransportException(f"Unknown error code: %s", code))
|
|
1963
|
+
async def result_set_get_stream(self, request: ResultSetGetStreamRequest) -> ResultSetGetStreamResponse:
|
|
1964
|
+
(code, response_bytes) = await self._transport.handle_message_async('DatabaseDriver', 'result_set_get_stream', request.SerializeToString())
|
|
1965
|
+
if code == 0:
|
|
1966
|
+
response = ResultSetGetStreamResponse()
|
|
1967
|
+
response.ParseFromString(response_bytes)
|
|
1968
|
+
return response
|
|
1969
|
+
elif code == 1:
|
|
1970
|
+
error = DriverException()
|
|
1971
|
+
error.ParseFromString(response_bytes)
|
|
1972
|
+
self._raise_error(ProtoApplicationException(error))
|
|
1973
|
+
elif code == 2:
|
|
1974
|
+
self._raise_error(ProtoTransportException(str(response_bytes)))
|
|
1975
|
+
else:
|
|
1976
|
+
self._raise_error(ProtoTransportException(f"Unknown error code: %s", code))
|
|
1977
|
+
async def result_set_get_chunks(self, request: ResultSetGetChunksRequest) -> ResultSetGetChunksResponse:
|
|
1978
|
+
(code, response_bytes) = await self._transport.handle_message_async('DatabaseDriver', 'result_set_get_chunks', request.SerializeToString())
|
|
1979
|
+
if code == 0:
|
|
1980
|
+
response = ResultSetGetChunksResponse()
|
|
1981
|
+
response.ParseFromString(response_bytes)
|
|
1982
|
+
return response
|
|
1983
|
+
elif code == 1:
|
|
1984
|
+
error = DriverException()
|
|
1985
|
+
error.ParseFromString(response_bytes)
|
|
1986
|
+
self._raise_error(ProtoApplicationException(error))
|
|
1987
|
+
elif code == 2:
|
|
1988
|
+
self._raise_error(ProtoTransportException(str(response_bytes)))
|
|
1989
|
+
else:
|
|
1990
|
+
self._raise_error(ProtoTransportException(f"Unknown error code: %s", code))
|
|
1991
|
+
async def result_set_release(self, request: ResultSetReleaseRequest) -> ResultSetReleaseResponse:
|
|
1992
|
+
(code, response_bytes) = await self._transport.handle_message_async('DatabaseDriver', 'result_set_release', request.SerializeToString())
|
|
1993
|
+
if code == 0:
|
|
1994
|
+
response = ResultSetReleaseResponse()
|
|
1995
|
+
response.ParseFromString(response_bytes)
|
|
1996
|
+
return response
|
|
1997
|
+
elif code == 1:
|
|
1998
|
+
error = DriverException()
|
|
1999
|
+
error.ParseFromString(response_bytes)
|
|
2000
|
+
self._raise_error(ProtoApplicationException(error))
|
|
2001
|
+
elif code == 2:
|
|
2002
|
+
self._raise_error(ProtoTransportException(str(response_bytes)))
|
|
2003
|
+
else:
|
|
2004
|
+
self._raise_error(ProtoTransportException(f"Unknown error code: %s", code))
|
|
2005
|
+
async def config_load_all_sections(self, request: ConfigLoadAllSectionsRequest) -> ConfigLoadAllSectionsResponse:
|
|
2006
|
+
(code, response_bytes) = await self._transport.handle_message_async('DatabaseDriver', 'config_load_all_sections', request.SerializeToString())
|
|
2007
|
+
if code == 0:
|
|
2008
|
+
response = ConfigLoadAllSectionsResponse()
|
|
2009
|
+
response.ParseFromString(response_bytes)
|
|
2010
|
+
return response
|
|
2011
|
+
elif code == 1:
|
|
2012
|
+
error = DriverException()
|
|
2013
|
+
error.ParseFromString(response_bytes)
|
|
2014
|
+
self._raise_error(ProtoApplicationException(error))
|
|
2015
|
+
elif code == 2:
|
|
2016
|
+
self._raise_error(ProtoTransportException(str(response_bytes)))
|
|
2017
|
+
else:
|
|
2018
|
+
self._raise_error(ProtoTransportException(f"Unknown error code: %s", code))
|
|
2019
|
+
async def config_get_paths(self, request: ConfigGetPathsRequest) -> ConfigGetPathsResponse:
|
|
2020
|
+
(code, response_bytes) = await self._transport.handle_message_async('DatabaseDriver', 'config_get_paths', request.SerializeToString())
|
|
2021
|
+
if code == 0:
|
|
2022
|
+
response = ConfigGetPathsResponse()
|
|
2023
|
+
response.ParseFromString(response_bytes)
|
|
2024
|
+
return response
|
|
2025
|
+
elif code == 1:
|
|
2026
|
+
error = DriverException()
|
|
2027
|
+
error.ParseFromString(response_bytes)
|
|
2028
|
+
self._raise_error(ProtoApplicationException(error))
|
|
2029
|
+
elif code == 2:
|
|
2030
|
+
self._raise_error(ProtoTransportException(str(response_bytes)))
|
|
2031
|
+
else:
|
|
2032
|
+
self._raise_error(ProtoTransportException(f"Unknown error code: %s", code))
|
|
2033
|
+
async def wif_create_attestation(self, request: WifCreateAttestationRequest) -> WifCreateAttestationResponse:
|
|
2034
|
+
(code, response_bytes) = await self._transport.handle_message_async('DatabaseDriver', 'wif_create_attestation', request.SerializeToString())
|
|
2035
|
+
if code == 0:
|
|
2036
|
+
response = WifCreateAttestationResponse()
|
|
2037
|
+
response.ParseFromString(response_bytes)
|
|
2038
|
+
return response
|
|
2039
|
+
elif code == 1:
|
|
2040
|
+
error = DriverException()
|
|
2041
|
+
error.ParseFromString(response_bytes)
|
|
2042
|
+
self._raise_error(ProtoApplicationException(error))
|
|
2043
|
+
elif code == 2:
|
|
2044
|
+
self._raise_error(ProtoTransportException(str(response_bytes)))
|
|
2045
|
+
else:
|
|
2046
|
+
self._raise_error(ProtoTransportException(f"Unknown error code: %s", code))
|
|
2047
|
+
async def telemetry_send_api_usage(self, request: TelemetrySendApiUsageRequest) -> TelemetrySendResponse:
|
|
2048
|
+
(code, response_bytes) = await self._transport.handle_message_async('DatabaseDriver', 'telemetry_send_api_usage', request.SerializeToString())
|
|
2049
|
+
if code == 0:
|
|
2050
|
+
response = TelemetrySendResponse()
|
|
2051
|
+
response.ParseFromString(response_bytes)
|
|
2052
|
+
return response
|
|
2053
|
+
elif code == 1:
|
|
2054
|
+
error = DriverException()
|
|
2055
|
+
error.ParseFromString(response_bytes)
|
|
2056
|
+
self._raise_error(ProtoApplicationException(error))
|
|
2057
|
+
elif code == 2:
|
|
2058
|
+
self._raise_error(ProtoTransportException(str(response_bytes)))
|
|
2059
|
+
else:
|
|
2060
|
+
self._raise_error(ProtoTransportException(f"Unknown error code: %s", code))
|
|
2061
|
+
async def telemetry_send_wrapper_error(self, request: TelemetrySendWrapperErrorRequest) -> TelemetrySendResponse:
|
|
2062
|
+
(code, response_bytes) = await self._transport.handle_message_async('DatabaseDriver', 'telemetry_send_wrapper_error', request.SerializeToString())
|
|
2063
|
+
if code == 0:
|
|
2064
|
+
response = TelemetrySendResponse()
|
|
2065
|
+
response.ParseFromString(response_bytes)
|
|
2066
|
+
return response
|
|
2067
|
+
elif code == 1:
|
|
2068
|
+
error = DriverException()
|
|
2069
|
+
error.ParseFromString(response_bytes)
|
|
2070
|
+
self._raise_error(ProtoApplicationException(error))
|
|
2071
|
+
elif code == 2:
|
|
2072
|
+
self._raise_error(ProtoTransportException(str(response_bytes)))
|
|
2073
|
+
else:
|
|
2074
|
+
self._raise_error(ProtoTransportException(f"Unknown error code: %s", code))
|
|
2075
|
+
async def telemetry_send_log(self, request: TelemetrySendLogRequest) -> TelemetrySendResponse:
|
|
2076
|
+
(code, response_bytes) = await self._transport.handle_message_async('DatabaseDriver', 'telemetry_send_log', request.SerializeToString())
|
|
2077
|
+
if code == 0:
|
|
2078
|
+
response = TelemetrySendResponse()
|
|
2079
|
+
response.ParseFromString(response_bytes)
|
|
2080
|
+
return response
|
|
2081
|
+
elif code == 1:
|
|
2082
|
+
error = DriverException()
|
|
2083
|
+
error.ParseFromString(response_bytes)
|
|
2084
|
+
self._raise_error(ProtoApplicationException(error))
|
|
2085
|
+
elif code == 2:
|
|
2086
|
+
self._raise_error(ProtoTransportException(str(response_bytes)))
|
|
2087
|
+
else:
|
|
2088
|
+
self._raise_error(ProtoTransportException(f"Unknown error code: %s", code))
|