Skip to content

Commit acf59cf

Browse files
[ASYNC] apply #2254 to async code
1 parent 34a873b commit acf59cf

File tree

1 file changed

+9
-4
lines changed

1 file changed

+9
-4
lines changed

src/snowflake/connector/aio/_network.py

Lines changed: 9 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -71,7 +71,12 @@
7171
)
7272
from ..network import SessionPool as SessionPoolSync
7373
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+
)
7580
from ..secret_detector import SecretDetector
7681
from ..sqlstate import (
7782
SQLSTATE_CONNECTION_NOT_EXISTS,
@@ -236,7 +241,7 @@ async def request(
236241
return await self._post_request(
237242
url,
238243
headers,
239-
json.dumps(body),
244+
json.dumps(body, cls=SnowflakeRestfulJsonEncoder),
240245
token=self.token,
241246
_no_results=_no_results,
242247
timeout=timeout,
@@ -298,7 +303,7 @@ async def _token_request(self, request_type):
298303
ret = await self._post_request(
299304
url,
300305
headers,
301-
json.dumps(body),
306+
json.dumps(body, cls=SnowflakeRestfulJsonEncoder),
302307
token=header_token,
303308
)
304309
if ret.get("success") and ret.get("data", {}).get("sessionToken"):
@@ -396,7 +401,7 @@ async def delete_session(self, retry: bool = False) -> None:
396401
ret = await self._post_request(
397402
url,
398403
headers,
399-
json.dumps(body),
404+
json.dumps(body, cls=SnowflakeRestfulJsonEncoder),
400405
token=self.token,
401406
timeout=5,
402407
no_retry=True,

0 commit comments

Comments
 (0)