From d203f68977cee868640cbcf6d224029e3eb0e61f Mon Sep 17 00:00:00 2001 From: Shane Harvey Date: Tue, 8 Aug 2023 11:21:01 -0700 Subject: [PATCH] PYTHON-3904 Fix socket vs _socket pyopenssl typing (#1345) --- pymongo/pyopenssl_context.py | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/pymongo/pyopenssl_context.py b/pymongo/pyopenssl_context.py index d6dac84f4..c16917381 100644 --- a/pymongo/pyopenssl_context.py +++ b/pymongo/pyopenssl_context.py @@ -42,7 +42,6 @@ from pymongo.socket_checker import _errno_from_exception from pymongo.write_concern import validate_boolean if TYPE_CHECKING: - import socket from ssl import VerifyMode from cryptography.x509 import Certificate @@ -106,7 +105,7 @@ def _ragged_eof(exc: BaseException) -> bool: # https://docs.python.org/3/library/ssl.html#notes-on-non-blocking-sockets class _sslConn(_SSL.Connection): def __init__( - self, ctx: _SSL.Context, sock: Optional[socket.socket], suppress_ragged_eofs: bool + self, ctx: _SSL.Context, sock: Optional[_socket.socket], suppress_ragged_eofs: bool ): self.socket_checker = _SocketChecker() self.suppress_ragged_eofs = suppress_ragged_eofs @@ -370,7 +369,7 @@ class SSLContext: def wrap_socket( self, - sock: socket.socket, + sock: _socket.socket, server_side: bool = False, do_handshake_on_connect: bool = True, suppress_ragged_eofs: bool = True,