From 82bb64f388cc62c4e735ea6e37d4d01542cb9c67 Mon Sep 17 00:00:00 2001
From: jxxghp <jxxghp@163.com>
Date: Fri, 9 Jun 2023 18:13:21 +0800
Subject: [PATCH] add plsywright timeout

---
 app/helper/playwright.py | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/app/helper/playwright.py b/app/helper/playwright.py
index ee8b449f..a49532a1 100644
--- a/app/helper/playwright.py
+++ b/app/helper/playwright.py
@@ -9,7 +9,8 @@ class PlaywrightHelper:
                         cookie: str = None,
                         ua: str = None,
                         proxy: dict = None,
-                        headless: bool = True):
+                        headless: bool = True,
+                        timeout: int = 30) -> str:
         """
         获取网页源码
         :param url: 网页地址
@@ -17,6 +18,7 @@ class PlaywrightHelper:
         :param ua: user-agent
         :param proxy: 代理
         :param headless: 是否无头模式
+        :param timeout: 超时时间
         """
         with sync_playwright() as playwright:
             browser = playwright[self.browser_type].launch(headless=headless)
@@ -25,7 +27,7 @@ class PlaywrightHelper:
             if cookie:
                 page.set_extra_http_headers({"cookie": cookie})
             page.goto(url)
-            page.wait_for_load_state("networkidle")
+            page.wait_for_load_state("networkidle", timeout=timeout)
             source = page.content()
             browser.close()