用Python对网页进行截图

方法一、使用PyQt4的QtWebKit组件

?

1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48

#!/usr/bin/env python # -*- coding: UTF-8 -*- # 来源 http://www.oschina.net/code/snippet_219811_14920   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.adssfwewfdsfdsf.com")      shotter = PageShotter( "http://www.youku.com/" , 'shot.png' )      shotter.shot()      sys.exit(app.exec_())

方法二、使用selenium

?

1 2 3 4 5 6 7 8 9 10 11 12 13 14

#!/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.yooli.com/" ) browser.find_element_by_id( "idClose" ).click() time.sleep( 5 )   browser.save_screenshot( "shot.png" ) browser.quit()

网友评论，共0条