推薦一款新的自動化測試框架:DrissionPage!

今天給大家推薦一款基於Python的網頁自動化工具:DrissionPage。這款工具既能控制瀏覽器,也能收發數據包,甚至能把兩者合而爲一,簡單來說:集合了WEB瀏覽器自動化的便利性和 requests 的高效率。

一、DrissionPage產生背景

實現網頁自動化,會有兩類形式:

  • 直接向服務器發送請求數據包,獲取需要的數據
  • 模擬真實用戶操作行爲,控制瀏覽器跟網頁進行交互

前者輕量級,速度快,例如requests 庫。但requests面對需要登錄的網站時,往往還要應付驗證碼、JS 混淆、簽名參數等反爬手段,門檻較高。若數據是由 JS 計算生成的,還須重現計算過程,開發效率不高。

而後者直接使用瀏覽器,模擬用戶行爲,如Selenium庫,可以很大程度上繞過這些坑,但瀏覽器運行效率不高。

因此,DrissionPag設計初衷,是將它們合而爲一,能夠在不同須要時切換相應模式,並提供一種人性化的使用方法,提高開發和運行效率。

爲什麼叫DrissionPag

Selenium框架用於操作瀏覽器的對象叫 Driver,requests 用於管理請求連接的對象叫 Session,因此Drission 就是它們兩者的合體。在舊版本中,是通過對 selenium 和 requests 的重新封裝實現的。

但從 3.0 版開始,作者另起爐竈,用 chromium 協議自行實現了 selenium 全部功能,從而擺脫了對 selenium 的依賴,功能更多更強,運行效率更高,開發更靈活。

二、DrissionPage亮點特色

本庫採用全自研的內核,對比 selenium,有以下優點:

  • 無 webdriver 特徵,不會被網站識別,無需爲不同版本的瀏覽器下載不同的驅動
  • 運行速度更快,可以跨 iframe 查找元素,無需切入切出,iframe 看作普通元素,獲取後可直接在其中查找元素,邏輯更清晰
  • 可以同時操作瀏覽器中的多個標籤頁,即使標籤頁爲非激活狀態,無需切換
  • 內置等待和自動重試功能。使不穩定的網絡變得易於控制,程序更穩定,編寫更省心
  • 允許反覆使用已經打開的瀏覽器。無須每次運行從頭啓動瀏覽器,調試超方便
  • 極簡的語法規則。集成大量常用功能,定位元素更加容易,功能更強大穩定
  • 使用 POM 模式封裝,可直接用於測試,便於擴展
  • 等等。。。

三、安裝搭建

說了這麼多,相信很多人已經躍躍欲試了,怎麼快速搭建這套框架,先要準備一些基礎環境。

環境準備

  • 操作系統:Windows、Linux 或 Mac。
  • python 版本:3.6 及以上
  • 支持瀏覽器:Chromium 內核(如 Chrome 和 Edge)

項目地址:

https://gitee.com/g1879/DrissionPage

使用 pip 安裝 DrissionPage:

pip install DrissionPage

另外在開始之前,我們先進行一些簡單設置。如果只使用收發數據包功能,無須任何準備工作。

如果要控制瀏覽器,須設置瀏覽器路徑。程序默認設置控制 Chrome,所以以下用 Chrome 作爲演示。如果要使用 Edge 或其它 Chromium 內核瀏覽器,設置方法是一樣的。

設置瀏覽器路徑:

from DrissionPage.easy_set import set_paths
set_paths(browser_path=r'這裏修改爲您的Chrome瀏覽器exe文件路徑')

這段代碼會記錄 Chrome 瀏覽器路徑到配置文件。由於路徑設置只要運行一次即可,不要寫到正式程序裏。一般建議新建一個臨時 py 文件,並輸入以下代碼,填入您電腦裏的 Chrome 瀏覽器 exe 文件路徑,然後執行。

四、實戰一下

1、嘗試啓動瀏覽器:

from DrissionPage import ChromiumPage

page = ChromiumPage()
page.get('https://www.baidu.com')

執行以下代碼,如果正常啓動了瀏覽器並且訪問了百度,說明可直接使用,跳過後面的步驟即可。

執行代碼前,如果已有打開的 Chrome 瀏覽器,請關閉。

2、與selenium框架代碼對比

#案例一:用顯性等待方式查找第一個文本包含 some text 的元素。
# 使用 selenium:
element = WebDriverWait(driver).until(ec.presence_of_element_located((By.XPATH, '//*[contains(text(), "some text")]')))

# 使用 DrissionPage:
element = page('some text')

#案例二:跳轉到第一個標籤頁
# 使用 selenium:
driver.switch_to.window(driver.window_handles[0])

# 使用 DrissionPage:
page.to_tab(0)

# 案例三:拖拽一個元素
# 使用 selenium:
ActionChains(driver).drag_and_drop(ele1, ele2).perform()

# 使用 DrissionPage:
ele1.drag_to(ele2)

以上代碼實現一模一樣的功能,但DrissionPage代碼明顯更簡潔優雅。

3、與requests框架代碼對比

# 案例一:獲取元素內容
url = 'https://baike.baidu.com/item/python'

# 使用 requests:
from lxml import etree
headers = {'User-Agent':'Mozilla/5.0 (Windows NT 6.3; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/41.0.2272.118 Safari/537.36'}
response = requests.get(url, headers = headers)
html = etree.HTML(response.text)
element = html.xpath('//h1')[0]
title = element.text

# 使用 DrissionPage:
page = WebPage('s')
page.get(url)
title = page('tag:h1').text

4、DrissionPage不同模式切換

例如:用瀏覽器登錄網站,然後切換到 requests 讀取網頁。兩者會共享登錄信息。

from DrissionPage import WebPage
from time import sleep

# 創建頁面對象,默認 d 模式
page = WebPage()  
# 訪問個人中心頁面(未登錄,重定向到登錄頁面)
page.get('https://gitee.com/profile')  

# 使用 selenium 輸入賬號密碼登錄
page.ele('@id:user_login').input('your_user_name')  
page.ele('@id:user_password').input('your_password\n')
sleep(1)

# 切換到 s 模式
page.change_mode()  
# 登錄後 session 模式的輸出
print('登錄後title:', page.title, '\n')  

五、小結

DrissionPage體驗一番後,雖然也發現了一些不足的地方,但整得來講,很多設計理念還是非常值得借鑑的,更多功能就留給大家去探索了,項目文檔地址:

http://g1879.gitee.io/drissionpagedocs/
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章