q792602257/Douyin_Web_Live
Archived
0
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

op_tab快捷方法

Browse Source
This commit is contained in:
Jerry Yan 2022-06-06 15:07:07 +08:00
parent d57287507a
commit 218082fed6
3 changed files with 42 additions and 18 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
browser/IDriver.py
View File

@ -1,12 +1,32 @@
import contextlib
class IDriver(): class IDriver():
def new_tab(self) -> str: def new_tab(self) -> str:
... ...
def get_current_tab(self) -> str:
...
def change_tab(self, tab_handler: str): def change_tab(self, tab_handler: str):
... ...
def open_url(self, url: str, tab_handler: str = ""): def open_url(self, url: str, tab_handler: str = ""):
... ...
@contextlib.contextmanager
def op_tab(self, tab_handler: str):
cur_handle = self.get_current_tab()
if tab_handler == "":
tab_handler = cur_handle
try:
self.change_tab(tab_handler)
yield self
finally:
self.change_tab(cur_handle)
def refresh(self, tab_handler: str = ""):
...
def screenshot(self, tab_handler: str = "") -> str: def screenshot(self, tab_handler: str = "") -> str:
... ...

18
browser/chrome.py
View File

@ -45,15 +45,17 @@ class ChromeDriver(IDriver):
return return
self.browser.switch_to.window(tab_handler) self.browser.switch_to.window(tab_handler)
def get_current_tab(self) -> str:
return self.browser.current_window_handle
def open_url(self, url: str, tab_handler: str = ""): def open_url(self, url: str, tab_handler: str = ""):
if tab_handler != "": with self.op_tab(tab_handler):
self.change_tab(tab_handler)
self.browser.get(url) self.browser.get(url)
def refresh(self, tab_handler: str = ""):
with self.op_tab(tab_handler):
self.browser.refresh()
def screenshot(self, tab_handler: str = "") -> str: def screenshot(self, tab_handler: str = "") -> str:
cur_handle = self.browser.current_window_handle with self.op_tab(tab_handler):
if tab_handler != "": return self.browser.get_screenshot_as_base64()
self.change_tab(tab_handler)
b64 = self.browser.get_screenshot_as_base64()
self.change_tab(cur_handle)
return b64

18
browser/edge.py
View File

@ -38,6 +38,9 @@ class EdgeDriver(IDriver):
return _handle return _handle
return "" return ""
def get_current_tab(self) -> str:
return self.browser.current_window_handle
def change_tab(self, tab_handler: str): def change_tab(self, tab_handler: str):
if tab_handler not in self.browser.window_handles: if tab_handler not in self.browser.window_handles:
return return
@ -46,14 +49,13 @@ class EdgeDriver(IDriver):
self.browser.switch_to.window(tab_handler) self.browser.switch_to.window(tab_handler)
def open_url(self, url: str, tab_handler: str = ""): def open_url(self, url: str, tab_handler: str = ""):
if tab_handler != "": with self.op_tab(tab_handler):
self.change_tab(tab_handler)
self.browser.get(url) self.browser.get(url)
def refresh(self, tab_handler: str = ""):
with self.op_tab(tab_handler):
self.browser.refresh()
def screenshot(self, tab_handler: str = "") -> str: def screenshot(self, tab_handler: str = "") -> str:
cur_handle = self.browser.current_window_handle with self.op_tab(tab_handler):
if tab_handler != "": return self.browser.get_screenshot_as_base64()
self.change_tab(tab_handler)
b64 = self.browser.get_screenshot_as_base64()
self.change_tab(cur_handle)
return b64