1. 程式人生 > 其它 >Python Selenium — 封裝瀏覽器引擎類

Python Selenium — 封裝瀏覽器引擎類

一、前言

  記錄下封裝瀏覽器引擎類的寫法,本篇是參考以下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'''
        if
self.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的值。