本文实例讲述了Python实现可自定义大小的截屏功能。分享给大家供大家参考，具体如下：
蝈蝈这两天正忙着收拾家当去公司报道，结果做PHP的发小蛐蛐找到了他，说是想要一个可以截图工具。
大致需要做出这样的效果。
虽然已经很久不写Python代码了，但是没办法，盛情难却啊，只好硬着头皮上了。
关于这个需求，蝈蝈想了想，脑海里大概有这么几个实现的方式。
① 调用QQ的截图工具。
② 自己写一个。
这第一个嘛，应了那句老话。理想很丰满，现实很骨感。因为被集成的缘故，剖不出来是没办法用的，自认为技术还不到家的蝈蝈很快放弃了这个方法。
那么只能自己写一个了。从谷哥那了解到PIL的ImageGrab可以很方便的截图，默认截图是全屏范围，当然也可以传递一个Bbox元组来实现截图的范围截图。于是思路就很明确了：获取鼠标位置，调用ImageGrab截图
获取鼠标位置
这个嘛，其实还是很简单的。借助pyHook就可以啦。
|
1
2
3
4
5
|
global old_x, old_y, new_x, new_y, full, hm if event.MessageName == "mouse left down": old_x, old_y = event.Position if event.MessageName == "mouse left up": new_x, new_y = event.Position |
按下鼠标的那一刻开始记录初始坐标，然后鼠标抬起的那一刻更新结束坐标。这两个坐标的范围就是要截图的范围。这里面需要注意的就是鼠标坐标默认从左上角（0, 0）开始。
截图的具体实现
关于具体实现，无非是一个full标记，默认也是截全屏的图，当full为False的时候，按照两次鼠标的绝对位置实现范围截图。
|
1
2
3
4
5
6
|
# 划屏 if full: image = ImageGrab.grab((0, 0, gsm(0), gsm(1))) else: image = ImageGrab.grab((old_x, old_y, new_x, new_y)) image.show() |
好啦，核心功能已经做好啦。为了方便蛐蛐进行自定义的拓展，蝈蝈把源码发给了他。
|
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
|
# coding: utf8# @Author: 郭 璞# @File: capture.py# @Time: 2017/7/24# @Contact: 1064319632@qq.com# @blog: http://blog.csdn.net/marksinoberg# @Description: 根据鼠标移动进行划屏截图import pyHookimport pythoncomimport win32guifrom PIL import Image, ImageGrabfrom win32api import GetSystemMetrics as gsm# 提前绑定鼠标位置事件old_x, old_y = 0, 0new_x, new_y = 0, 0def hotkey(key=None): """绑定热键，开始进行划屏截图操作""" passdef on_mouse_event(event): global old_x, old_y, new_x, new_y, full, hm if event.MessageName == "mouse left down": old_x, old_y = event.Position if event.MessageName == "mouse left up": new_x, new_y = event.Position # 解除事件绑定 hm.UnhookMouse() hm = None # 划屏 if full: image = ImageGrab.grab((0, 0, gsm(0), gsm(1))) else: image = ImageGrab.grab((old_x, old_y, new_x, new_y)) image.show()full = Falsehm = Nonedef capture(): hm = pyHook.HookManager() hm.SubscribeMouseAll(on_mouse_event) hm.HookMouse() pythoncom.PumpMessages()capture() |
核心功能已经算是完成了，虽然貌似并没有什么太大的用处。
因为就要走了，所以蝈蝈没有多少时间来润色，只能这样匆匆交差了。除了代码，蝈蝈特意嘱咐了下面这几句话：
① 增加保存到本地功能。
② 绑定系统快捷键，这样打游戏的时候也可以截图。
③ 增加蒙层，截图的时候提供更好的用户体验。
蛐蛐听完之后，貌似也有了自己的想法，然后就自己琢磨去了。其实他不知道的是，蝈蝈对于截到的图的另一层处理。
简易图片相似度分析
|
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
|
# coding: utf8# @Author: 郭 璞# @File: similar.py# @Time: 2017/7/23# @Contact: 1064319632@qq.com# @blog: http://blog.csdn.net/marksinoberg# @Description: 两张图片相似度计算实现。from PIL import Imagedef pixel_way(img1, img2): image1 = Image.open(img1, 'r') image2 = Image.open(img2, 'r') return get_pixel_details(image1)==get_pixel_details(image2)def get_pixel_details(img): pixels = img.load() r, g, b = 0, 0, 0 counter = 0 for x in range(img.size[0]): for y in range(img.size[1]): counter += 1 r1, g1, b1 = pixels[x, y] r += r1 g += g1 b += b1 return (r/counter, g/counter, b/counter)if __name__ == '__main__': image1 = r'./1.png' image2 = r'./1.png' img = Image.open(image1, 'r') img.resize((256,256)).convert("RGB") print(pixel_way(image1, image2)) |
图片像素直方图
|
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
|
# coding: utf8# @Author: 郭 璞# @File: pixel-compare.py# @Time: 2017/7/24# @Contact: 1064319632@qq.com# @blog: http://blog.csdn.net/marksinoberg# @Description: 计算RGB值相关from PIL import Imagefrom PIL import ImageDrawim = Image.open('1.png')im = im.convert("L")width, height = im.sizepix = im.load()a = [0]*256for w in range(width): for h in range(height): p = pix[w, h] a[p] = a[p] + 1x = max(a)print(a, "---", len(a), '-----', x)image = Image.new('RGB', (256, 256), (255, 255, 255))draw = ImageDraw.Draw(image)for k in range(256): a[k] = a[k]*200/x source = (k, 255) target = (k, 255-a[k]) draw.line([source, target], (100, 100, 100))image.show() |
还有很多更好玩的，但是有时候，话多，不是一件好事，想到这里，蝈蝈又不自觉的回忆起了那段不堪的帮忙的经历，无奈……
希望本文所述对大家Python程序设计有所帮助。
原文链接：http://blog.csdn.net/marksinoberg/article/details/76039134
