Python實現(xiàn)的網(wǎng)頁截圖功能【PyQt4與selenium組件】

更新時間：2018年07月12日 08:40:50 作者：wang1472jian1110

這篇文章主要介紹了Python實現(xiàn)的網(wǎng)頁截圖功能,結(jié)合實例形式分別描述了使用PyQt4組件與selenium組件進行網(wǎng)頁截圖操作的相關(guān)實現(xiàn)技巧與注意事項,需要的朋友可以參考下

本文實例講述了Python實現(xiàn)的網(wǎng)頁截圖功能。分享給大家供大家參考，具體如下：

方法一、使用PyQt4的QtWebKit組件

#!/usr/bin/env python
# -*- coding: UTF-8 -*-
import sys
import os.path
from PyQt4 import QtGui,QtCore,QtWebKit
class PageShotter(QtGui.QWidget):
  def __init__(self,url,filename,parent=None):
    QtGui.QWidget.__init__(self,parent)
    self.url = url
    self.filename = filename
    self.webpage = None
  def shot(self):
    webview = QtWebKit.QWebView(self)
    webview.load(QtCore.QUrl(self.url))
    self.webpage = webview.page()
    self.connect(webview,QtCore.SIGNAL("loadFinished(bool)"),self.save_page)
  def save_page(self,finished):
    #print finished
    if finished:
      print u"開始截圖！"
      size = self.webpage.mainFrame().contentsSize()
      print u"頁面寬：%d，頁面高：%d" % (size.width(),size.height())
      self.webpage.setViewportSize(QtCore.QSize(size.width()+16,size.height()))
      img = QtGui.QImage(size, QtGui.QImage.Format_ARGB32)
      painter = QtGui.QPainter(img)
      self.webpage.mainFrame().render(painter)
      painter.end()
      filename= self.filename;
      if img.save(filename):
        filepath = os.path.join(os.path.dirname(__file__), filename)
        print u"截圖完畢：%s" % filepath
      else:
        print u"截圖失敗";
    else:
      print u"網(wǎng)頁加載失敗！"
    self.close()
if __name__=="__main__":
  app = QtGui.QApplication(sys.argv)
  shotter = PageShotter("http://www.dbjr.com.cn/", 'shot.png')
  shotter.shot()
  sys.exit(app.exec_())

運行后輸出：

QFont::setPixelSize: Pixel size <= 0 (0)
開始截圖！
頁面寬：1058，頁面高：9819
截圖完畢：C:\py\jb51PyDemo\src\Demo\shot.png

注：

Python2的32位操作系統(tǒng)安裝包可至此下載：http://www.dbjr.com.cn/softs/548192.html
64位操作系統(tǒng)下對應PyQt安裝包可至此下載：http://www.dbjr.com.cn/softs/548197.html）

Python3可直接使用pip命令安裝PyQt5庫，如：

pip3 install PyQt5-sip

方法二、使用selenium

#!/usr/bin/env python
# -*- coding: UTF-8 -*-
import time
from selenium import webdriver
browser = webdriver.Firefox()
browser.set_window_size(1055, 800)
browser.get("http://www.dbjr.com.cn/")
browser.find_element_by_id("idClose").click()
time.sleep(5)
browser.save_screenshot("shot.png")
browser.quit()

PS：selenium庫同樣可以使用pip命令安裝：

pip install selenium

另外，使用selenium時還需要下載geckodriver來驅(qū)動第三方瀏覽器，對于selenium3.x版本都會使用geckodriver來驅(qū)動firefox，所以需要下載geckodriver.exe,下載地址：https://github.com/mozilla/geckodriver/releases

下載后將geckodriver.exe放在C:\Python27即可（查看環(huán)境變量path中是否添加C:\Python27該路徑）

更多關(guān)于Python相關(guān)內(nèi)容感興趣的讀者可查看本站專題：《Python圖片操作技巧總結(jié)》、《Python數(shù)據(jù)結(jié)構(gòu)與算法教程》、《Python函數(shù)使用技巧總結(jié)》、《Python字符串操作技巧匯總》、《Python入門與進階經(jīng)典教程》及《Python文件與目錄操作技巧匯總》

希望本文所述對大家Python程序設(shè)計有所幫助。

您可能感興趣的文章: