添加重试

check_station.py

@@ -34,6 +34,17 @@ class CheckStation:
         # # 确保Appium服务器正在运行,不在运行则启动
         # if not driver_utils.check_server_status(4723):
         #     driver_utils.start_appium_server()
+        try:
+            if not driver_utils.check_session_valid(self.driver, self.device_id):
+                logging.warning(f"设备 {self.device_id} 会话无效，尝试重新连接驱动...")
+                self.driver, self.wait = driver_utils.reconnect_driver(self.device_id, self.driver)
+                if not self.driver:
+                    logging.error(f"设备 {self.device_id} 驱动重连失败")
+                    return False
+        except Exception as inner_e:
+            logging.warning(f"设备 {self.device_id} 检查会话状态时出错: {str(inner_e)}")
+            return False
+
 
         # 检查应用是否成功启动
         if driver_utils.is_app_launched(self.driver):
@@ -82,6 +93,17 @@ class CheckStation:
     #     return True
     def add_transition_point(self):
         """添加转点"""
+        try:
+            if not driver_utils.check_session_valid(self.driver, self.device_id):
+                logging.warning(f"设备 {self.device_id} 会话无效，尝试重新连接驱动...")
+                self.driver, self.wait = driver_utils.reconnect_driver(self.device_id, self.driver)
+                if not self.driver:
+                    logging.error(f"设备 {self.device_id} 驱动重连失败")
+                    return False
+        except Exception as inner_e:
+            logging.warning(f"设备 {self.device_id} 检查会话状态时出错: {str(inner_e)}")
+            return False
+
         try:
             # 查找并点击添加转点按钮
             add_transition_btn = self.wait.until(