fix: handle first screenshot in loop()

2026-10-10 04:52:05 +00:00 · 2026-07-06 22:41:00 +08:00
parent d033dd17c7
commit eaf806e02d
1 changed files with 5 additions and 1 deletions
@@ -91,7 +91,11 @@ class ModuleBase:
                skip_first = False
            else:
                self.device.screenshot()
-            yield self.device.image
+            try:
+                yield self.device.image
+            except AttributeError:
+                self.device.screenshot()
+                yield self.device.image

    def loop_hierarchy(self, skip_first=True):
        """