55 lines
2.6 KiB
Python
55 lines
2.6 KiB
Python
import logging
|
|
from selenium import webdriver
|
|
from selenium.webdriver import Proxy, DesiredCapabilities
|
|
from selenium.webdriver.common.proxy import ProxyType
|
|
|
|
from browser.IDriver import IDriver
|
|
from selenium.webdriver.chrome.options import Options
|
|
|
|
_log = logging.getLogger("ChromeDriver")
|
|
_log.setLevel(logging.DEBUG)
|
|
|
|
class ChromeDriver(IDriver):
|
|
def __init__(self, config_manager):
|
|
super(ChromeDriver, self).__init__(config_manager)
|
|
options = Options()
|
|
if self._config_manager.config['webdriver']['headless']:
|
|
options.add_argument("--headless")
|
|
options.add_argument("--window-size=1920,1080")
|
|
options.add_argument('--proxy-server=%s:%s' % (self._config_manager.config['mitm']['host'], self._config_manager.config['mitm']['port']))
|
|
options.add_argument('--ignore-certificate-errors')
|
|
options.add_argument('--ignore-ssl-errors')
|
|
options.add_argument('--incognito')
|
|
options.add_experimental_option('excludeSwitches', ['ignore-certificate-errors'])
|
|
if self._config_manager.config['webdriver']['chrome']['no_sandbox']:
|
|
_log.debug("添加启动参数NoSandbox")
|
|
options.add_argument('--no-sandbox')
|
|
proxy = Proxy()
|
|
proxy.proxy_type = ProxyType.MANUAL
|
|
proxy.http_proxy = "%s:%s" % (self._config_manager.config['mitm']['host'], self._config_manager.config['mitm']['port'])
|
|
proxy.ssl_proxy = "%s:%s" % (self._config_manager.config['mitm']['host'], self._config_manager.config['mitm']['port'])
|
|
capabilities = DesiredCapabilities.CHROME
|
|
proxy.add_to_capabilities(capabilities)
|
|
|
|
self.browser = webdriver.Chrome(options=options,
|
|
desired_capabilities=capabilities,
|
|
executable_path=self._config_manager.config['webdriver']['chrome']['bin']
|
|
)
|
|
_log.info("浏览器启动完毕")
|
|
|
|
def new_tab(self) -> str:
|
|
current_window_handles = self.browser.window_handles
|
|
self.browser.execute_script("window.open('')")
|
|
new_window_handles = self.browser.window_handles
|
|
for _handle in new_window_handles:
|
|
if _handle not in current_window_handles:
|
|
_log.debug("新窗口句柄：%s", _handle)
|
|
return _handle
|
|
_log.warning("打开新窗口，未发现新句柄")
|
|
return ""
|
|
|
|
def change_tab(self, tab_handler: str):
|
|
if tab_handler not in self.browser.window_handles:
|
|
return
|
|
self.browser.switch_to.window(tab_handler)
|