Python天天美味(36) - 用Python实现Spy++

Spy++是微软出品的用来获取Window窗口信息的一个小工具。实现的原理其实不难，通过调用某些特定的Windows API即可。于是，我打算用Python也实现一个功能简化版本的小工具，取名叫PySpy++。Python中调用Windows API一般使用pywin32这套库，界面库我使用PyQT4。

Spy++原理

Spy++中，最常用的一个功能，就是识别窗口。其中主要需要用到的Windows API有：

获取当前鼠标位置：

BOOL GetCursorPos( LPPOINT lpPoint );

获取位于指定位置的窗口句柄：

HWND WindowFromPoint( POINT Point );

获取窗口类别：

int  GetClassName( HWND hWnd, LPTSTR lpClassName,  int  nMaxCount );

获取窗口内容或标题：

方法一：

int  GetWindowText( HWND hWnd, LPTSTR lpString,  int  nMaxCount );

这个API有时候不能取到某些控件的值，因此，使用方法二。

方法二：

给窗口发送WM_GETTEXT消息：

LRESULT SendMessage( HWND hWnd, UINT Msg, WPARAM  wParam, LPARAM lParam );

高亮选中的窗口：

先获取当前窗口的大小，然后画一个矩形框。

BOOL GetWindowRect( HWND hWnd, LPRECT lpRect );

BOOL Rectangle(
    HDC hdc,  //  handle to DC
     int  nLeftRect,  //  x-coord of upper-left corner of rectangle
     int  nTopRect,  //  y-coord of upper-left corner of rectangle
     int  nRightRect,  //  x-coord of lower-right corner of rectangle
     int  nBottomRect  //  y-coord of lower-right corner of rectangle
);

鼠标移开窗口后，窗口需要恢复原状，需要重新刷新：

BOOL InvalidateRect(
    HWND hWnd,  //  handle to window
    CONST RECT *  lpRect,  //  rectangle coordinates
    BOOL bErase  //  erase state
);

BOOL UpdateWindow(
    HWND hWnd  //  handle to window
);

BOOL RedrawWindow(
    HWND hWnd,  //  handle to window
    CONST RECT  * lprcUpdate,  //  update rectangle
    HRGN hrgnUpdate,  //  handle to update region
    UINT flags  //  array of redraw flags
);

PyWin32对应的函数

在Python中调用Windows API，首先下载PyWin32，地址：http://pywin32.sourceforge.net/

安装完成后，打开帮助文档Python for Windows Documentation，里面有所有需要的东西，随时用来查看。

常用的API在win32api模块里，界面相关的API在win32gui模块里，API参数中定义的一些常量在win32con模块中。上面的Windows API对应PyWin32中的函数为：

(int, int)  =  win32gui. GetCursorPos ()
int  =  win32gui. WindowFromPoint (point)
string  =  win32gui. GetClassName (hwnd)
string  =  win32gui. GetWindowText (hwnd)
int  =  win32gui. SendMessage (hwnd, message , wparam , lparam )
(left, top, right, bottom)  =  win32gui. GetWindowRect (hwnd)
win32gui. Rectangle (hdc, LeftRect, TopRect, RightRect, BottomRect)
win32gui. InvalidateRect (hWnd, Rect, Erase)
win32gui. UpdateWindow (hwnd)
win32gui. RedrawWindow (hWnd, rcUpdate, hrgnUpdate, flags)

代码实现

界面库使用PyQT4，参考资料可以从我之前的一篇博客里了解：PyQt4 学习资料汇总

工具对话框窗口有两个控件，一个是QLabel控件，一个是QTextEdit控件。QLabel控件就是那个用来鼠标按下去后去捕捉窗口，QTextEdit控件用来显示窗口的信息。为了让QTextEdit响应自定义的鼠标事件，我创建了一个自定义QLabel控件SpyLabel，继承自QLabel。

class  SpyLabel(QtGui.QLabel):
     def   __init__ (self, parent  =  None):
        QtGui.QLabel. __init__ (self, parent)
        self.parent  =  parent
        self.spying  =  False
        self.rectanglePen  =  win32gui.CreatePen(win32con.PS_SOLID,  3 , win32api.RGB( 255 , 0, 0))
        self.prevWindow  =  None
        self.setCursor(QtCore.Qt.SizeAllCursor)

SpyLabel中处理鼠标移动事件：

def  mouseMoveEvent(self, event):
     if  self.spying:
        curX, curY  =  win32gui.GetCursorPos()
        hwnd  =  win32gui. WindowFromPoint ((curX, curY))

         if  self.checkWindowValidity(hwnd):               
             if  self.prevWindow:
                self.refreshWindow(self.prevWindow)
            self.prevWindow  =  hwnd
            self.highlightWindow(hwnd)
            self.displayWindowInformation(hwnd)

鼠标松开事件：

def  mouseReleaseEvent(self, event):
     if  self.spying:
         if  self.prevWindow:
            self.refreshWindow(self.prevWindow)
        win32gui.ReleaseCapture()
        self.spying  =  False

高亮窗口的函数：

def  highlightWindow(self, hwnd):
    left, top, right, bottom  =  win32gui.GetWindowRect(hwnd)
    windowDc  =  win32gui.GetWindowDC(hwnd)
     if  windowDc:
        prevPen  =  win32gui.SelectObject(windowDc, self.rectanglePen)
        prevBrush  =  win32gui.SelectObject(windowDc, win32gui.GetStockObject(win32con.HOLLOW_BRUSH))

        win32gui. Rectangle (windowDc, 0, 0, right  -  left, bottom  -  top)
        win32gui.SelectObject(windowDc, prevPen)
        win32gui.SelectObject(windowDc, prevBrush)
        win32gui.ReleaseDC(hwnd, windowDc)

刷新窗口的函数：

def  refreshWindow(self, hwnd):
    win32gui. InvalidateRect (hwnd, None, True)
    win32gui. UpdateWindow (hwnd)
    win32gui. RedrawWindow (hwnd, 
        None, 
        None,  
        win32con.RDW_FRAME |
            win32con.RDW_INVALIDATE |
            win32con.RDW_UPDATENOW |
            win32con.RDW_ALLCHILDREN)

显示窗口信息：

def  displayWindowInformation(self, hwnd):
    className  =  win32gui.GetClassName(hwnd)
    buf_size  =   1   +  win32gui. SendMessage (hwnd, win32con.WM_GETTEXTLENGTH, 0, 0)
    buffer  =  win32gui.PyMakeBuffer(buf_size)
    win32gui. SendMessage (hwnd, win32con.WM_GETTEXT, buf_size, buffer)
    windowText  =  buffer[:buf_size]

     try :
        windowText  =  unicode(windowText,  ' gbk ' )
     except :
         pass

    message  =  [ ' Handle:\t '   +  str(hwnd),
                ' Class Name:\t '   +  className,
                ' Window Text:\t '   +  windowText]
    self.output( ' \r\n ' .join(message))

注意到上面SendMessage函数，需要传入一个分配的缓冲区，用于获取返回的内容。这里使用了：

buffer  =  win32gui.PyMakeBuffer(buf_size)

由于返回的内容中可能有中文，因此使用unicode(windowText, 'gbk')进行一下转换。

演示

 

二进制下载：

http://pyspyplusplus.googlecode.com/files/pyspy++.exe 

源代码：

http://code.google.com/p/pyspyplusplus/ 

 

 

Python 天天美味系列（总）

Python 天天美味(31) - python数据结构与算法之插入排序 

Python 天天美味(32) - python数据结构与算法之堆排序 

Python 天天美味(33) - 五分钟理解元类（Metaclasses）[转]

Python 天天美味(34) - Decorators详解

Python 天天美味(35) - 细品lambda