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

修改关闭标签页功能，未完成

Browse Source
This commit is contained in:
g1879 2022-01-07 18:15:20 +08:00
parent 0172757ccf
commit 3af48126ea
1 changed files with 27 additions and 16 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

43
DrissionPage/driver_page.py
View File

@ -303,31 +303,28 @@ class DriverPage(BasePage):
:param num_or_handles: 要保留的标签页序号或handle，可传入handle组成的列表或元组
:return: None
"""
try:
tab = int(num_or_handles)
except (ValueError, TypeError):
tab = num_or_handles
tabs = self.driver.window_handles
if tab is None:
page_handle = (self.current_tab_handle,)
elif isinstance(tab, int):
page_handle = (tabs[tab],)
elif isinstance(tab, str):
page_handle = (tab,)
elif isinstance(tab, (list, tuple)):
page_handle = tab
if num_or_handles is None:
reserve_tabs = (self.current_tab_handle,)
elif isinstance(num_or_handles, (int, str)):
reserve_tabs = (num_or_handles,)
elif isinstance(num_or_handles, (list, tuple)):
reserve_tabs = num_or_handles
else:
raise TypeError('num_or_handle参数只能是int、str、list 或 tuple类型。')
tabs = self.driver.window_handles
reserve_tabs = set(i if isinstance(i, str) else tabs[i] for i in reserve_tabs)
for i in tabs: # 遍历所有标签页，关闭非保留的
if i not in page_handle:
if i not in reserve_tabs:
self.driver.switch_to.window(i)
self.driver.close()
self.to_tab(0)
def close_tab(self, num_or_handles: Union[int, str, list, tuple] = None):
pass
def to_tab(self, num_or_handle: Union[int, str] = 0) -> None:
"""跳转到标签页 \n
:param num_or_handle: 标签页序号或handle字符串，序号第一个为0，最后为-1
@ -525,3 +522,17 @@ class DriverPage(BasePage):
alert.accept()
return text
def _get_handles(page: DriverPage, num_or_handles: Union[int, str, list, tuple] = None):
if num_or_handles is None:
reserve_tabs = (page.current_tab_handle,)
elif isinstance(num_or_handles, (int, str)):
reserve_tabs = (num_or_handles,)
elif isinstance(num_or_handles, (list, tuple)):
reserve_tabs = num_or_handles
else:
raise TypeError('num_or_handle参数只能是int、str、list 或 tuple类型。')
tabs = page.driver.window_handles
return set(i if isinstance(i, str) else tabs[i] for i in reserve_tabs)