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

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

方法一、使用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"网页加载失败！"

									    self.close()

									if __name__=="__main__":

									  app = QtGui.QApplication(sys.argv)

									  shotter = PageShotter("http://www.zzvips.com/", '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位操作系统安装包
64位操作系统下对应PyQt安装包

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

1	`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.zzvips.com/")

									browser.find_element_by_id("idClose").click()

									time.sleep(5)

									browser.save_screenshot("shot.png")

									browser.quit()

PS：selenium库同样可以使用pip命令安装：

1	`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程序设计有所帮助。

原文链接：https://blog.csdn.net/wang1472jian1110/article/details/72642605

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

相关文章

热门资讯