使用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_()) |
SRE实战 互联网时代守护先锋，助力企业售后服务体系运筹帷幄！一键直达领取阿里云限量特价优惠。
方法二、使用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() |
更多精彩
