使用python对整个网页进行截图

Posted 2020-11-24 hello2020888

方法一、使用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()