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