Python Selenium — 封裝瀏覽器引擎類
阿新 • • 發佈:2021-10-17
一、前言
記錄下封裝瀏覽器引擎類的寫法,本篇是參考以下2篇部落格:
- https://blog.csdn.net/u011541946/article/details/70171401
- https://www.cnblogs.com/yudx/p/11864786.html
封裝瀏覽器引擎類後,可以通過更改一個字串的值,利用if語句去判斷和控制啟動那個瀏覽器。這裡我們暫時,支援兩大瀏覽器(Chrome,Firefox)。
這裡有一個前提條件,在基礎篇中,啟動三大瀏覽器的driver檔案,檢查下你的Python安裝路徑下有沒有這三個driver外掛,如果沒有,自行百度下。
二、封裝瀏覽器引擎
在CommonMethod目錄下新建browser_engine.py檔案,程式碼如下:
from selenium import webdriver class BrowserEngine(object): '''定義一個瀏覽器引擎,根據browser_type的值去,控制啟動不同的瀏覽器,這裡主要是firefox,chrome''' def __init__(self, driver): self.driver = driver browser_type = "Chrome" def get_browser(self): '''通過if語句,來控制初始化不同瀏覽器的啟動,預設是啟動firefox''' ifself.browser_type == "Firefox": driver = webdriver.Firefox() elif self.browser_type == "Chrome": driver = webdriver.Chrome() else: driver = webdriver.Firefox() driver.maximize_window() driver.implicitly_wait(10) return driver
三、測試呼叫
在testcase目錄下新建test.py檔案,進行測試呼叫。
from CommonMethod.browser_engine import BrowserEngine class TestB(object): def open_browser(self): browserengine = BrowserEngine(self) self.driver = browserengine.get_browser() self.driver.get("https://blog.csdn.net") t = TestB() t.open_browser()
四、總結
目前,自定義的瀏覽器引擎類到這裡就封裝好了,如需要開啟不同瀏覽器,只需要手動修改引擎類中browser_type的值。