Update nb4x.py

nb4x.py

@@ -24,6 +24,11 @@ logging.basicConfig(
 )
 logger = logging.getLogger(__name__)
 
+# 在文件开头添加配置
+MAX_CONSECUTIVE_FAILURES = 3  # 最大连续失败次数
+ENABLE_FAILURE_STOP = True    # 失败停止开关
+consecutive_failures = 0      # 连续失败计数器
+
 def get_verification_code(driver, email, max_retries=10, delay=2):
     email_url = f'https://linshiyou.com/#/{email}'
     
@@ -55,7 +60,7 @@ def get_verification_code(driver, email, max_retries=10, delay=2):
                 return verification_code
                 
         except Exception as e:
-            logger.error(f"获取验证码时发生错误: {str(e)}")
+            logger.error(f"获取���证码时发生错误: {str(e)}")
             
         time.sleep(delay)
         logger.info(f"等待验证码，重试第 {_ + 1} 次")
@@ -122,15 +127,23 @@ def wait_for_url(driver, url, timeout=10):
 
 def wait_for_email_input(driver, timeout=10):
     """等待邮箱输入框出现并返回"""
+    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] + "...")  # 只打印前500字符
+            logger.info("=" * 50)
+            
             # 获取页面标题
             logger.info(f"当前页面标题: {driver.title}")
             
             # 检查页面源代码中是否包含关键元素
-            logger.info("检查页面源代码...")
-            if "descope-wc" in driver.page_source:
+            if "descope-wc" in page_source:
                 logger.info("找到 descope-wc 元素")
             else:
                 logger.info("未找到 descope-wc 元素")
@@ -179,6 +192,15 @@ def wait_for_email_input(driver, timeout=10):
         logger.info("等待0.5秒后重试...")
         time.sleep(0.5)
         
+    # 更新失败计数
+    if ENABLE_FAILURE_STOP:
+        consecutive_failures += 1
+        logger.warning(f"连续失败次数: {consecutive_failures}")
+        if consecutive_failures >= MAX_CONSECUTIVE_FAILURES:
+            logger.critical(f"连续失败达到{MAX_CONSECUTIVE_FAILURES}次，停止脚本运行")
+            import sys
+            sys.exit(1)  # 强制停止脚本
+    
     logger.error("邮箱输入框未出现")
     return False
 
@@ -221,6 +243,8 @@ def input_verification_code(driver, code):
 
 def registration_process(driver, wait):
     try:
+        global consecutive_failures  # 使用全局计数器
+        
         # 生成邮箱地址
         email = generate_email()
         if not email: