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

ShadowElement增加__call__()

Browse Source
This commit is contained in:
g1879 2020-11-13 00:09:38 +08:00
parent 70cc8da88d
commit cad7d62eb0
1 changed files with 16 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
DrissionPage/shadow_root_element.py
View File

@ -1,7 +1,7 @@
#!/usr/bin/env python #!/usr/bin/env python
# -*- coding:utf-8 -*- # -*- coding:utf-8 -*-
from re import split as re_SPLIT from re import split as re_SPLIT
from typing import Union, Any from typing import Union, Any, Tuple
from selenium.webdriver.remote.webelement import WebElement from selenium.webdriver.remote.webelement import WebElement
@ -19,6 +19,19 @@ class ShadowRootElement(DrissionElement):
def __repr__(self): def __repr__(self):
return f'<ShadowRootElement in {self.parent_ele} >' return f'<ShadowRootElement in {self.parent_ele} >'
def __call__(self,
loc_or_str: Union[Tuple[str, str], str],
mode: str = 'single',
timeout: float = None):
"""实现查找元素的简化写法 \n
例：ele2 = ele1('@id=ele_id') \n
:param loc_or_str: 元素的定位信息，可以是loc元组，或查询字符串
:param mode: 'single' 或 'all'，对应查找一个或全部
:param timeout: 超时时间
:return: DriverElement对象
"""
return self.ele(loc_or_str, mode, timeout or self.timeout)
@property @property
def driver(self): def driver(self):
"""返回控制元素的WebDriver对象""" """返回控制元素的WebDriver对象"""
@ -58,7 +71,7 @@ class ShadowRootElement(DrissionElement):
return self.parent_ele.ele(loc, timeout=0.1) return self.parent_ele.ele(loc, timeout=0.1)
def ele(self, def ele(self,
loc_or_str: Union[tuple, str], loc_or_str: Union[Tuple[str, str], str],
mode: str = 'single', mode: str = 'single',
timeout: float = None): timeout: float = None):
"""返回当前元素下级符合条件的子元素，默认返回第一个 \n """返回当前元素下级符合条件的子元素，默认返回第一个 \n
@ -101,7 +114,7 @@ class ShadowRootElement(DrissionElement):
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)
def eles(self, def eles(self,
loc_or_str: Union[tuple, str], loc_or_str: Union[Tuple[str, str], str],
timeout: float = None): timeout: float = None):
"""返回当前元素下级所有符合条件的子元素 \n """返回当前元素下级所有符合条件的子元素 \n
示例： \n 示例： \n