46
46
_HAVE_SSL = False
47
47
48
48
try :
49
- from pymongo .pyopenssl_context import _sslConn as _pysslConn
49
+ from pymongo .pyopenssl_context import _sslConn
50
50
51
51
_HAVE_PYOPENSSL = True
52
52
except ImportError :
53
53
_HAVE_PYOPENSSL = False
54
- _pysslConn = SSLSocket # type: ignore[assignment, misc]
54
+ _sslConn = SSLSocket # type: ignore[assignment, misc]
55
55
56
56
from pymongo .ssl_support import (
57
57
BLOCKING_IO_LOOKUP_ERROR ,
76
76
77
77
# These socket-based I/O methods are for KMS requests and any other network operations that do not use
78
78
# the MongoDB wire protocol
79
- async def async_socket_sendall (sock : Union [socket .socket , _pysslConn ], buf : bytes ) -> None :
79
+ async def async_socket_sendall (sock : Union [socket .socket , _sslConn ], buf : bytes ) -> None :
80
80
timeout = sock .gettimeout ()
81
81
sock .settimeout (0.0 )
82
82
loop = asyncio .get_running_loop ()
83
83
try :
84
- if _HAVE_SSL and isinstance (sock , (SSLSocket , _pysslConn )):
84
+ if _HAVE_SSL and isinstance (sock , (SSLSocket , _sslConn )):
85
85
await asyncio .wait_for (_async_socket_sendall_ssl (sock , buf , loop ), timeout = timeout )
86
86
else :
87
87
await asyncio .wait_for (loop .sock_sendall (sock , buf ), timeout = timeout ) # type: ignore[arg-type]
@@ -95,7 +95,7 @@ async def async_socket_sendall(sock: Union[socket.socket, _pysslConn], buf: byte
95
95
if sys .platform != "win32" :
96
96
97
97
async def _async_socket_sendall_ssl (
98
- sock : Union [socket .socket , _pysslConn ], buf : bytes , loop : AbstractEventLoop
98
+ sock : Union [socket .socket , _sslConn ], buf : bytes , loop : AbstractEventLoop
99
99
) -> None :
100
100
view = memoryview (buf )
101
101
sent = 0
@@ -138,7 +138,7 @@ def _is_ready(fut: Future) -> None:
138
138
loop .remove_writer (fd )
139
139
140
140
async def _async_socket_receive_ssl (
141
- conn : _pysslConn , length : int , loop : AbstractEventLoop , once : Optional [bool ] = False
141
+ conn : _sslConn , length : int , loop : AbstractEventLoop , once : Optional [bool ] = False
142
142
) -> memoryview :
143
143
mv = memoryview (bytearray (length ))
144
144
total_read = 0
@@ -192,7 +192,7 @@ def _is_ready(fut: Future) -> None:
192
192
# https://docs.python.org/3/library/asyncio-platforms.html#asyncio-platform-support
193
193
# Note: In PYTHON-4493 we plan to replace this code with asyncio streams.
194
194
async def _async_socket_sendall_ssl (
195
- sock : Union [socket .socket , _pysslConn ], buf : bytes , dummy : AbstractEventLoop
195
+ sock : Union [socket .socket , _sslConn ], buf : bytes , dummy : AbstractEventLoop
196
196
) -> None :
197
197
view = memoryview (buf )
198
198
total_length = len (buf )
@@ -213,7 +213,7 @@ async def _async_socket_sendall_ssl(
213
213
total_sent += sent
214
214
215
215
async def _async_socket_receive_ssl (
216
- conn : _pysslConn , length : int , dummy : AbstractEventLoop , once : Optional [bool ] = False
216
+ conn : _sslConn , length : int , dummy : AbstractEventLoop , once : Optional [bool ] = False
217
217
) -> memoryview :
218
218
mv = memoryview (bytearray (length ))
219
219
total_read = 0
@@ -239,7 +239,7 @@ async def _async_socket_receive_ssl(
239
239
return mv
240
240
241
241
242
- def sendall (sock : Union [socket .socket , _pysslConn ], buf : bytes ) -> None :
242
+ def sendall (sock : Union [socket .socket , _sslConn ], buf : bytes ) -> None :
243
243
sock .sendall (buf )
244
244
245
245
@@ -252,15 +252,15 @@ async def _poll_cancellation(conn: AsyncConnection) -> None:
252
252
253
253
254
254
async def async_receive_data_socket (
255
- sock : Union [socket .socket , _pysslConn ], length : int
255
+ sock : Union [socket .socket , _sslConn ], length : int
256
256
) -> memoryview :
257
257
sock_timeout = sock .gettimeout ()
258
258
timeout = sock_timeout
259
259
260
260
sock .settimeout (0.0 )
261
261
loop = asyncio .get_running_loop ()
262
262
try :
263
- if _HAVE_SSL and isinstance (sock , (SSLSocket , _pysslConn )):
263
+ if _HAVE_SSL and isinstance (sock , (SSLSocket , _sslConn )):
264
264
return await asyncio .wait_for (
265
265
_async_socket_receive_ssl (sock , length , loop , once = True ), # type: ignore[arg-type]
266
266
timeout = timeout ,
@@ -435,7 +435,7 @@ def sock(self) -> socket.socket:
435
435
436
436
437
437
class NetworkingInterface (NetworkingInterfaceBase ):
438
- def __init__ (self , conn : Union [socket .socket , _pysslConn ]):
438
+ def __init__ (self , conn : Union [socket .socket , _sslConn ]):
439
439
super ().__init__ (conn )
440
440
441
441
def gettimeout (self ) -> float | None :
@@ -451,11 +451,11 @@ def is_closing(self) -> bool:
451
451
return self .conn .is_closing ()
452
452
453
453
@property
454
- def get_conn (self ) -> Union [socket .socket , _pysslConn ]:
454
+ def get_conn (self ) -> Union [socket .socket , _sslConn ]:
455
455
return self .conn
456
456
457
457
@property
458
- def sock (self ) -> Union [socket .socket , _pysslConn ]:
458
+ def sock (self ) -> Union [socket .socket , _sslConn ]:
459
459
return self .conn
460
460
461
461
def fileno (self ) -> int :
0 commit comments