python截图库_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安装：

$ sudo apt-get install python-imaging

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

$ sudo easy_install PIL

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

$ pip install pillow

2、实现屏幕截图

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下：

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下：

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

2、实现网页截图

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

#!/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

importsysimportos.pathimportrequestsimporturlparseimporttime

sys.path.append('../')from PyQt4 importQtGui, QtCore, QtWebKitfrom PyQt4.QtNetwork importQNetworkRequestclassWebStatus(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"}defclear_url(self, url):if noturl:returnNoneif ',' inurl:

url= url.split(',')[0]

url=url.strip()if url.startswith('http://') or url.startswith('https://'):pass

else:

url= 'http://' +urltry:

parse=urlparse.urlparse(url)

url_new= parse.scheme + '://' +parse.netlocexcept:

url_new=urlif url_new.endswith('.'):

url_new= url_new.rstrip('.')returnurl_newdefisAccessible(self, url):'''说明：获取网页状态码,判断该网页是否可正常访问的，200表示网页可正常访问的。

参数：url 网址；

返回：

True：状态码200，网页正常可访问；

False：网页状态码非200，或者请求异常。'''url=self.clear_url(url)

tries= self.__triesstatus=False

r=Nonewhile tries >0:try:

r= requests.get(url=url, headers=self.__headers, timeout=self.__timeout)except:

tries-= 1status=Falseelse:if r.status_code == 200:

status=urlbreak

finally:ifr:

r.close()

r=Nonereturnstatusdef __del__(self):pass

classPageShotter(QtGui.QWidget):def __init__(self, url, parent=None,pic_path = './pic_path'):

path=os.path.exists(pic_path)if notpath:

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')defshot(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)returnself.dir_pathdefsavePage(self, finished):iffinished:

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()ifimg.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")ifstatus:

app=QtGui.QApplication(sys.argv)

shotter= PageShotter(status,pic_path = './dirpath_test')

obj=shotter.shot()printobj

sys.exit(app.exec_())else:print 'Invalid URL'

pyqt4截图加强版

selenium +phantomjs

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

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