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

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

先看看browser_engine.py中的代碼：

# coding=utf-8

from selenium import webdriver

class BrowserEngine(object):

'''

定義一個瀏覽器引擎類，根據(jù)browser_type的值去，控制啟動不同的瀏覽器，這里主要是IE，F(xiàn)irefox, 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的值。看起來功能簡單，但是我們只是需要學習這種做事的方式和思維，在下一個部分，框架設計的時候，我會再告訴大家如何去加強這個引擎類的功能，到時候去修改配置文件中的瀏覽器類型，而不是修改代碼中的字段。通過修改配置文件，從而去打開不同瀏覽器，并開始測試相關(guān)腳本。