频道栏目
首页 > 程序开发 > Web开发 > Python > 正文
Python:一个多功能的抓图工具开发(附源码)
2012-04-06 09:56:20      个评论      
收藏   我要投稿

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

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

一、代码:


[python]
#!/usr/bin/env python  
#coding=gb2312  
 
#此模块主要提供抓图功能,支持以下三种抓图方式:  
#1、抓取全屏,快捷键CTRL+F1  
#2、抓取当前窗口,快捷键CTRL+F2  
#3、抓取所选区域,快捷键CTRL+F3  
#抓到之后,会自动弹出保存对话框,选择路径保存即可  
#*******************************************  
#更新记录  
#0.1 2012-03-10 create by dyx1024  
#********************************************  
 
import pyhk 
import wx 
import os 
import sys 
from PIL import ImageGrab 
import ctypes 
import win32gui 
import ctypes.wintypes  
 
 
def capture_fullscreen(): 
    '''''
    Function:全屏抓图
    Input:NONE
    Output: NONE
    author: socrates
    blog:http://blog.csdn.net/dyx1024
    date:2012-03-10
    '''   
    #抓图     
    pic = ImageGrab.grab() 
     
    #保存图片  
    save_pic(pic) 
     
def capture_current_windows(): 
    '''''
    Function:抓取当前窗口
    Input:NONE
    Output: NONE
    author: socrates
    blog:http://blog.csdn.net/dyx1024
    date:2012-03-10
    '''  
    #窗口结构         
    class RECT(ctypes.Structure): 
        _fields_ = [('left', ctypes.c_long), 
                ('top', ctypes.c_long), 
                ('right', ctypes.c_long), 
                ('bottom', ctypes.c_long)] 
        def __str__(self): 
            return str((self.left, self.top, self.right, self.bottom)) 
     
    rect = RECT() 
     
    #获取当前窗口句柄  
    HWND = win32gui.GetForegroundWindow() 
     
    #取当前窗口坐标  
    ctypes.windll.user32.GetWindowRect(HWND,ctypes.byref(rect)) 
 
    #调整坐标  
    rangle = (rect.left+2,rect.top+2,rect.right-2,rect.bottom-2) 
     
    #抓图  
    pic = ImageGrab.grab(rangle) 
     
    #保存  
    save_pic(pic) 
     
def capture_choose_windows(): 
    '''''
    Function:抓取选择的区域,没有自己写这个,借用QQ抓图功能
    Input:NONE
    Output: NONE
    author: socrates
    blog:http://blog.csdn.net/dyx1024
    date:2012-03-10
    '''      
    try: 
         #加载QQ抓图使用的dll  
         dll_handle = ctypes.cdll.LoadLibrary('CameraDll.dll')  
    except Exception: 
             try: 
                 #如果dll加载失败,则换种方法使用,直接运行,如果还失败,退出  
                 os.system("Rundll32.exe CameraDll.dll, CameraSubArea") 
             except Exception: 
                 return     
    else: 
         try: 
             #加载dll成功,则调用抓图函数,注:没有分析清楚这个函数带的参数个数  
             #及类型,所以此语句执行后会报参数缺少4个字节,但不影响抓图功能,所  
             #以直接忽略了些异常  
             dll_handle.CameraSubArea(0) 
         except Exception: 
             return            
 
def save_pic(pic, filename = '未命令图片.png'): 
    '''''
    Function:使用文件对框,保存图片
    Input:NONE
    Output: NONE
    author: socrates
    blog:http://blog.csdn.net/dyx1024
    date:2012-03-10
    '''      
    app = wx.PySimpleApp() 
     
    wildcard = "PNG(*.png)|*.png" 
    dialog = wx.FileDialog(None, "Select a place", os.getcwd(), 
                           filename, wildcard, wx.SAVE) 
    if dialog.ShowModal() == wx.ID_OK: 
        pic.save(dialog.GetPath().encode('gb2312')) 
    else: 
        pass 
     
    dialog.Destroy()     
     
 
def main(): 
    '''''
    Function:主函数,注册快捷键
    Input:NONE
    Output: NONE
    author: socrates
    blog:http://blog.csdn.net/dyx1024
    date:2012-03-10
    '''          
     
    #创建hotkey句柄   
    hot_handle = pyhk.pyhk() 
  
    #注册抓取全屏快捷键CTRL+F1  
    hot_handle.addHotkey(['Ctrl', 'F1'], capture_fullscreen) 
     
    #注册抓取当前窗口快捷键CTRL+F2  
    hot_handle.addHotkey(['Ctrl', 'F2'], capture_current_windows) 
     
    #注册抓取所选区域快捷键CTRL+F3  
    hot_handle.addHotkey(['Ctrl', 'F3'], capture_choose_windows) 
  
    #开始运行  
    hot_handle.start() 
     
if __name__ == "__main__": 
    main()  
#!/usr/bin/env python
#coding=gb2312

#此模块主要提供抓图功能,支持以下三种抓图方式:
#1、抓取全屏,快捷键CTRL+F1
#2、抓取当前窗口,快捷键CTRL+F2
#3、抓取所选区域,快捷键CTRL+F3
#抓到之后,会自动弹出保存对话框,选择路径保存即可
#*******************************************
#更新记录
#0.1 2012-03-10 create by dyx1024
#********************************************

import pyhk
import wx
import os
import sys
from PIL import ImageGrab
import ctypes
import win32gui
import ctypes.wintypes


def capture_fullscreen():
    '''
    Function:全屏抓图
    Input:NONE
    Output: NONE
    author: socrates
    blog:http://blog.csdn.net/dyx1024
    date:2012-03-10
    ''' 
    #抓图  
    pic = ImageGrab.grab()
   
    #保存图片
    save_pic(pic)
   
def capture_current_windows():
    '''
    Function:抓取当前窗口
    Input:NONE
    Output: NONE
    author: socrates
    blog:http://blog.csdn.net/dyx1024
    date:2012-03-10
    '''
    #窗口结构      
    class RECT(ctypes.Structure):
        _fields_ = [('left', ctypes.c_long),
                ('top', ctypes.c_long),
                ('right', ctypes.c_long),
                ('bottom', ctypes.c_long)]
        def __str__(self):
            return str((self.left, self.top, self.right, self.bottom))
   
    rect = RECT()
   
    #获取当前窗口句柄
    HWND = win32gui.GetForegroundWindow()
   
    #取当前窗口坐标
    ctypes.windll.user32.GetWindowRect(HWND,ctypes.byref(rect))

    #调整坐标
    rangle = (rect.left+2,rect.top+2,rect.right-2,rect.bottom-2)
   
    #抓图
    pic = ImageGrab.grab(rangle)
   
    #保存
    save_pic(pic)
   
def capture_choose_windows():
    '''
    Function:抓取选择的区域,没有自己写这个,借用QQ抓图功能
    Input:NONE
    Output: NONE
    author: socrates
    blog:http://blog.csdn.net/dyx1024
    date:2012-03-10
    '''    
    try:
         #加载QQ抓图使用的dll
         dll_handle = ctypes.cdll.LoadLibrary('CameraDll.dll')
    except Exception:
             try:
                 #如果dll加载失败,则换种方法使用,直接运行,如果还失败,退出
                 os.system("Rundll32.exe CameraDll.dll, CameraSubArea")
             except Exception:
                 return   
    else:
         try:
             #加载dll成功,则调用抓图函数,注:没有分析清楚这个函数带的参数个数
             #及类型,所以此语句执行后会报参数缺少4个字节,但不影响抓图功能,所
             #以直接忽略了些异常 www.2cto.com
             dll_handle.CameraSubArea(0)
         except Exception:
             return          

def save_pic(pic, filename = '未命令图片.png'):
    '''
    Function:使用文件对框,保存图片
    Input:NONE
    Output: NONE
    author: socrates
    blog:http://blog.csdn.net/dyx1024
    date:2012-03-10
    '''    
    app = wx.PySimpleApp()
   
    wildcard = "PNG(*.png)|*.png"
    dialog = wx.FileDialog(None, "Select a place", os.getcwd(),
                           filename, wildcard, wx.SAVE)
    if dialog.ShowModal() == wx.ID_OK:
        pic.save(dialog.GetPath().encode('gb2312'))
    else:
        pass
   
    dialog.Destroy()   
   

def main():
    '''
    Function:主函数,注册快捷键
    Input:NONE
    Output: NONE
    author: socrates
    blog:http://blog.csdn.net/dyx1024
    date:2012-03-10
    '''        
   
    #创建hotkey句柄
    hot_handle = pyhk.pyhk()
 
    #注册抓取全屏快捷键CTRL+F1
    hot_handle.addHotkey(['Ctrl', 'F1'], capture_fullscreen)
   
    #注册抓取当前窗口快捷键CTRL+F2
    hot_handle.addHotkey(['Ctrl', 'F2'], capture_current_windows)
   
    #注册抓取所选区域快捷键CTRL+F3
    hot_handle.addHotkey(['Ctrl', 'F3'], capture_choose_windows)
 
    #开始运行
    hot_handle.start()
   
if __name__ == "__main__":
    main()
二、测试:

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

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

  
2、抓全屏(CTRL+F1)

 
2、抓当前窗口(CTRL+F2)

3、抓所选区域(CTRL+F3)

 
三、有待改进:
  后面考虑增加托盘显示、用户自定义快捷键等功能,最终以exe格式发布(目前只做到了windows下使用,Linux下)
  

摘自  Socrates的专栏
 

点击复制链接 与好友分享!回本站首页
相关TAG标签 源码 多功能 工具
上一篇:用python实现的字典树
下一篇:Python:程序最小化到托盘功能实现
相关文章
图文推荐
点击排行

关于我们 | 联系我们 | 广告服务 | 投资合作 | 版权申明 | 在线帮助 | 网站地图 | 作品发布 | Vip技术培训 | 举报中心

版权所有: 红黑联盟--致力于做实用的IT技术学习网站