fix: handle first screenshot in loop()

This commit is contained in:
2026-07-06 22:41:00 +08:00
parent d033dd17c7
commit eaf806e02d
+4
View File
@@ -91,6 +91,10 @@ class ModuleBase:
skip_first = False skip_first = False
else: else:
self.device.screenshot() self.device.screenshot()
try:
yield self.device.image
except AttributeError:
self.device.screenshot()
yield self.device.image yield self.device.image
def loop_hierarchy(self, skip_first=True): def loop_hierarchy(self, skip_first=True):