串起截图和上传

page_objects/login_page.py

@@ -73,22 +73,32 @@ class LoginPage:
                 pass
             
             # 点击登录按钮
-            login_btn = self.wait.until(
-                EC.element_to_be_clickable((AppiumBy.ID, ids.LOGIN_BTN))
-            )
-            login_btn.click()
-            self.logger.info("已点击登录按钮")
+            max_retries = 3
+            retry_count = 0
             
-            # 等待登录完成
-            time.sleep(3)
+            while retry_count < max_retries:
+                login_btn = self.wait.until(
+                    EC.element_to_be_clickable((AppiumBy.ID, ids.LOGIN_BTN))
+                )
+                login_btn.click()
+                self.logger.info(f"已点击登录按钮 (尝试 {retry_count + 1}/{max_retries})")
+                
+                # 等待登录完成
+                time.sleep(3)
+                
+                # 检查是否登录成功
+                if self.is_login_successful():
+                    self.logger.info("登录成功")
+                    return True
+                else:
+                    self.logger.warning("登录后未检测到主页面元素，准备重试")
+                    retry_count += 1
+                    if retry_count < max_retries:
+                        self.logger.info(f"等待2秒后重新尝试登录...")
+                        time.sleep(2)
             
-            # 检查是否登录成功
-            if self.is_login_successful():
-                self.logger.info("登录成功")
-                return True
-            else:
-                self.logger.warning("登录后未检测到主页面元素")
-                return False
+            self.logger.error(f"登录失败，已尝试 {max_retries} 次")
+            return False
                 
         except Exception as e:
             self.logger.error(f"登录过程中出错: {str(e)}")