Skip to content

Commit 8455689

Browse files
committed
PYTHON-2560 Fix asyncio.sleep, synchro can't handle it
1 parent e413334 commit 8455689

File tree

2 files changed

+10
-6
lines changed

2 files changed

+10
-6
lines changed

pymongo/asynchronous/encryption.py

Lines changed: 5 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -19,6 +19,7 @@
1919
import contextlib
2020
import enum
2121
import socket
22+
import time as time # noqa: PLC0414 # needed in sync version
2223
import uuid
2324
import weakref
2425
from copy import deepcopy
@@ -176,9 +177,10 @@ async def kms_request(self, kms_context: MongoCryptKmsContext) -> None:
176177
ssl_context=ctx,
177178
)
178179
host, port = parse_host(endpoint, _HTTPS_PORT)
179-
sleep_usec = kms_context.usleep
180-
if sleep_usec:
181-
await asyncio.sleep(float(sleep_usec) / 1e6)
180+
sleep_u = kms_context.usleep
181+
if sleep_u:
182+
sleep_sec = float(sleep_u) / 1e6
183+
await asyncio.sleep(sleep_sec)
182184
try:
183185
conn = await _configured_socket((host, port), opts)
184186
try:

pymongo/synchronous/encryption.py

Lines changed: 5 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -19,6 +19,7 @@
1919
import contextlib
2020
import enum
2121
import socket
22+
import time as time # noqa: PLC0414 # needed in sync version
2223
import uuid
2324
import weakref
2425
from copy import deepcopy
@@ -176,9 +177,10 @@ def kms_request(self, kms_context: MongoCryptKmsContext) -> None:
176177
ssl_context=ctx,
177178
)
178179
host, port = parse_host(endpoint, _HTTPS_PORT)
179-
sleep_usec = kms_context.usleep
180-
if sleep_usec:
181-
asyncio.sleep(float(sleep_usec) / 1e6)
180+
sleep_u = kms_context.usleep
181+
if sleep_u:
182+
sleep_sec = float(sleep_u) / 1e6
183+
time.sleep(sleep_sec)
182184
try:
183185
conn = _configured_socket((host, port), opts)
184186
try:

0 commit comments

Comments
 (0)