From 6719be85db9d6b19320676698e90e311969b6db4 Mon Sep 17 00:00:00 2001
From: MangoFanFanw <mangofanfanw@icloud.com>
Date: Tue, 21 Apr 2026 22:17:45 +0800
Subject: [PATCH] =?UTF-8?q?=E6=B5=8B=E8=AF=95=20x2?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

设置最长截图等待时间，避免报错，至少把图截出来。
---
 router/enhance/screenshot.py | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/router/enhance/screenshot.py b/router/enhance/screenshot.py
index 8ee40f0..55d9bad 100644
--- a/router/enhance/screenshot.py
+++ b/router/enhance/screenshot.py
@@ -24,8 +24,10 @@ class ScreenShot(PlayContextManager):
 
     async def shot(self, save_path: str) -> None:
         await self.page.mouse.move(0, 0)
-        await self.page.wait_for_load_state("networkidle")
-        await self.page.screenshot(path=save_path)
+        try:
+            await self.page.wait_for_load_state("networkidle", timeout=1000)
+        finally:
+            await self.page.screenshot(path=save_path)
         logger.debug(f"截图 | 截图已经保存在 {save_path=}")
         return