Python 實現網頁自動截圖的示例講解

背景介紹

最近在為部門編寫一個自動化測試工具，工具涉及到一個功能，即 將自動化測試生成的html報告截圖，作為郵件正文，html文件上傳到web服務器以鏈接形式添加到郵件中，最后發送郵件。

任務難點

之前從未接觸過頁面自動截圖相關的方面，因此如何自動進行頁面截圖成為本地調研方向。

方案思考

在剛接到這個任務時，并不認同目前的方案。曾經一度認為，將html報告的內容寫入郵件正文，即可通過html的形式發送郵件了。經過嘗試后發現，郵件不支持帶javascript的html。因此，選擇了預覽html并自動截圖的方案。

編程語言

python 2.7

使用方案 :

selenium + phantomjs

方案介紹

實現自動預覽html并截圖有以下幾個步驟：

1. 瀏覽器打開html

2. 對瀏覽器頁面進行截圖

3. 將截圖保存到指定位置

最初，筆者使用了這個實現方案：

(1).使用webbrowser 庫打開默認瀏覽器，顯示url

(2).使用PIL.ImageGrab庫的對屏幕進行進行截圖

到此為止，筆者已經獲得html的預覽截圖，一切似乎很順利，但接下來

發現了以下幾個問題：

(1).打開默認瀏覽器時，默認瀏覽器是什么，瀏覽器處于什么狀態都不知曉。

(2).瀏覽器顯示html，會有一個顯示的打開瀏覽器，瀏覽器打開html的動作出現，如果用戶此時產生了其他動作，必定會對之后的截圖產生影響。

(3).屏幕截圖，將整個屏幕都進行截圖，截圖中除了html頁面正文，同時也包含了瀏覽器等當前屏幕包含的所有元素，暴漏了用戶隱私。

(4).如果頁面較大，頁面會產生上下翻頁的效果，屏幕截圖無法應對此種頁面，截圖看到的報告只有一部分。

基于上述問題，筆者放棄了這個看起來簡單有效的方案。從而

對方案挖掘了一些深層需求：

(1).打開瀏覽器必須隱式調用，用戶無法看到工具在做什么，從而不會通過誤操作對工具產生影響。

(2).截圖必須針對瀏覽器頁面正文 ，保證無其他冗余信息，且截到頁面全圖根據這些需求，筆者經過一系列嘗試終于發現了selenium + phantomjs。

python selenium : 是python對瀏覽器自動化測試框架selenium的擴展，selenium庫支持了selenium所包含大部分功能。

phantomjs ： 是一個無界面的，可腳本編程的webkit瀏覽器，python selenium同樣提供對phantomjs的支持。

接下來一切就很容易了：

(1).安裝python selenium 庫，推薦使用pip快速安裝最新版本

(2).下載phantomjs.exe，將其添加到環境變量 path 中 。方便起見可以直接放在python的安裝目錄根目錄下

(3).編寫測試代碼

簡單講解一下代碼：

fromselenium importwebdriver     #從selenium庫導入webdirver

brower=webdriver.PhantomJS()    #使用webdirver.PhantomJS()方法新建一個phantomjs的對象，這里會使用到phantomjs.exe，環境變量path中找不到phantomjs.exe，則會報錯

brower.get(url)           #使用get()方法，打開指定頁面。注意這里是phantomjs是無界面的，所以不會有任何頁面顯示

brower.maximize_window()      #設置phantomjs瀏覽器全屏顯示

brower.save_screenshot(picName)   #使用save_screenshot將瀏覽器正文部分截圖，即使正文本分無法一頁顯示完全，save_screenshot也可以完全截圖

brower.close()           #關閉phantomjs瀏覽器，不要忽略了這一步，否則你會在任務瀏覽器中發現許多phantomjs進程

執行了上述代碼后，我們看一下截圖效果吧：

以上這篇Python 實現網頁自動截圖的示例講解就是小編分享給大家的全部內容了，希望能給大家一個參考，也希望大家多多支持億速云。