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

修复获取同域iframe出错问题

Browse Source
This commit is contained in:
g1879 2022-11-22 15:25:20 +08:00
parent b9bfa1a1bc
commit 74caffc796
1 changed files with 7 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
DrissionPage/chromium_element.py
View File

@ -11,6 +11,7 @@ from pathlib import Path
from re import search from re import search
from typing import Union, Tuple, List, Any from typing import Union, Tuple, List, Any
from time import perf_counter, sleep from time import perf_counter, sleep
from urllib.parse import urlparse
from pychrome import Tab from pychrome import Tab
from requests import Session from requests import Session
@ -1467,6 +1468,11 @@ class ChromiumBase(BasePage):
for i in nodeIds['nodeIds']: for i in nodeIds['nodeIds']:
ele = ChromiumElement(self, node_id=i) ele = ChromiumElement(self, node_id=i)
if ele.tag in ('iframe', 'frame'): if ele.tag in ('iframe', 'frame'):
src = ele.attr('src')
if src:
netloc1 = urlparse(src).netloc
netloc2 = urlparse(self.url).netloc
if netloc1 != netloc2:
ele = ChromiumFrame(self, ele) ele = ChromiumFrame(self, ele)
eles.append(ele) eles.append(ele)