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.
Files changed (204) hide show
  1. snowflake/connector/__init__.py +148 -0
  2. snowflake/connector/_common/__init__.py +0 -0
  3. snowflake/connector/_common/extras.py +134 -0
  4. snowflake/connector/_common/telemetry.py +221 -0
  5. snowflake/connector/_common/wif_util.py +92 -0
  6. snowflake/connector/_core/sf_core_python.cp315-win_amd64.pyd +0 -0
  7. snowflake/connector/_core/sf_core_python.pyi +79 -0
  8. snowflake/connector/_internal/__init__.py +6 -0
  9. snowflake/connector/_internal/_private_key_helper.py +31 -0
  10. snowflake/connector/_internal/api_client/__init__.py +93 -0
  11. snowflake/connector/_internal/api_client/bridge.py +41 -0
  12. snowflake/connector/_internal/api_client/client_api.py +976 -0
  13. snowflake/connector/_internal/arrow.py +47 -0
  14. snowflake/connector/_internal/arrow_context.py +240 -0
  15. snowflake/connector/_internal/arrow_stream_async.py +266 -0
  16. snowflake/connector/_internal/arrow_stream_iterator.cp311-win_amd64.pyd +0 -0
  17. snowflake/connector/_internal/arrow_stream_iterator.cp312-win_amd64.pyd +0 -0
  18. snowflake/connector/_internal/arrow_stream_iterator.cp313-win_amd64.pyd +0 -0
  19. snowflake/connector/_internal/arrow_stream_iterator.cp314-win_amd64.pyd +0 -0
  20. snowflake/connector/_internal/arrow_stream_iterator.cp314t-win_amd64.pyd +0 -0
  21. snowflake/connector/_internal/arrow_stream_iterator.cp315-win_amd64.pyd +0 -0
  22. snowflake/connector/_internal/arrow_stream_iterator.pyi +42 -0
  23. snowflake/connector/_internal/arrow_stream_utils.py +171 -0
  24. snowflake/connector/_internal/authenticator_type.py +53 -0
  25. snowflake/connector/_internal/backward_compatibility.py +245 -0
  26. snowflake/connector/_internal/binding_converters.py +741 -0
  27. snowflake/connector/_internal/config_utils.py +71 -0
  28. snowflake/connector/_internal/connection/__init__.py +39 -0
  29. snowflake/connector/_internal/connection/connection.py +616 -0
  30. snowflake/connector/_internal/connection/connection_types.py +13 -0
  31. snowflake/connector/_internal/connection/constants.py +25 -0
  32. snowflake/connector/_internal/connection/decorators.py +31 -0
  33. snowflake/connector/_internal/connection/freezable_proxy.py +104 -0
  34. snowflake/connector/_internal/connection/queries.py +10 -0
  35. snowflake/connector/_internal/connection_config_mixin.py +755 -0
  36. snowflake/connector/_internal/cursor/__init__.py +25 -0
  37. snowflake/connector/_internal/cursor/base.py +732 -0
  38. snowflake/connector/_internal/cursor/cursor_types.py +13 -0
  39. snowflake/connector/_internal/cursor/decorators.py +56 -0
  40. snowflake/connector/_internal/cursor/query_result.py +202 -0
  41. snowflake/connector/_internal/cursor/query_result_waiter.py +98 -0
  42. snowflake/connector/_internal/cursor/result_metadata.py +255 -0
  43. snowflake/connector/_internal/cursor/result_set_wrapper.py +66 -0
  44. snowflake/connector/_internal/decorators.py +539 -0
  45. snowflake/connector/_internal/error_kinds.py +111 -0
  46. snowflake/connector/_internal/errorcode.py +35 -0
  47. snowflake/connector/_internal/errorhandler.py +278 -0
  48. snowflake/connector/_internal/fips.py +28 -0
  49. snowflake/connector/_internal/logging/__init__.py +52 -0
  50. snowflake/connector/_internal/logging/config.py +134 -0
  51. snowflake/connector/_internal/logging/core_logger.py +218 -0
  52. snowflake/connector/_internal/logging/native_extension_logger.py +144 -0
  53. snowflake/connector/_internal/logout_config_mapping.py +187 -0
  54. snowflake/connector/_internal/nanoarrow_cpp/ArrowIterator/ArrayConverter.cpp +57 -0
  55. snowflake/connector/_internal/nanoarrow_cpp/ArrowIterator/ArrayConverter.hpp +29 -0
  56. snowflake/connector/_internal/nanoarrow_cpp/ArrowIterator/ArrowTableConverter.cpp +943 -0
  57. snowflake/connector/_internal/nanoarrow_cpp/ArrowIterator/ArrowTableConverter.hpp +63 -0
  58. snowflake/connector/_internal/nanoarrow_cpp/ArrowIterator/BinaryConverter.cpp +18 -0
  59. snowflake/connector/_internal/nanoarrow_cpp/ArrowIterator/BinaryConverter.hpp +26 -0
  60. snowflake/connector/_internal/nanoarrow_cpp/ArrowIterator/BooleanConverter.cpp +21 -0
  61. snowflake/connector/_internal/nanoarrow_cpp/ArrowIterator/BooleanConverter.hpp +23 -0
  62. snowflake/connector/_internal/nanoarrow_cpp/ArrowIterator/CArrowIterator.cpp +118 -0
  63. snowflake/connector/_internal/nanoarrow_cpp/ArrowIterator/CArrowIterator.hpp +116 -0
  64. snowflake/connector/_internal/nanoarrow_cpp/ArrowIterator/CArrowStreamIterator.cpp +323 -0
  65. snowflake/connector/_internal/nanoarrow_cpp/ArrowIterator/CArrowStreamIterator.hpp +133 -0
  66. snowflake/connector/_internal/nanoarrow_cpp/ArrowIterator/CArrowStreamTableIterator.cpp +275 -0
  67. snowflake/connector/_internal/nanoarrow_cpp/ArrowIterator/CArrowStreamTableIterator.hpp +168 -0
  68. snowflake/connector/_internal/nanoarrow_cpp/ArrowIterator/ConverterUtil.cpp +439 -0
  69. snowflake/connector/_internal/nanoarrow_cpp/ArrowIterator/ConverterUtil.hpp +30 -0
  70. snowflake/connector/_internal/nanoarrow_cpp/ArrowIterator/DateConverter.cpp +48 -0
  71. snowflake/connector/_internal/nanoarrow_cpp/ArrowIterator/DateConverter.hpp +46 -0
  72. snowflake/connector/_internal/nanoarrow_cpp/ArrowIterator/DecFloatConverter.cpp +76 -0
  73. snowflake/connector/_internal/nanoarrow_cpp/ArrowIterator/DecFloatConverter.hpp +35 -0
  74. snowflake/connector/_internal/nanoarrow_cpp/ArrowIterator/DecimalConverter.cpp +90 -0
  75. snowflake/connector/_internal/nanoarrow_cpp/ArrowIterator/DecimalConverter.hpp +63 -0
  76. snowflake/connector/_internal/nanoarrow_cpp/ArrowIterator/FixedSizeListConverter.cpp +62 -0
  77. snowflake/connector/_internal/nanoarrow_cpp/ArrowIterator/FixedSizeListConverter.hpp +28 -0
  78. snowflake/connector/_internal/nanoarrow_cpp/ArrowIterator/FloatConverter.cpp +29 -0
  79. snowflake/connector/_internal/nanoarrow_cpp/ArrowIterator/FloatConverter.hpp +35 -0
  80. snowflake/connector/_internal/nanoarrow_cpp/ArrowIterator/IColumnConverter.hpp +17 -0
  81. snowflake/connector/_internal/nanoarrow_cpp/ArrowIterator/IntConverter.cpp +23 -0
  82. snowflake/connector/_internal/nanoarrow_cpp/ArrowIterator/IntConverter.hpp +41 -0
  83. snowflake/connector/_internal/nanoarrow_cpp/ArrowIterator/IntervalConverter.cpp +65 -0
  84. snowflake/connector/_internal/nanoarrow_cpp/ArrowIterator/IntervalConverter.hpp +54 -0
  85. snowflake/connector/_internal/nanoarrow_cpp/ArrowIterator/LICENSE.txt +209 -0
  86. snowflake/connector/_internal/nanoarrow_cpp/ArrowIterator/MapConverter.cpp +79 -0
  87. snowflake/connector/_internal/nanoarrow_cpp/ArrowIterator/MapConverter.hpp +30 -0
  88. snowflake/connector/_internal/nanoarrow_cpp/ArrowIterator/ObjectConverter.cpp +52 -0
  89. snowflake/connector/_internal/nanoarrow_cpp/ArrowIterator/ObjectConverter.hpp +29 -0
  90. snowflake/connector/_internal/nanoarrow_cpp/ArrowIterator/Python/Common.cpp +8 -0
  91. snowflake/connector/_internal/nanoarrow_cpp/ArrowIterator/Python/Common.hpp +117 -0
  92. snowflake/connector/_internal/nanoarrow_cpp/ArrowIterator/Python/Helpers.cpp +54 -0
  93. snowflake/connector/_internal/nanoarrow_cpp/ArrowIterator/Python/Helpers.hpp +34 -0
  94. snowflake/connector/_internal/nanoarrow_cpp/ArrowIterator/SnowflakeType.cpp +33 -0
  95. snowflake/connector/_internal/nanoarrow_cpp/ArrowIterator/SnowflakeType.hpp +48 -0
  96. snowflake/connector/_internal/nanoarrow_cpp/ArrowIterator/StringConverter.cpp +18 -0
  97. snowflake/connector/_internal/nanoarrow_cpp/ArrowIterator/StringConverter.hpp +26 -0
  98. snowflake/connector/_internal/nanoarrow_cpp/ArrowIterator/TimeConverter.cpp +39 -0
  99. snowflake/connector/_internal/nanoarrow_cpp/ArrowIterator/TimeConverter.hpp +31 -0
  100. snowflake/connector/_internal/nanoarrow_cpp/ArrowIterator/TimeStampConverter.cpp +326 -0
  101. snowflake/connector/_internal/nanoarrow_cpp/ArrowIterator/TimeStampConverter.hpp +138 -0
  102. snowflake/connector/_internal/nanoarrow_cpp/ArrowIterator/Util/macros.hpp +14 -0
  103. snowflake/connector/_internal/nanoarrow_cpp/ArrowIterator/Util/time.cpp +65 -0
  104. snowflake/connector/_internal/nanoarrow_cpp/ArrowIterator/Util/time.hpp +68 -0
  105. snowflake/connector/_internal/nanoarrow_cpp/ArrowIterator/arrow_stream_iterator.cpp +13278 -0
  106. snowflake/connector/_internal/nanoarrow_cpp/ArrowIterator/arrow_stream_iterator.pyx +204 -0
  107. snowflake/connector/_internal/nanoarrow_cpp/ArrowIterator/flatcc/flatcc_accessors.h +109 -0
  108. snowflake/connector/_internal/nanoarrow_cpp/ArrowIterator/flatcc/flatcc_alloc.h +125 -0
  109. snowflake/connector/_internal/nanoarrow_cpp/ArrowIterator/flatcc/flatcc_assert.h +45 -0
  110. snowflake/connector/_internal/nanoarrow_cpp/ArrowIterator/flatcc/flatcc_builder.h +1914 -0
  111. snowflake/connector/_internal/nanoarrow_cpp/ArrowIterator/flatcc/flatcc_emitter.h +208 -0
  112. snowflake/connector/_internal/nanoarrow_cpp/ArrowIterator/flatcc/flatcc_endian.h +128 -0
  113. snowflake/connector/_internal/nanoarrow_cpp/ArrowIterator/flatcc/flatcc_epilogue.h +7 -0
  114. snowflake/connector/_internal/nanoarrow_cpp/ArrowIterator/flatcc/flatcc_flatbuffers.h +58 -0
  115. snowflake/connector/_internal/nanoarrow_cpp/ArrowIterator/flatcc/flatcc_identifier.h +144 -0
  116. snowflake/connector/_internal/nanoarrow_cpp/ArrowIterator/flatcc/flatcc_iov.h +31 -0
  117. snowflake/connector/_internal/nanoarrow_cpp/ArrowIterator/flatcc/flatcc_prologue.h +8 -0
  118. snowflake/connector/_internal/nanoarrow_cpp/ArrowIterator/flatcc/flatcc_refmap.h +144 -0
  119. snowflake/connector/_internal/nanoarrow_cpp/ArrowIterator/flatcc/flatcc_rtconfig.h +160 -0
  120. snowflake/connector/_internal/nanoarrow_cpp/ArrowIterator/flatcc/flatcc_types.h +98 -0
  121. snowflake/connector/_internal/nanoarrow_cpp/ArrowIterator/flatcc/flatcc_verifier.h +243 -0
  122. snowflake/connector/_internal/nanoarrow_cpp/ArrowIterator/flatcc/portable/flatcc_portable.h +14 -0
  123. snowflake/connector/_internal/nanoarrow_cpp/ArrowIterator/flatcc/portable/paligned_alloc.h +207 -0
  124. snowflake/connector/_internal/nanoarrow_cpp/ArrowIterator/flatcc/portable/pattributes.h +80 -0
  125. snowflake/connector/_internal/nanoarrow_cpp/ArrowIterator/flatcc/portable/pdiagnostic.h +83 -0
  126. snowflake/connector/_internal/nanoarrow_cpp/ArrowIterator/flatcc/portable/pdiagnostic_pop.h +20 -0
  127. snowflake/connector/_internal/nanoarrow_cpp/ArrowIterator/flatcc/portable/pdiagnostic_push.h +51 -0
  128. snowflake/connector/_internal/nanoarrow_cpp/ArrowIterator/flatcc/portable/pendian.h +201 -0
  129. snowflake/connector/_internal/nanoarrow_cpp/ArrowIterator/flatcc/portable/pendian_detect.h +108 -0
  130. snowflake/connector/_internal/nanoarrow_cpp/ArrowIterator/flatcc/portable/pinline.h +19 -0
  131. snowflake/connector/_internal/nanoarrow_cpp/ArrowIterator/flatcc/portable/pinttypes.h +52 -0
  132. snowflake/connector/_internal/nanoarrow_cpp/ArrowIterator/flatcc/portable/portable.h +2 -0
  133. snowflake/connector/_internal/nanoarrow_cpp/ArrowIterator/flatcc/portable/portable_basic.h +25 -0
  134. snowflake/connector/_internal/nanoarrow_cpp/ArrowIterator/flatcc/portable/pstatic_assert.h +67 -0
  135. snowflake/connector/_internal/nanoarrow_cpp/ArrowIterator/flatcc/portable/pstdalign.h +164 -0
  136. snowflake/connector/_internal/nanoarrow_cpp/ArrowIterator/flatcc/portable/pstdint.h +926 -0
  137. snowflake/connector/_internal/nanoarrow_cpp/ArrowIterator/flatcc/portable/punaligned.h +182 -0
  138. snowflake/connector/_internal/nanoarrow_cpp/ArrowIterator/flatcc/portable/pversion.h +6 -0
  139. snowflake/connector/_internal/nanoarrow_cpp/ArrowIterator/flatcc/portable/pwarnings.h +52 -0
  140. snowflake/connector/_internal/nanoarrow_cpp/ArrowIterator/flatcc.c +3052 -0
  141. snowflake/connector/_internal/nanoarrow_cpp/ArrowIterator/nanoarrow.c +3019 -0
  142. snowflake/connector/_internal/nanoarrow_cpp/ArrowIterator/nanoarrow.h +3354 -0
  143. snowflake/connector/_internal/nanoarrow_cpp/ArrowIterator/nanoarrow.hpp +344 -0
  144. snowflake/connector/_internal/nanoarrow_cpp/ArrowIterator/nanoarrow_ipc.c +29305 -0
  145. snowflake/connector/_internal/nanoarrow_cpp/ArrowIterator/nanoarrow_ipc.h +381 -0
  146. snowflake/connector/_internal/nanoarrow_cpp/Logging/logging.cpp +108 -0
  147. snowflake/connector/_internal/nanoarrow_cpp/Logging/logging.hpp +47 -0
  148. snowflake/connector/_internal/nanoarrow_cpp/scripts/.clang-format +8 -0
  149. snowflake/connector/_internal/nanoarrow_cpp/scripts/format.sh +8 -0
  150. snowflake/connector/_internal/protobuf_gen/__init__.py +0 -0
  151. snowflake/connector/_internal/protobuf_gen/database_driver_v1_pb2.py +408 -0
  152. snowflake/connector/_internal/protobuf_gen/database_driver_v1_pb2.pyi +4487 -0
  153. snowflake/connector/_internal/protobuf_gen/database_driver_v1_services.py +2088 -0
  154. snowflake/connector/_internal/protobuf_gen/proto_exception.py +17 -0
  155. snowflake/connector/_internal/result_batch.py +150 -0
  156. snowflake/connector/_internal/snowflake_restful.py +222 -0
  157. snowflake/connector/_internal/snowpark_backcompat_helpers.py +31 -0
  158. snowflake/connector/_internal/sqlstate.py +11 -0
  159. snowflake/connector/_internal/statement_utils.py +207 -0
  160. snowflake/connector/_internal/text_utils.py +112 -0
  161. snowflake/connector/_internal/type_codes.py +139 -0
  162. snowflake/connector/_internal/utils.py +65 -0
  163. snowflake/connector/_internal/write_pandas_operation.py +657 -0
  164. snowflake/connector/aio/README.md +168 -0
  165. snowflake/connector/aio/__init__.py +37 -0
  166. snowflake/connector/aio/connection/__init__.py +17 -0
  167. snowflake/connector/aio/connection/_connection.py +506 -0
  168. snowflake/connector/aio/cursor/__init__.py +35 -0
  169. snowflake/connector/aio/cursor/_base.py +1034 -0
  170. snowflake/connector/aio/cursor/_dict_cursor.py +78 -0
  171. snowflake/connector/aio/cursor/_result_set_wrapper.py +35 -0
  172. snowflake/connector/aio/cursor/_snowflake_cursor.py +72 -0
  173. snowflake/connector/aio/pandas_tools.py +279 -0
  174. snowflake/connector/aio/result_batch.py +165 -0
  175. snowflake/connector/auth/__init__.py +0 -0
  176. snowflake/connector/auth/workload_identity.py +5 -0
  177. snowflake/connector/compat.py +22 -0
  178. snowflake/connector/config_manager.py +392 -0
  179. snowflake/connector/connection/__init__.py +30 -0
  180. snowflake/connector/connection/_connection.py +641 -0
  181. snowflake/connector/connection_config.py +517 -0
  182. snowflake/connector/constants.py +52 -0
  183. snowflake/connector/cursor/__init__.py +37 -0
  184. snowflake/connector/cursor/_base.py +1169 -0
  185. snowflake/connector/cursor/_chunked_download_reader.py +75 -0
  186. snowflake/connector/cursor/_dict_cursor.py +78 -0
  187. snowflake/connector/cursor/_result_set_wrapper.py +35 -0
  188. snowflake/connector/cursor/_snowflake_cursor.py +72 -0
  189. snowflake/connector/description.py +7 -0
  190. snowflake/connector/errors.py +492 -0
  191. snowflake/connector/network.py +71 -0
  192. snowflake/connector/pandas_tools.py +191 -0
  193. snowflake/connector/py.typed +0 -0
  194. snowflake/connector/result_batch.py +216 -0
  195. snowflake/connector/telemetry.py +40 -0
  196. snowflake/connector/time_util.py +11 -0
  197. snowflake/connector/types.py +134 -0
  198. snowflake/connector/util_text.py +3 -0
  199. snowflake/connector/version.py +23 -0
  200. snowflake_connector_python-5.2.0.dist-info/DELVEWHEEL +2 -0
  201. snowflake_connector_python-5.2.0.dist-info/METADATA +190 -0
  202. snowflake_connector_python-5.2.0.dist-info/RECORD +204 -0
  203. snowflake_connector_python-5.2.0.dist-info/WHEEL +4 -0
  204. 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))