|
71 | 71 | )
|
72 | 72 | from ..network import SessionPool as SessionPoolSync
|
73 | 73 | from ..network import SnowflakeRestful as SnowflakeRestfulSync
|
74 |
| -from ..network import get_http_retryable_error, is_login_request, is_retryable_http_code |
| 74 | +from ..network import ( |
| 75 | + SnowflakeRestfulJsonEncoder, |
| 76 | + get_http_retryable_error, |
| 77 | + is_login_request, |
| 78 | + is_retryable_http_code, |
| 79 | +) |
75 | 80 | from ..secret_detector import SecretDetector
|
76 | 81 | from ..sqlstate import (
|
77 | 82 | SQLSTATE_CONNECTION_NOT_EXISTS,
|
@@ -236,7 +241,7 @@ async def request(
|
236 | 241 | return await self._post_request(
|
237 | 242 | url,
|
238 | 243 | headers,
|
239 |
| - json.dumps(body), |
| 244 | + json.dumps(body, cls=SnowflakeRestfulJsonEncoder), |
240 | 245 | token=self.token,
|
241 | 246 | _no_results=_no_results,
|
242 | 247 | timeout=timeout,
|
@@ -298,7 +303,7 @@ async def _token_request(self, request_type):
|
298 | 303 | ret = await self._post_request(
|
299 | 304 | url,
|
300 | 305 | headers,
|
301 |
| - json.dumps(body), |
| 306 | + json.dumps(body, cls=SnowflakeRestfulJsonEncoder), |
302 | 307 | token=header_token,
|
303 | 308 | )
|
304 | 309 | if ret.get("success") and ret.get("data", {}).get("sessionToken"):
|
@@ -396,7 +401,7 @@ async def delete_session(self, retry: bool = False) -> None:
|
396 | 401 | ret = await self._post_request(
|
397 | 402 | url,
|
398 | 403 | headers,
|
399 |
| - json.dumps(body), |
| 404 | + json.dumps(body, cls=SnowflakeRestfulJsonEncoder), |
400 | 405 | token=self.token,
|
401 | 406 | timeout=5,
|
402 | 407 | no_retry=True,
|
|
0 commit comments