🐛 Fix handle PlaywrightTimeoutError

PaiGramTeam · Nov 13, 2024 · d0681c6 · d0681c6
1 parent e1becdf
commit d0681c6
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/plugins/system/errorhandler.py b/plugins/system/errorhandler.py
@@ -6,6 +6,7 @@
 
 import aiofiles
 from httpx import HTTPError, TimeoutException
+from playwright.async_api import Error as PlaywrightError, TimeoutError as PlaywrightTimeoutError
 from simnet.errors import (
     DataNotPublic,
     BadRequest as SIMNetBadRequest,
@@ -276,6 +277,15 @@ async def process_public_cookies(self, update: object, context: CallbackContext)
         self.create_notice_task(update, context, config.notice.user_not_found)
         raise ApplicationHandlerStop
 
+    @error_handler()
+    async def process_playwright_exception(self, update: object, context: CallbackContext):
+        if not isinstance(context.error, PlaywrightError) or not isinstance(update, Update):
+            return
+        if isinstance(context.error, PlaywrightTimeoutError):
+            notice = self.ERROR_MSG_PREFIX + " 渲染超时 ~ 请稍后再试"
+            self.create_notice_task(update, context, notice)
+            raise ApplicationHandlerStop
+
     @error_handler(block=False)
     async def process_z_error(self, update: object, context: CallbackContext) -> None:
         # 必须 `process_` 加上 `z` 保证该函数最后一个注册