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


   前一篇文章我們知道了,如何去封裝幾個簡單的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的值。看起來功能簡單,但是我們只是需要學習這種做事的方式和思維,在下一個部分,框架設計的時候,我會再告訴大家如何去加強這個引擎類的功能,到時候去修改配置文件中的瀏覽器類型,而不是修改代碼中的字段。通過修改配置文件,從而去打開不同瀏覽器,並開始測試相關腳本。

 


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM