Python对Selenium调用浏览器

• 2020 年 1 月 16 日
• 筆記

""" 获取浏览器 打开本地浏览器 打开远程浏览器 关闭浏览器 打开网址 最大化 最小化 标题 url 刷新

Python对Selenium封装浏览器调用

——browser.py文件 """

from selenium import webdriver    from Common.tools.rw_ini import read_config  from .base import Base      class Browser(Base):      def get_web_driver(self) -> webdriver.Remote:          """获取浏览器实例            :return: 浏览器实例          """          rc = read_config("browser.ini")          local_browser = rc.getboolean("local", "local_browser", fallback=True)          wait_time = rc.getint("local", "wait_time", fallback=10)            browser_name = rc.get("browser", "name", fallback="chrome")          window_width = rc.get("browser", "window_width", fallback=None)          window_height = rc.get("browser", "window_height", fallback=None)            command_executor = rc.get("grid", "command_executor", fallback="http://127.0.0.1:4444/wd/hub")          headless = rc.getboolean("grid", "headless", fallback=False)          proxy = rc.get("grid", "proxy", fallback=None)            if local_browser:              # 打开本地浏览器              driver = self._get_local_web_driver(browser_name)          else:              # 打开远程浏览器              driver = self._get_remote_web_driver(browser_name, command_executor, headless, proxy)          self.logger.info(f"打开浏览器：{driver.name}")            self.logger.info(f"设置隐式等待：{wait_time}s")          driver.implicitly_wait(wait_time)            if window_width is None or window_width == "" or window_height is None or window_height == "":              self.logger.info("最大化浏览器")              driver.maximize_window()          else:              driver.set_window_size(window_width, window_height)          return driver        def _get_local_web_driver(self, browser_name: str) -> webdriver.Remote:          """获取本地浏览器实例            :param browser_name: 浏览器类型          :return: 浏览器实例          """          if browser_name.upper() == "CHROME":              driver = webdriver.Chrome()          elif browser_name.upper() == "FIREFOX":              driver = webdriver.Firefox()          elif browser_name.upper() == "IE":              driver = webdriver.Ie()          else:              self.logger.error(f"浏览器类型错误：{browser_name}")              raise ValueError(f"浏览器类型错误：{browser_name}")          return driver        def _get_remote_web_driver(self, browser_name, command_executor, headless, proxy) -> webdriver.Remote:          """获取远程浏览器实例            :param browser_name: 浏览器类型          :param command_executor: HUB 地址          :param headless: 使用无头浏览器          :param proxy: 代理          :return: 浏览器实例          """          if browser_name.upper() == "CHROME":              options = webdriver.ChromeOptions()          elif browser_name.upper() == "FIREFOX":              options = webdriver.FirefoxOptions()          elif browser_name.upper() == "IE":              options = webdriver.IeOptions()          else:              self.logger.error(f"浏览器类型错误：{browser_name}")              raise ValueError(f"浏览器类型错误：{browser_name}")            if proxy is not None:              options.add_argument(f"--proxy-server={proxy}")          # 无头浏览器          options.headless = headless            driver = webdriver.Remote(              command_executor=command_executor,              options=options          )          return driver        def quit(self):          """退出浏览器"""          self.logger.info(f"关闭浏览器：{self.driver.name}")          self.driver.quit()        def get(self, url: str, new: bool = False):          """打开 url            :param url: 网址          :param new: 是否新窗口打开          :return: None          """          if new:              self.logger.info(f"新窗口打开 url：{url}")              self.driver.execute_script(f"window.open('{url}')")          else:              self.logger.info(f"当前窗口打开 url：{url}")              self.driver.get(url)        def get_session_id(self):          """返回当前浏览器 session id"""          session_id = self.driver.session_id          self.logger.info(f"当前浏览器 session_id：{session_id}")          return session_id        def get_source(self) -> str:          """获取当前页面HTML源代码"""          page_source = self.driver.page_source          self.logger.info(f"当前页面源代码：n{page_source}")          return page_source        def get_title(self):          """返回当前页面的标题"""          title = self.driver.title          self.logger.info(f"当前页面的标题：{title}")          return title        def get_current_url(self):          """返回当前页面的URL"""          current_url = self.driver.current_url          self.logger.info(f"当前页面URL：{current_url}")          return current_url        def reload_page(self):          """模拟用户重新加载页面"""          self.logger.info("刷新页面")          self.driver.refresh()

浏览器配置文件：

——browser.ini文件

[local]  local_browser = False  wait_time = 10    [browser]  name = chrome  ;name = firefox  ;name = ie  window_width =  window_height =    [grid]  command_executor = http://127.0.0.1:4444/wd/hub  headless = False  proxy =

rw_ini.py读取配置文件请查看下面链接： https://www.cnblogs.com/CesareZhang/p/10738541.html