From eaf806e02daf1ffb56467e09c805cdba05a2ea8d Mon Sep 17 00:00:00 2001
From: YoursFunny <admin@yoursfunny.top>
Date: Mon, 6 Jul 2026 20:14:11 +0800
Subject: [PATCH] fix: handle first screenshot in loop()

---
 module/base/base.py | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/module/base/base.py b/module/base/base.py
index 612842f..7d47c34 100644
--- a/module/base/base.py
+++ b/module/base/base.py
@@ -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):
         """