nite/Byparr
1
Fork 0
mirror of https://github.com/ThePhaseless/Byparr.git synced 2025-03-15 01:40:21 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix new issue

Browse Source
This commit is contained in:
Thephaseless 2024-09-13 18:04:22 +00:00
parent b8f77474e7
commit 19ef18016a
1 changed files with 5 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
src/utils/browser.py
View File

@ -1,5 +1,4 @@
import asyncio
from typing import List
import nodriver as webdriver
from nodriver.core.element import Element
@ -83,7 +82,7 @@ async def bypass_cloudflare(page: webdriver.Tab):
elem = elem.parent
# Get the element containing the shadow root
for _ in range(4):
for _ in range(3):
if elem is not None:
elem = get_first_div(elem)
else:
@ -115,9 +114,10 @@ def get_first_div(elem):
The first div element found, or the original element if no div is found.
"""
if isinstance(elem.children, List):
elem = next(x for x in elem.children if x.tag_name == "div")
return elem
for child in elem.children:
if child.tag_name == "div":
return child
raise InvalidElementError
class InvalidElementError(Exception):