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

通用方法不下放，提取部分方法

Browse Source
This commit is contained in:
Jerry Yan
2022-07-15 16:00:10 +08:00
parent 134cf0186a
commit d0e067ae44
3 changed files with 39 additions and 22 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
browser/IDriver.py
View File

@ -27,10 +27,12 @@ class IDriver():
...
def close_tab(self, tab_handler: str):
...
with self.op_tab(tab_handler):
self.browser.close()
def open_url(self, url: str, tab_handler: str = ""):
...
with self.op_tab(tab_handler):
self.browser.get(url)
@contextlib.contextmanager
def op_tab(self, tab_handler: str):
@ -47,7 +49,9 @@ class IDriver():
_log.debug("切换至旧Tab：%s", cur_handle)
def refresh(self, tab_handler: str = ""):
...
with self.op_tab(tab_handler):
self.browser.refresh()
def screenshot(self, tab_handler: str = "") -> str:
...
with self.op_tab(tab_handler):
return self.browser.get_screenshot_as_base64()