codfrm/MoviePilot
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

add plsywright timeout

Browse Source
This commit is contained in:
jxxghp 2023-06-09 18:13:21 +08:00
parent 83b1c2375a
commit 82bb64f388
1 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
app/helper/playwright.py
View File

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