blackteay/DrissionPage
1
0
Fork 0
mirror of https://gitee.com/g1879/DrissionPage.git synced 2024-12-10 04:00:23 +08:00
Code Issues Projects Releases Wiki Activity

ShadowRootElement增加.和#方式用于查找元素

Browse Source
This commit is contained in:
g1879 2020-11-23 12:46:16 +08:00
parent a3efe2a772
commit b070077642
1 changed files with 36 additions and 33 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

43
DrissionPage/shadow_root_element.py
View File

@ -5,16 +5,15 @@ from typing import Union, Any, Tuple
from selenium.webdriver.remote.webelement import WebElement from selenium.webdriver.remote.webelement import WebElement
from .common import DrissionElement, format_html from .common import format_html, DrissionElement
from .driver_element import execute_driver_find from .driver_element import execute_driver_find, DriverElement
class ShadowRootElement(DrissionElement): class ShadowRootElement(DrissionElement):
def __init__(self, inner_ele: WebElement, parent_ele, timeout: float = 10): def __init__(self, inner_ele: WebElement, parent_ele: DriverElement, timeout: float = 10):
super().__init__(inner_ele) super().__init__(inner_ele, parent_ele.page)
self.parent_ele = parent_ele self.parent_ele = parent_ele
self.timeout = timeout self.timeout = timeout
self._driver = inner_ele.parent
def __repr__(self): def __repr__(self):
return f'<ShadowRootElement in {self.parent_ele} >' return f'<ShadowRootElement in {self.parent_ele} >'
@ -32,21 +31,19 @@ class ShadowRootElement(DrissionElement):
""" """
return self.ele(loc_or_str, mode, timeout or self.timeout) return self.ele(loc_or_str, mode, timeout or self.timeout)
@property
def driver(self):
"""返回控制元素的WebDriver对象"""
return self._driver
@property @property
def tag(self): def tag(self):
"""元素标签名"""
return 'shadow-root' return 'shadow-root'
@property @property
def html(self): def html(self):
"""内部html文本"""
return format_html(self.inner_ele.get_attribute('innerHTML')) return format_html(self.inner_ele.get_attribute('innerHTML'))
@property @property
def parent(self): def parent(self):
"""shadow-root所依赖的父元素"""
return self.parent_ele return self.parent_ele
def parents(self, num: int = 1): def parents(self, num: int = 1):
@ -113,7 +110,7 @@ class ShadowRootElement(DrissionElement):
timeout = timeout or self.timeout timeout = timeout or self.timeout
if loc_or_str[0] == 'css selector': if loc_or_str[0] == 'css selector':
return execute_driver_find(self.inner_ele, loc_or_str, mode, timeout) return execute_driver_find(self, loc_or_str, mode, timeout)
elif loc_or_str[0] == 'text': elif loc_or_str[0] == 'text':
return self._find_eles_by_text(loc_or_str[1], loc_or_str[2], loc_or_str[3], mode) return self._find_eles_by_text(loc_or_str[1], loc_or_str[2], loc_or_str[3], mode)
@ -179,32 +176,34 @@ class ShadowRootElement(DrissionElement):
""" """
# 获取所有元素 # 获取所有元素
eles = self.run_script('return arguments[0].querySelectorAll("*")') eles = self.run_script('return arguments[0].querySelectorAll("*")')
from .driver_element import DriverElement
results = [] results = []
# 遍历所有元素，找到符合条件的 # 遍历所有元素，找到符合条件的
for ele in eles: for ele in eles:
if tag and tag != ele.tag_name: if tag and tag != ele.tag_name:
continue continue
txt = self.driver.execute_script(
txt = self.page.driver.execute_script(
'if(arguments[0].firstChild!=null){return arguments[0].firstChild.nodeValue}', ele) 'if(arguments[0].firstChild!=null){return arguments[0].firstChild.nodeValue}', ele)
txt = txt or '' txt = txt or ''
# 匹配没有文本的元素或精确匹配 # 匹配没有文本的元素或精确匹配
if text == '' or match == 'exact': if text == '' or match == 'exact':
if text == txt: if text == txt:
if mode == 'single': if mode == 'single':
return DriverElement(ele) return DriverElement(ele, self.page, self.timeout)
elif mode == 'all': elif mode == 'all':
results.append(DriverElement(ele)) results.append(DriverElement(ele, self.page, self.timeout))
# 模糊匹配 # 模糊匹配
elif match == 'fuzzy': elif match == 'fuzzy':
if text in txt: if text in txt:
if mode == 'single': if mode == 'single':
return DriverElement(ele) return DriverElement(ele, self.page, self.timeout)
elif mode == 'all': elif mode == 'all':
results.append(DriverElement(ele)) results.append(DriverElement(ele, self.page, self.timeout))
return None if mode == 'single' else results return None if mode == 'single' else results
@ -212,9 +211,12 @@ class ShadowRootElement(DrissionElement):
def str_to_css_loc(loc: str) -> tuple: def str_to_css_loc(loc: str) -> tuple:
"""处理元素查找语句 \n """处理元素查找语句 \n
查找方式：属性、tag name及属性、文本、css selector \n 查找方式：属性、tag name及属性、文本、css selector \n
=表示精确匹配，:表示模糊匹配，无控制字符串时默认搜索该字符串 \n @表示属性，.表示class，#表示id，=表示精确匹配，:表示模糊匹配，无控制字符串时默认搜索该字符串 \n
=表示精确匹配，:表示模糊匹配，无控制字符串时默认搜索该字符串 \n
示例： \n 示例： \n
.ele_class - class为ele_class的元素 \n
.:ele_class - class含有ele_class的元素 \n
#ele_id - id为ele_id的元素 \n
#:ele_id - id含有ele_id的元素 \n
@class:ele_class - class含有ele_class的元素 \n @class:ele_class - class含有ele_class的元素 \n
@class=ele_class - class等于ele_class的元素 \n @class=ele_class - class等于ele_class的元素 \n
@class - 带class属性的元素 \n @class - 带class属性的元素 \n
@ -225,7 +227,7 @@ def str_to_css_loc(loc: str) -> tuple:
tag:div@text()=search_text - 文本等于search_text的div元素 \n tag:div@text()=search_text - 文本等于search_text的div元素 \n
text:search_text - 文本含有search_text的元素 \n text:search_text - 文本含有search_text的元素 \n
text=search_text - 文本等于search_text的元素 \n text=search_text - 文本等于search_text的元素 \n
css:div.ele_class \n css:div.ele_class - 符合css selector的元素 \n
""" """
loc_by = 'css selector' loc_by = 'css selector'
@ -256,6 +258,7 @@ def str_to_css_loc(loc: str) -> tuple:
elif loc.startswith(('tag=', 'tag:')): elif loc.startswith(('tag=', 'tag:')):
if '@' not in loc[4:]: if '@' not in loc[4:]:
loc_str = f'{loc[4:]}' loc_str = f'{loc[4:]}'
else: else:
at_lst = loc[4:].split('@', maxsplit=1) at_lst = loc[4:].split('@', maxsplit=1)
r = re_SPLIT(r'([:=])', at_lst[1], maxsplit=1) r = re_SPLIT(r'([:=])', at_lst[1], maxsplit=1)