From c0c3218389cf6c2131849251d3f0e271838023ce Mon Sep 17 00:00:00 2001
From: Hengyu <h3ngyu@gmail.com>
Date: Fri, 30 Jun 2023 16:45:46 +0800
Subject: [PATCH] Upd: put load_rows in the loop when select_row

---
 module/ui/draggable_list.py | 9 +++++----
 1 file changed, 5 insertions(+), 4 deletions(-)

diff --git a/module/ui/draggable_list.py b/module/ui/draggable_list.py
index a5aa906b4..d450ea499 100644
--- a/module/ui/draggable_list.py
+++ b/module/ui/draggable_list.py
@@ -200,6 +200,10 @@ class DraggableList:
         Returns:
             If success
         """
+        result = self.insight_row(
+            row, main=main, skip_first_screenshot=skip_first_screenshot)
+        if not result:
+            return False
         logger.info(f'Select row: {row}')
         skip_first_screenshot = True
         interval = Timer(5)
@@ -208,11 +212,8 @@ class DraggableList:
                 skip_first_screenshot = False
             else:
                 main.device.screenshot()
+                self.load_rows(main=main)
 
-            result = self.insight_row(
-                row, main=main, skip_first_screenshot=skip_first_screenshot)
-            if not result:
-                return False
             button = self.keyword2button(row)
             if not button:
                 return False