debug

Author: blink1073

pymongo/asynchronous/auth_oidc.py

@@ -99,10 +99,15 @@ async def authenticate(self, conn: AsyncConnection) -> Optional[Mapping[str, Any
         # If it succeeded, we are done.
         ctx = conn.auth_ctx
         if ctx and ctx.speculate_succeeded():
+            print("speculative auth succeeded")
             resp = ctx.speculative_authenticate
             if resp and resp["done"]:
                 conn.oidc_token_gen_id = self.token_gen_id
                 return resp
+        elif ctx:
+            print("speculative auth failed")
+        else:
+            print("no speculative auth")
 
         # If spec auth failed, call the appropriate auth logic for the callback type.
         # We cannot assume that the token is invalid, because a proxy may have been

pymongo/asynchronous/helpers.py

@@ -45,11 +45,13 @@ async def inner(*args: Any, **kwargs: Any) -> Any:
             return await func(*args, **kwargs)
         except OperationFailure as exc:
             if no_reauth:
+                print("failure with no reauth")
                 raise
             if exc.code == _REAUTHENTICATION_REQUIRED_CODE:
                 # Look for an argument that either is a AsyncConnection
                 # or has a connection attribute, so we can trigger
                 # a reauth.
+                print("got reauth")
                 conn = None
                 for arg in args:
                     if isinstance(arg, AsyncConnection):
@@ -59,8 +61,10 @@ async def inner(*args: Any, **kwargs: Any) -> Any:
                         conn = arg.conn  # type: ignore[assignment]
                         break
                 if conn:
+                    print("running reauth")
                     await conn.authenticate(reauthenticate=True)
                 else:
+                    print("not running reauth, no conn")
                     raise
                 return func(*args, **kwargs)
             raise

pymongo/synchronous/auth_oidc.py

@@ -99,10 +99,15 @@ def authenticate(self, conn: Connection) -> Optional[Mapping[str, Any]]:
         # If it succeeded, we are done.
         ctx = conn.auth_ctx
         if ctx and ctx.speculate_succeeded():
+            print("speculative auth succeeded")
             resp = ctx.speculative_authenticate
             if resp and resp["done"]:
                 conn.oidc_token_gen_id = self.token_gen_id
                 return resp
+        elif ctx:
+            print("speculative auth failed")
+        else:
+            print("no speculative auth")
 
         # If spec auth failed, call the appropriate auth logic for the callback type.
         # We cannot assume that the token is invalid, because a proxy may have been

pymongo/synchronous/helpers.py

@@ -45,11 +45,13 @@ def inner(*args: Any, **kwargs: Any) -> Any:
             return func(*args, **kwargs)
         except OperationFailure as exc:
             if no_reauth:
+                print("failure with no reauth")
                 raise
             if exc.code == _REAUTHENTICATION_REQUIRED_CODE:
                 # Look for an argument that either is a Connection
                 # or has a connection attribute, so we can trigger
                 # a reauth.
+                print("got reauth")
                 conn = None
                 for arg in args:
                     if isinstance(arg, Connection):
@@ -59,8 +61,10 @@ def inner(*args: Any, **kwargs: Any) -> Any:
                         conn = arg.conn  # type: ignore[assignment]
                         break
                 if conn:
+                    print("running reauth")
                     conn.authenticate(reauthenticate=True)
                 else:
+                    print("not running reauth, no conn")
                     raise
                 return func(*args, **kwargs)
             raise

pyproject.toml

@@ -193,6 +193,7 @@ ignore = [
   "EM103",  # Exception must not use a `.format()` string directly, assign to variable first
   "C408",   # Unnecessary `dict` call (rewrite as a literal)
   "SIM117", # Use a single `with` statement with multiple contexts instead of nested `with` statements
+  "T20",
 ]
 unfixable = [
   "RUF100", # Unused noqa