利用Python實現一個多功能的截屏工具

jopen 12年前發布 | 114K 次閱讀 Python Python開發

最近幾年，各種論壇回帖中經常會見到這樣的回復：1. 無圖無真相，2. no picture you say a XXX，還有經常寫工作周報或者實驗結果時，均要貼幾張圖上去。所以，抓圖的需求在工作和生活中應用算是比較多的了，如果在家，我個人一直使用PrtSc 鍵或QQ的抓圖工具，用PrtScr鍵的一個問題是，我必須打開畫圖板（mspaint.exe）,把剪貼板中的圖片黏貼到畫布上，再另存為，還是比較麻煩的。如果在公司，因為用不了QQ，此時，要抓個一塊區域上的圖，還要通過畫圖板來編輯一下，也是比較浪費時間的。

為解決一上問題，今天使用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()

二、測試：

程序運行后，可以看別的事情，如果要抓圖，根據要抓的圖片類型，按下相應的快捷鍵即可,以下圖片除第一張外，其他均使用本程序抓區。

1、抓到后自動彈出保存對話框

利用Python實現一個多功能的截屏工具