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