Update nb4x.py

nb4x.py

@@ -129,69 +129,85 @@ def wait_for_email_input(driver, timeout=30):
     """等待邮箱输入框出现并返回"""
     global consecutive_failures
     
-    start_time = time.time()
-    while time.time() - start_time < timeout:
-        try:
-            # 获取并记录页面源码
-            page_source = driver.page_source
-            logger.info("当前页面源码:")
-            logger.info("=" * 50)
-            logger.info(page_source[:500] + "...")
-            logger.info("=" * 50)
-            
-            # 获取页面标题
-            logger.info(f"当前页面标题: {driver.title}")
-            
-            # 检查页面源代码中是否包含关键元素
-            if "descope-wc" in page_source:
-                logger.info("找到 descope-wc 元素")
-            else:
-                logger.info("未找到 descope-wc 元素")
-            
-            # 尝试获取 Shadow DOM 元素
+    # 先等待页面加载完成
+    logger.info("等待页面完全加载...")
+    try:
+        WebDriverWait(driver, 20).until(
+            lambda d: d.execute_script('return document.readyState') == 'complete'
+        )
+        logger.info("页面加载完成")
+        
+        # 再等待10秒，确保JavaScript完全执行
+        logger.info("等待10秒让JavaScript完全执行...")
+        time.sleep(10)
+        
+        # 现在开始检查元素
+        start_time = time.time()
+        while time.time() - start_time < timeout:
             try:
-                wrapper = driver.find_element(By.CSS_SELECTOR, "#AppProvider_Wrapper > form > descope-wc")
-                logger.info("找到 wrapper 元素")
-            except:
-                logger.info("未找到 wrapper 元素")
+                # 获取并记录页面源码
+                page_source = driver.page_source
+                logger.info("当前页面源码:")
+                logger.info("=" * 50)
+                logger.info(page_source[:500] + "...")
+                logger.info("=" * 50)
                 
-            # 原来的元素查找逻辑
-            input_js = """
-            var wrapper = document.querySelector("#AppProvider_Wrapper > form > descope-wc");
-            if (!wrapper) {
-                return "wrapper not found";
-            }
-            var shadow1 = wrapper.shadowRoot;
-            if (!shadow1) {
-                return "shadow1 not found";
-            }
-            var elem = shadow1.querySelector("#TExZpnv5m6");
-            if (!elem) {
-                return "elem not found";
-            }
-            var shadow2 = elem.shadowRoot;
-            if (!shadow2) {
-                return "shadow2 not found";
-            }
-            var input = shadow2.querySelector("#input-vaadin-email-field-3");
-            if (!input) {
-                return "input not found";
-            }
-            return "found";
-            """
-            result = driver.execute_script(input_js)
-            logger.info(f"元素查找结果: {result}")
-            
-            if result == "found":
-                logger.info("邮箱输入框已加载")
-                return True
+                # 获取页面标题
+                logger.info(f"当前页面标题: {driver.title}")
                 
-        except Exception as e:
-            logger.error(f"查找元素时发生错误: {str(e)}")
-            
-        logger.info("等待10秒后重试...")
-        time.sleep(10)
+                # 检查页面源代码中是否包含关键元素
+                if "descope-wc" in page_source:
+                    logger.info("找到 descope-wc 元素")
+                else:
+                    logger.info("未找到 descope-wc 元素")
+                
+                # 尝试获取 Shadow DOM 元素
+                try:
+                    wrapper = driver.find_element(By.CSS_SELECTOR, "#AppProvider_Wrapper > form > descope-wc")
+                    logger.info("找到 wrapper 元素")
+                except:
+                    logger.info("未找到 wrapper 元素")
+                    
+                # 原来的元素查找逻辑
+                input_js = """
+                var wrapper = document.querySelector("#AppProvider_Wrapper > form > descope-wc");
+                if (!wrapper) {
+                    return "wrapper not found";
+                }
+                var shadow1 = wrapper.shadowRoot;
+                if (!shadow1) {
+                    return "shadow1 not found";
+                }
+                var elem = shadow1.querySelector("#TExZpnv5m6");
+                if (!elem) {
+                    return "elem not found";
+                }
+                var shadow2 = elem.shadowRoot;
+                if (!shadow2) {
+                    return "shadow2 not found";
+                }
+                var input = shadow2.querySelector("#input-vaadin-email-field-3");
+                if (!input) {
+                    return "input not found";
+                }
+                return "found";
+                """
+                result = driver.execute_script(input_js)
+                logger.info(f"元素查找结果: {result}")
+                
+                if result == "found":
+                    logger.info("邮箱输入框已加载")
+                    return True
+                    
+            except Exception as e:
+                logger.error(f"查找元素时发生错误: {str(e)}")
+                
+            logger.info("等待10秒后重试...")
+            time.sleep(10)
         
+    except Exception as e:
+        logger.error(f"等待页面加载时发生错误: {str(e)}")
+    
     # 更新失败计数
     if ENABLE_FAILURE_STOP:
         consecutive_failures += 1