23云恋49枫
posts - 167,comments - 59,views - 34万
Python各种花式截图工具，截到你手软

前言：

　　　最近，项目中遇到了一个关于实现通过给定URL，实现对网页屏幕进行截图的一个功能，前面代码中已经用python的第三方库实现了截图功能，但在上线以后出现了一些bug，所以就改bug的任务就落在了我的头上，这方面前面没有接触过，正好利用这个机会学习一下python中各种截图方法，下面就是我要介绍的一些常用截图功能的基本使用，希望可以帮到你，也可以提出文中不足的地方。

PIL和Pillow

PIL：Python Imaging Library，已经是Python平台事实上的图像处理标准库了。PIL功能非常强大，但API却非常简单易用，不过只支持到Python 2.7

Pillow：是PIL的一个派生分支，但如今已经发展成为比PIL本身更具活力的图像处理库。目前最新版本是3.0.0

1、安装

在Debian/Ubuntu Linux下直接通过apt安装：

?
1
$ sudo apt-get install python-imaging

Mac和其他版本的Linux可以直接使用easy_install或pip安装，安装前需要把编译环境装好：

?
1
$ sudo easy_install PIL

Windows平台就去PIL官方网站下载exe安装包或pip

?
1
$ pip install pillow

2、实现屏幕截图

?
1
2
3
from PIL import ImageGrab
im = ImageGrab.grab()
im.save("1.png")    #定义保存的路径和保存的图片格式

PyQt4

PyQt是一个创建GUI应用程序的工具包。它是Python编程语言和Qt库的成功融合。Qt库是目前最强大的库之一。PyQt是由Phil Thompson 开发。PyQt是Qt库的Python版本。PyQt3支持Qt1到Qt3。 PyQt4支持Qt4。它的首次发布也是在1998年，但是当时它叫 PyKDE，因为开始的时候SIP和PyQt没有分开。PyQt是用SIP写的。PyQt 提供 GPL版和商业版。

1、安装

windows下：

?
1
2
32位： http://sourceforge.net/projects/pyqt/files/PyQt4/PyQt-4.11.4/PyQt4-4.11.4-gpl-Py2.7-Qt4.8.7-x32.exe
64位： http://sourceforge.net/projects/pyqt/files/PyQt4/PyQt-4.11.4/PyQt4-4.11.4-gpl-Py2.7-Qt4.8.7-x64.exe

linux下：

?
1
https://www.riverbankcomputing.com/software/pyqt/download

2、实现网页截图

有些图片较多，或是页面较长的网站，截出来的图，会出现图片未加载完毕或页面截取不完整等情况，找了很多资料也没解决，如果那个小伙伴会的话，可以教一下我

?
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
#!/usr/bin/env python
# -*- coding:utf-8 -*-
import sys,time
import os.path
from PyQt4 import QtGui, QtCore, QtWebKit
 
class PageShotter(QtGui.QWidget):
    def __init__(self, url, parent=None):
        QtGui.QWidget.__init__(self, parent)
        self.url = url
 
    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.savePage)
 
    def savePage(self, finished):
        if finished:
            print "开始截图！"
            size = self.webPage.mainFrame().contentsSize()
            print "页面宽：%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 = "shot.png"
            if img.save(fileName):
                filePath = os.path.join(os.path.dirname(__file__), fileName)
                print "截图完毕：%s" % filePath
            else:
                print "截图失败"
        else:
            print "网页加载失败！"
        self.close()
 
 
if __name__ == "__main__":
    app = QtGui.QApplication(sys.argv)
    shotter = PageShotter("https://www.jd.com/")
    shotter.shot()
    sys.exit(app.exec_())

加强版

复制代码
#!/usr/bin/python
# coding:utf-8

import sys
import os.path
import requests
import urlparse
import time

sys.path.append('../')
from PyQt4 import QtGui, QtCore, QtWebKit
from PyQt4.QtNetwork import QNetworkRequest


class WebStatus(object):
        def __init__(self, timeout, tries):
                '''
                类说明：获取网页的状态码。如200表示“网页正常，可访问”
                参数：timeout(s) 请求等待时间；tries 请求次数；
                '''
                self.__timeout = timeout
                self.__tries = tries

                self.__headers = {
                        "Accept": "text/html,application/xhtml+xml,application/xml;q=0.9,*/*;q=0.8",
                        "Accept-Language": "en-US,en;q=0.5",
                        "Accept-Encoding": "gzip, deflate",
                        "User-Agent": "Mozilla/5.0 (X11; Linux x86_64; rv:45.0) Gecko/20100101 Firefox/45.0"
                }

        def clear_url(self, url):
                if not url:
                        return None
                if ',' in url:
                        url = url.split(',')[0]
                url = url.strip()
                if url.startswith('http://') or url.startswith('https://'):
                        pass
                else:
                        url = 'http://' + url
                try:
                        parse = urlparse.urlparse(url)
                        url_new = parse.scheme + '://' + parse.netloc
                except:
                        url_new = url
                if url_new.endswith('.'):
                        url_new = url_new.rstrip('.')
                return url_new

        def isAccessible(self, url):
                '''
                说明：获取网页状态码,判断该网页是否可正常访问的，200表示网页可正常访问的。
                参数：url 网址；
                返回：
                                True：状态码200，网页正常可访问；
                                False：网页状态码非200，或者请求异常。
                '''
                url = self.clear_url(url)
                tries = self.__tries
                status = False
                r = None
                while tries > 0:
                        try:
                                r = requests.get(url=url, headers=self.__headers, timeout=self.__timeout)
                        except:
                                tries -= 1
                                status = False
                        else:
                                if r.status_code == 200:
                                        status = url
                                break
                        finally:
                                if r:
                                        r.close()
                                        r = None
                return status

        def __del__(self):
                pass


class PageShotter(QtGui.QWidget):
        def __init__(self, url, parent=None,pic_path = './pic_path'):
                path = os.path.exists(pic_path)
                if not path:
                        os.mkdir(pic_path)
                self.request = QNetworkRequest()
                QtGui.QWidget.__init__(self, parent)
                self.url = url
                self.dir_path = os.path.join(pic_path,str(time.strftime('%Y-%m-%d %H:%M:%S', time.localtime(time.time()))) + '_' + urlparse.urlparse(self.url).netloc + '.png')

        def shot(self):
                webView = QtWebKit.QWebView(self)
                self.request.setRawHeader("User-Agent", "Mozilla/5.0 (X11; Linux x86_64; rv:45.0) Gecko/20100101 Firefox/45.0")
                webView.load(QtCore.QUrl(self.url))
                self.webPage = webView.page()
                self.connect(webView, QtCore.SIGNAL("loadFinished(bool)"), self.savePage)
                return self.dir_path

        def savePage(self, finished):
                if finished:
                        size = self.webPage.mainFrame().contentsSize()
                        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()
                        if img.save(self.dir_path):
                                filePath = os.path.join(os.path.dirname(__file__),self.dir_path)
                                #print "截图完毕：%s" % filePath
                        else:
                                print "截图失败"
                else:
                        print "网页加载失败！"
                self.close()


if __name__ == "__main__":
        obj1 = WebStatus(10, 2)
        status = obj1.isAccessible("https://www.jd.com")
        if status:
                app = QtGui.QApplication(sys.argv)
                shotter = PageShotter(status,pic_path = './dirpath_test')
                obj = shotter.shot()
                print obj
                sys.exit(app.exec_())
        else:
                print 'Invalid URL'
pyqt4截图加强版
复制代码

selenium +phantomjs

安装请参考  http://www.cnblogs.com/luxiaojun/p/6144748.html

?
1
2
3
4
5
6
7
8
9
10
from selenium import webdriver
from selenium.webdriver.common.desired_capabilities import DesiredCapabilities
  
dcap = dict(DesiredCapabilities.PHANTOMJS)  #设置userAgent
dcap["phantomjs.page.settings.userAgent"] = ("Mozilla/5.0 (Macintosh; Intel Mac OS X 10.9; rv:25.0) Gecko/20100101 Firefox/25.0 ")
  
obj = webdriver.PhantomJS(executable_path='C:\Python27\Scripts\phantomjs.exe',desired_capabilities=dcap) #加载网址
obj.get('http://wap.95533pc.com')#打开网址
obj.save_screenshot("1.png")   #截图保存
obj.quit()

另外还有利用subprocess中的popen方法指令linux命令进行截图的方法，如 cutycapt，会在以后的文章中继续写到，同样也可以参考以下文章：

http://www.111cn.net/sys/linux/81361.htm

 

posted on   23云恋49枫  阅读(17671)  评论(2)  编辑  收藏  举报
编辑推荐：
· AI与.NET技术实操系列：基于图像分类模型对图像进行分类
· go语言实现终端里的倒计时
· 如何编写易于单元测试的代码
· 10年+ .NET Coder 心语，封装的思维：从隐藏、稳定开始理解其本质意义
· .NET Core 中如何实现缓存的预热？
阅读排行：
· 分享一个免费、快速、无限量使用的满血 DeepSeek R1 模型，支持深度思考和联网搜索！
· 25岁的心里话
· 基于 Docker 搭建 FRP 内网穿透开源项目（很简单哒）
· ollama系列01：轻松3步本地部署deepseek，普通电脑可用
· 按钮权限的设计及实现
< 2025年3月 >
日 一 二 三 四 五 六
23 24 25 26 27 28 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 1 2 3 4 5

Copyright © 2025 23云恋49枫
Powered by .NET 9.0 on Kubernetes Powered By博客园
点击右上角即可分享
微信分享提示