Python:一个多功能的抓图工具开发（附源码）

   最近几年，各种论坛回帖中经常会见到这样的回复：无图无真相。还有经常写工作周报或者实验结果时，均要贴几张图上去。所以，抓图的需求在工作和生活中应用算是比较多的了，如果在家，我个人一直使用PrtSc键或QQ的抓图工具，用PrtScr键的一个问题是，我必须打开画图板（mspaint.exe）,把剪贴板中的图片黏贴到画布上，再另存为，还是比较麻烦的。如果在公司，因为用不了QQ，此时，要抓个一块区域上的图，还要通过画图板来编辑一下，也是比较浪费时间的。

   为解决一上问题，今天使用python写了一个抓图的工具，支持使用快捷键抓取屏幕、当前窗口，所选区域三种需求，抓到之后并弹出保存对话框，再也不用使用画图板中转一下了，进入正题：

一、代码：

[python]  view plain copy print ?

1. #!/usr/bin/env python  
2. #coding=gb2312  
3.   
4. #此模块主要提供抓图功能，支持以下三种抓图方式：  
5. #1、抓取全屏,快捷键CTRL+F1  
6. #2、抓取当前窗口，快捷键CTRL+F2  
7. #3、抓取所选区域，快捷键CTRL+F3  
8. #抓到之后，会自动弹出保存对话框，选择路径保存即可  
9. #*******************************************  
10. #更新记录  
11. #0.1 2012-03-10 create by dyx1024  
12. #********************************************  
13.   
14. import pyhk  
15. import wx  
16. import os  
17. import sys  
18. from PIL import ImageGrab  
19. import ctypes  
20. import win32gui  
21. import ctypes.wintypes   
22.   
23.   
24. def capture_fullscreen():  
25.     ''''' 
26.     Function:全屏抓图 
27.     Input：NONE 
28.     Output: NONE 
29.     author: socrates 
30.     blog:http://blog.youkuaiyun.com/dyx1024 
31.     date:2012-03-10 
32.     '''    
33.     #抓图     
34.     pic = ImageGrab.grab()  
35.       
36.     #保存图片  
37.     save_pic(pic)  
38.       
39. def capture_current_windows():  
40.     ''''' 
41.     Function:抓取当前窗口 
42.     Input：NONE 
43.     Output: NONE 
44.     author: socrates 
45.     blog:http://blog.youkuaiyun.com/dyx1024 
46.     date:2012-03-10 
47.     '''   
48.     #窗口结构         
49.     class RECT(ctypes.Structure):  
50.         _fields_ = [('left', ctypes.c_long),  
51.                 ('top', ctypes.c_long),  
52.                 ('right', ctypes.c_long),  
53.                 ('bottom', ctypes.c_long)]  
54.         def __str__(self):  
55.             return str((self.left, self.top, self.right, self.bottom))  
56.       
57.     rect = RECT()  
58.       
59.     #获取当前窗口句柄  
60.     HWND = win32gui.GetForegroundWindow()  
61.       
62.     #取当前窗口坐标  
63.     ctypes.windll.user32.GetWindowRect(HWND,ctypes.byref(rect))  
64.   
65.     #调整坐标  
66.     rangle = (rect.left+2,rect.top+2,rect.right-2,rect.bottom-2)  
67.       
68.     #抓图  
69.     pic = ImageGrab.grab(rangle)  
70.       
71.     #保存  
72.     save_pic(pic)  
73.       
74. def capture_choose_windows():  
75.     ''''' 
76.     Function:抓取选择的区域，没有自己写这个，借用QQ抓图功能 
77.     Input：NONE 
78.     Output: NONE 
79.     author: socrates 
80.     blog:http://blog.youkuaiyun.com/dyx1024 
81.     date:2012-03-10 
82.     '''       
83.     try:  
84.          #加载QQ抓图使用的dll  
85.          dll_handle = ctypes.cdll.LoadLibrary('CameraDll.dll')   
86.     except Exception:  
87.              try:  
88.                  #如果dll加载失败，则换种方法使用，直接运行，如果还失败，退出  
89.                  os.system("Rundll32.exe CameraDll.dll, CameraSubArea")  
90.              except Exception:  
91.                  return      
92.     else:  
93.          try:  
94.              #加载dll成功，则调用抓图函数，注:没有分析清楚这个函数带的参数个数  
95.              #及类型，所以此语句执行后会报参数缺少4个字节，但不影响抓图功能，所  
96.              #以直接忽略了些异常  
97.              dll_handle.CameraSubArea(0)  
98.          except Exception:  
99.              return             
100.   
101. def save_pic(pic, filename = '未命令图片.png'):  
102.     ''''' 
103.     Function:使用文件对框，保存图片 
104.     Input：NONE 
105.     Output: NONE 
106.     author: socrates 
107.     blog:http://blog.youkuaiyun.com/dyx1024 
108.     date:2012-03-10 
109.     '''       
110.     app = wx.PySimpleApp()  
111.       
112.     wildcard = "PNG(*.png)|*.png"  
113.     dialog = wx.FileDialog(None, "Select a place", os.getcwd(),  
114.                            filename, wildcard, wx.SAVE)  
115.     if dialog.ShowModal() == wx.ID_OK:  
116.         pic.save(dialog.GetPath().encode('gb2312'))  
117.     else:  
118.         pass  
119.       
120.     dialog.Destroy()      
121.       
122.   
123. def main():  
124.     ''''' 
125.     Function:主函数，注册快捷键 
126.     Input：NONE 
127.     Output: NONE 
128.     author: socrates 
129.     blog:http://blog.youkuaiyun.com/dyx1024 
130.     date:2012-03-10 
131.     '''           
132.       
133.     #创建hotkey句柄   
134.     hot_handle = pyhk.pyhk()  
135.    
136.     #注册抓取全屏快捷键CTRL+F1  
137.     hot_handle.addHotkey(['Ctrl', 'F1'], capture_fullscreen)  
138.       
139.     #注册抓取当前窗口快捷键CTRL+F2  
140.     hot_handle.addHotkey(['Ctrl', 'F2'], capture_current_windows)  
141.       
142.     #注册抓取所选区域快捷键CTRL+F3  
143.     hot_handle.addHotkey(['Ctrl', 'F3'], capture_choose_windows)  
144.    
145.     #开始运行  
146.     hot_handle.start()  
147.       
148. if __name__ == "__main__":  
149.     main()   

二、测试：

 程序运行后，可以看别的事情，如果要抓图，根据要抓的图片类型，按下相应的快捷键即可,以下图片除第一张外，其他均使用本程序抓区。

 1、抓到后自动弹出保存对话框

 

2、抓全屏（CTRL+F1）

2、抓当前窗口（CTRL+F2）

3、抓所选区域（CTRL+F3）

三、有待改进:

  后面考虑增加托盘显示、用户自定义快捷键等功能，最终以exe格式发布（目前只做到了windows下使用，Linux下有空再研究一下）