Selenium中級篇之4-封裝一個自己的類-瀏覽器引擎類

前一篇文章我們知道了，如何去封裝幾個簡單的Selenium方法到我們自定義的類，這次我們編寫一個類，叫瀏覽器引擎類，通過更改一個字元串的值，利用if語句去判斷和控制啟動那個瀏覽器。這裡我們暫時，支持三大瀏覽器（IE,Chrome,Firefox）。這裡有一個前提條件，在基礎篇中，啟動三大瀏覽器的driver文件，檢查下你的Python安裝路徑下有沒有這三個driver插件，如果沒有，請回到基礎篇的如何啟動火狐和IE瀏覽器文章去看看如何做。

我們繼續在test1這個包下新建一個browser_engine.py文件，然後在另外一個包下新建一個test.py文件去測試這個瀏覽器引擎類是否工作正常。這個瀏覽器引擎類，我們一開始寫簡單一點，只寫啟動瀏覽器。

先看看browser_engine.py中的代碼：

# coding=utf-8

from selenium import webdriver

class BrowserEngine(object):

"""

定義一個瀏覽器引擎類，根據browser_type的值去，控制啟動不同的瀏覽器，這裡主要是IE，Firefox, Chrome

"""

def __init__(self, driver):

self.driver = driver

browser_type = "IE" # maybe Firefox, Chrome, IE

def get_browser(self):

"""

通過if語句，來控制初始化不同瀏覽器的啟動，默認是啟動Chrome

:return: driver

"""

if self.browser_type == "Firefox":

driver = webdriver.Firefox()

elif self.browser_type == "Chrome":

driver = webdriver.Chrome()

elif self.browser_type == "IE":

driver = webdriver.Ie()

else: driver = webdriver.Chrome()

driver.maximize_window()

driver.implicitly_wait(10)

return driver

再看看test.py代碼，進行測試，更改browser_engine.py中browser_type的值，去測試三大瀏覽器是否啟動正常。

# coding=utf-8

import time

from test1.browser_engine import BrowserEngine

class TestBrowserEngine(object):

def open_browser(self):

browserengine = BrowserEngine(self)

driver = browserengine.get_browser()

tbe = TestBrowserEngine()

tbe.open_browser()

目前，自定義的瀏覽器引擎類到這裡就封裝好了，只支持打開不同瀏覽器，需要手動修改，引擎類中browser_type的值。看起來功能簡單，但是我們只是需要學習這種做事的方式和思維，在下一個部分，框架設計的時候，我會再告訴大家如何去加強這個引擎類的功能，到時候去修改配置文件中的瀏覽器類型，而不是修改代碼中的欄位。通過修改配置文件，從而去打開不同瀏覽器，並開始測試相關腳本。

喜歡這篇文章嗎？立刻分享出去讓更多人知道吧！