Python实现的网页截图功能【PyQt4与selenium组件】

本文实例讲述了Python实现的网页截图功能。分享给大家供大家参考，具体如下：

方法一、使用PyQt4的QtWebKit组件

#!/usr/bin/env python# -*- coding: UTF-8 -*-import sysimport os.pathfrom PyQt4 import QtGui,QtCore,QtWebKitclass 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"网页加载失败！"    self.close()if __name__=="__main__":  app = QtGui.QApplication(sys.argv)  shotter = PageShotter("https://www.jb51.net/", '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位操作系统安装包可至此下载：https://www.jb51.net/softs/548192.html
64位操作系统下对应PyQt安装包可至此下载：https://www.jb51.net/softs/548197.html）

Python3可直接使用pip命令安装PyQt5库，如：

pip3 install PyQt5-sip

方法二、使用selenium

#!/usr/bin/env python# -*- coding: UTF-8 -*-import timefrom selenium import webdriverbrowser = webdriver.Firefox()browser.set_window_size(1055, 800)browser.get("https://www.jb51.net/")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来驱动第三方浏览器，对于selenium3.x版本都会使用geckodriver来驱动firefox，所以需要下载geckodriver.exe,下载地址：https://github.com/mozilla/geckodriver/releases

下载后将geckodriver.exe放在C:/Python27即可（查看环境变量path中是否添加C:/Python27该路径）

更多关于Python相关内容感兴趣的读者可查看本站专题：《Python图片操作技巧总结》、《Python数据结构与算法教程》、《Python函数使用技巧总结》、《Python字符串操作技巧汇总》、《Python入门与进阶经典教程》及《Python文件与目录操作技巧汇总》