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

s模式xpath可获取上级及平级的元素，且优化nexts,parents,prevs函数

Browse Source
This commit is contained in:
g1879 2020-08-10 22:36:20 +08:00
parent 2696229a02
commit b065e7994f
1 changed files with 5 additions and 23 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

28
DrissionPage/session_element.py
View File

@ -69,35 +69,21 @@ class SessionElement(DrissionElement):
:param num: 第几级父元素 :param num: 第几级父元素
:return: SessionElement对象 :return: SessionElement对象
""" """
try: return self.ele(f'xpath:..{"/.." * (num - 1)}')
return SessionElement(
Element(element=self.inner_ele.element.xpath(f'..{"/.." * (num - 1)}')[0], url=self.inner_ele.url))
except IndexError:
return None
def nexts(self, num: int = 1): def nexts(self, num: int = 1):
"""返回后面第num个兄弟元素 \n """返回后面第num个兄弟元素 \n
:param num: 后面第几个兄弟元素 :param num: 后面第几个兄弟元素
:return: SessionElement对象 :return: SessionElement对象
""" """
try: return self.ele(f'xpath:./following-sibling::*[{num}]')
return SessionElement(
Element(element=self.inner_ele.element.xpath(f'./following-sibling::*[{num}]')[0],
url=self.inner_ele.url))
except IndexError:
return None
def prevs(self, num: int = 1): def prevs(self, num: int = 1):
"""返回前面第num个兄弟元素 \n """返回前面第num个兄弟元素 \n
:param num: 前面第几个兄弟元素 :param num: 前面第几个兄弟元素
:return: SessionElement对象 :return: SessionElement对象
""" """
try: return self.ele(f'xpath:./preceding-sibling::*[{num}]')
return SessionElement(
Element(element=self.inner_ele.element.xpath(f'./preceding-sibling::*[{num}]')[0],
url=self.inner_ele.url))
except IndexError:
return None
def ele(self, loc_or_str: Union[tuple, str], mode: str = None, show_errmsg: bool = False): def ele(self, loc_or_str: Union[tuple, str], mode: str = None, show_errmsg: bool = False):
"""返回当前元素下级符合条件的子元素，默认返回第一个 \n """返回当前元素下级符合条件的子元素，默认返回第一个 \n
@ -223,7 +209,6 @@ def execute_session_find(page_or_ele: BaseParser,
if mode not in ['single', 'all']: if mode not in ['single', 'all']:
raise ValueError("Argument mode can only be 'single' or 'all'.") raise ValueError("Argument mode can only be 'single' or 'all'.")
loc_by, loc_str = loc loc_by, loc_str = loc
result = None
try: try:
ele = None ele = None
if loc_by == 'xpath': if loc_by == 'xpath':
@ -236,13 +221,10 @@ def execute_session_find(page_or_ele: BaseParser,
ele = page_or_ele.find(loc_str) ele = page_or_ele.find(loc_str)
if mode == 'single': if mode == 'single':
result = SessionElement(ele[0]) if ele else None return SessionElement(ele[0]) if ele else None
elif mode == 'all': elif mode == 'all':
result = [SessionElement(e) for e in ele] return [SessionElement(e) for e in ele]
return result
except: except:
# raise
if show_errmsg: if show_errmsg:
print('Element(s) not found.', loc) print('Element(s) not found.', loc)
raise raise