用Python对网页进行截图

来源：互联网发布：spss软件下载中文版编辑：程序博客网时间：2024/05/13 15:25

方法一、使用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
 
importsys
importos.path
fromPyQt4 importQtGui,QtCore,QtWebKit
 
classPageShotter(QtGui.QWidget):
    def__init__(self,url,filename,parent=None):
        QtGui.QWidget.__init__(self,parent)
        self.url=url
        self.filename=filename
        self.webpage=None
 
    defshot(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)
 
    defsave_page(self,finished):
        #print finished
        iffinished:
            printu"开始截图！"
            size=self.webpage.mainFrame().contentsSize()
            printu"页面宽：%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;
            ifimg.save(filename):
                filepath=os.path.join(os.path.dirname(__file__), filename)
                printu"截图完毕：%s"% filepath
            else:
                printu"截图失败";
        else:
            printu"网页加载失败！"
        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 -*-
 
importtime
fromselenium importwebdriver
 
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条

0 0