fix logic

Author: blink1073

pymongo/asynchronous/pool.py

@@ -49,16 +49,13 @@
 from pymongo.errors import (  # type:ignore[attr-defined]
     AutoReconnect,
     ConfigurationError,
-    ConnectionFailure,
     DocumentTooLarge,
     ExecutionTimeout,
     InvalidOperation,
-    NetworkTimeout,
     NotPrimaryError,
     OperationFailure,
     PyMongoError,
     WaitQueueTimeoutError,
-    _OperationCancelled,
 )
 from pymongo.hello import Hello, HelloCompat
 from pymongo.helpers_shared import _get_timeout_details, format_timeout_details
@@ -1109,11 +1106,10 @@ async def connect(self, handler: Optional[_MongoClientErrorHandler] = None) -> A
             async with self.lock:
                 self.active_contexts.discard(conn.cancel_context)
             # Enter backoff mode and reconnect on establishment failure.
-            if isinstance(e, ConnectionFailure) and not isinstance(
-                e, (_OperationCancelled, NetworkTimeout)
-            ):
+            if type(e) == AutoReconnect:
                 self._backoff += 1
                 # TODO: emit a message about backoff.
+                print("backing off", self._backoff)  # noqa: T201
                 return await self.connect(handler)
             await conn.close_conn(ConnectionClosedReason.ERROR)
             raise

pymongo/synchronous/pool.py

@@ -46,16 +46,13 @@
 from pymongo.errors import (  # type:ignore[attr-defined]
     AutoReconnect,
     ConfigurationError,
-    ConnectionFailure,
     DocumentTooLarge,
     ExecutionTimeout,
     InvalidOperation,
-    NetworkTimeout,
     NotPrimaryError,
     OperationFailure,
     PyMongoError,
     WaitQueueTimeoutError,
-    _OperationCancelled,
 )
 from pymongo.hello import Hello, HelloCompat
 from pymongo.helpers_shared import _get_timeout_details, format_timeout_details
@@ -1105,11 +1102,10 @@ def connect(self, handler: Optional[_MongoClientErrorHandler] = None) -> Connect
             with self.lock:
                 self.active_contexts.discard(conn.cancel_context)
             # Enter backoff mode and reconnect on establishment failure.
-            if isinstance(e, ConnectionFailure) and not isinstance(
-                e, (_OperationCancelled, NetworkTimeout)
-            ):
+            if type(e) == AutoReconnect:
                 self._backoff += 1
                 # TODO: emit a message about backoff.
+                print("backing off", self._backoff)  # noqa: T201
                 return self.connect(handler)
             conn.close_conn(ConnectionClosedReason.ERROR)
             raise