notebook

Medidata/download_query_details.py

@@ -128,8 +128,12 @@ def _okta_mfa_present(page):
         'input[name*="code"]',
         'input[placeholder*="code" i]',
     ]:
+        try:
             if page.query_selector(sel):
                 return True
+        except Exception:
+            # Page navigated during selector, skip
+            pass
     return False
 
 
@@ -141,13 +145,20 @@ def _fill_otp(page, otp):
         'input[type="tel"]',
         'input[placeholder*="code" i]',
     ]:
+        try:
             el = page.query_selector(sel)
             if el:
                 el.fill(otp)
                 page.keyboard.press("Enter")
                 return
+        except Exception:
+            # Page navigated, continue to next selector
+            pass
+    try:
         page.keyboard.type(otp)
         page.keyboard.press("Enter")
+    except Exception:
+        pass
 
 
 # ---------------------------------------------------------------------------