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

attr('href')增加处理javascript、mailto、#功能

Browse Source
This commit is contained in:
g1879 2020-06-02 23:11:19 +08:00
parent 7d02f81d7e
commit 6d5b9b019d
1 changed files with 9 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
DrissionPage/session_element.py
View File

@ -106,12 +106,20 @@ class SessionElement(DrissionElement):
if attr == 'href': if attr == 'href':
# 如直接获取attr只能获取相对地址 # 如直接获取attr只能获取相对地址
link = self._inner_ele.attrs['href'] link = self._inner_ele.attrs['href']
if link.startswith('?'): # 避免当相对URL以?开头时requests-html丢失参数的bug if link.startswith(('javascript:', 'mailto:')):
return link
elif link.startswith('#'):
if '#' in self.inner_ele.url:
return re.sub(r'#.*', link, self.inner_ele.url)
else:
return f'{self.inner_ele.url}{link}'
elif link.startswith('?'): # 避免当相对URL以?开头时requests-html丢失参数的bug
if '?' in self.inner_ele.url: if '?' in self.inner_ele.url:
return re.sub(r'\?.*', link, self.inner_ele.url) return re.sub(r'\?.*', link, self.inner_ele.url)
else: else:
return f'{self.inner_ele.url}{link}' return f'{self.inner_ele.url}{link}'
else: else:
self._inner_ele.skip_anchors = False
for link in self._inner_ele.absolute_links: for link in self._inner_ele.absolute_links:
return link return link
elif attr == 'src': elif attr == 'src':