Python调用WinAPI实战：pywin32模块详解与应用

殷迎彤

1. 理解WinAPI与pywin32的桥梁作用

在Windows平台开发中，Win32 API（简称WinAPI）是操作系统最底层的编程接口。它像一座桥梁，连接着应用程序和Windows内核。而pywin32则是Python开发者调用这些WinAPI的必备工具包，相当于给Python装上了操作Windows系统的"瑞士军刀"。

我第一次接触pywin32是在开发一个自动化测试工具时。当时需要模拟用户操作Windows桌面程序，发现直接调用WinAPI可以实现最精准的控制。比如通过user32.dll里的FindWindow和SendMessage函数，就能定位窗口并发送点击事件。这种能力让Python突破了脚本语言的限制，获得了系统级的控制权。

2. pywin32的核心模块解析

2.1 win32api模块：基础功能入口

win32api模块封装了最常用的WinAPI函数。比如获取系统信息的经典操作：

python复制import win32api
# 获取屏幕分辨率
width = win32api.GetSystemMetrics(0)
height = win32api.GetSystemMetrics(1)
print(f"当前屏幕分辨率：{width}x{height}")

# 获取最后错误代码
last_error = win32api.GetLastError()

注意：调用WinAPI后务必检查错误码。许多函数成功时返回0，失败时返回非零值，这时需要用GetLastError()获取详细错误信息。

2.2 win32gui模块：图形界面控制

这个模块让我实现了自动化测试中的窗口操作：

python复制import win32gui

# 查找记事本窗口
hwnd = win32gui.FindWindow("Notepad", None)
if hwnd:
    # 获取窗口标题
    title = win32gui.GetWindowText(hwnd)
    # 最大化窗口
    win32gui.ShowWindow(hwnd, 3)  # SW_MAXIMIZE = 3

实测发现，窗口类名在不同语言系统下可能变化。比如中文系统的记事本类名是"Notepad"，而英文系统可能需要用"记事本"。

2.3 win32con模块：常量大全

这个模块包含了所有WinAPI定义的常量值。比如消息类型的常量：

python复制from win32con import WM_CLOSE, WM_KEYDOWN, VK_RETURN

# 发送回车键按下消息
win32gui.SendMessage(hwnd, WM_KEYDOWN, VK_RETURN, 0)

建议总是通过win32con引用常量，而不是直接使用数值。因为不同Windows版本中，某些常量的值可能变化。

3. 实战：用pywin32开发屏幕截图工具

3.1 获取设备上下文(DC)

python复制import win32gui
import win32ui
from win32con import SRCCOPY

# 获取桌面窗口句柄
desktop = win32gui.GetDesktopWindow()
# 获取设备上下文
dc = win32gui.GetWindowDC(desktop)
dc_obj = win32ui.CreateDCFromHandle(dc)

这里有个坑点：必须手动释放DC资源，否则会导致内存泄漏。正确的做法是：

python复制try:
    # 使用DC...
finally:
    dc_obj.DeleteDC()
    win32gui.ReleaseDC(desktop, dc)

3.2 创建兼容DC并截图

python复制# 创建内存DC
mem_dc = dc_obj.CreateCompatibleDC()

# 创建位图对象
width = win32api.GetSystemMetrics(0)
height = win32api.GetSystemMetrics(1)
bitmap = win32ui.CreateBitmap()
bitmap.CreateCompatibleBitmap(dc_obj, width, height)

# 将位图选入内存DC
mem_dc.SelectObject(bitmap)

# 复制屏幕内容
mem_dc.BitBlt((0,0), (width,height), dc_obj, (0,0), SRCCOPY)

3.3 保存位图到文件

python复制# 保存为BMP文件
bitmap.SaveBitmapFile(mem_dc, "screenshot.bmp")

# 或者转换为PIL Image对象
from PIL import Image
import numpy as np

bmp_info = bitmap.GetInfo()
bmp_str = bitmap.GetBitmapBits(True)
image = Image.frombuffer(
    'RGB',
    (bmp_info['bmWidth'], bmp_info['bmHeight']),
    bmp_str, 'raw', 'BGRX', 0, 1
)
image.save("screenshot.png")

4. 高级应用：钩子(Hook)技术

4.1 安装键盘钩子

python复制import win32api
import win32con
import pywintypes

def keyboard_hook(nCode, wParam, lParam):
    if wParam == win32con.WM_KEYDOWN:
        vk_code = lParam[0]
        print(f"按键按下: {chr(vk_code)}")
    return win32api.CallNextHookEx(hook, nCode, wParam, lParam)

# 安装钩子
hook = win32api.SetWindowsHookEx(
    win32con.WH_KEYBOARD_LL,
    keyboard_hook,
    win32api.GetModuleHandle(None),
    0
)

# 保持钩子运行
try:
    msg = win32gui.GetMessage(None, 0, 0)
    while msg:
        win32gui.TranslateMessage(msg)
        win32gui.DispatchMessage(msg)
        msg = win32gui.GetMessage(None, 0, 0)
finally:
    win32api.UnhookWindowsHookEx(hook)

警告：全局钩子会影响系统性能，调试时可能导致键盘无响应。建议在虚拟机中测试。

4.2 鼠标事件监控

python复制def mouse_hook(nCode, wParam, lParam):
    if wParam == win32con.WM_LBUTTONDOWN:
        x, y = lParam[0], lParam[1]
        print(f"左键点击位置: ({x}, {y})")
    return win32api.CallNextHookEx(hook, nCode, wParam, lParam)

hook = win32api.SetWindowsHookEx(
    win32con.WH_MOUSE_LL,
    mouse_hook,
    win32api.GetModuleHandle(None),
    0
)

5. 常见问题排查指南

5.1 错误代码处理

WinAPI错误通常以错误代码形式返回。pywin32提供了错误转换功能：

python复制try:
    # 调用可能失败的WinAPI
except pywintypes.error as e:
    print(f"错误代码: {e.winerror}")
    print(f"错误描述: {e.strerror}")

常见错误代码：

2: 文件未找到
5: 访问被拒绝
87: 参数错误
1400: 无效窗口句柄

5.2 句柄管理问题

窗口句柄是WinAPI编程的核心。常见问题包括：

窗口关闭后仍使用其句柄
跨进程传递句柄未使用DuplicateHandle
未正确释放GDI对象导致泄漏

建议的调试方法：

python复制# 检查窗口是否有效
if not win32gui.IsWindow(hwnd):
    print("窗口句柄已失效")

# 枚举所有顶层窗口
def enum_windows_callback(hwnd, extra):
    if win32gui.IsWindowVisible(hwnd):
        print(win32gui.GetWindowText(hwnd))
win32gui.EnumWindows(enum_windows_callback, None)

5.3 32/64位兼容性问题

在64位Python中调用32位DLL时可能出现问题。解决方法：

python复制from ctypes import windll
# 显式加载32位DLL
user32 = windll.user32
# 或者使用绝对路径
dll_path = r"C:\Windows\SysWOW64\user32.dll"
user32 = windll.LoadLibrary(dll_path)

6. 性能优化技巧

6.1 减少GDI对象创建

频繁创建/销毁GDI对象（如DC、位图）会导致性能下降。建议：

复用对象池
使用CreateCompatibleDC而不是每次都创建新DC
批量操作后统一释放资源

6.2 异步消息处理

对于耗时操作，使用PostMessage而非SendMessage：

python复制win32gui.PostMessage(hwnd, win32con.WM_USER+100, 0, 0)

SendMessage会阻塞直到消息处理完成，而PostMessage只是将消息放入队列。

6.3 内存管理

pywin32对象底层是COM接口，需要注意引用计数：

python复制# 错误示例：循环引用
bitmap = win32ui.CreateBitmap()
dc.SelectObject(bitmap)
# 必须显式释放
dc.SelectObject(None)
bitmap.DeleteObject()

7. 安全注意事项

7.1 UAC权限问题

某些操作需要管理员权限。检测当前权限：

python复制import win32security
import win32con

def is_admin():
    try:
        return win32security.OpenProcessToken(
            win32api.GetCurrentProcess(),
            win32con.TOKEN_QUERY
        ).GetTokenInformation(win32security.TokenElevation)
    except:
        return False

7.2 输入模拟的安全风险

自动化输入工具可能被安全软件拦截。建议：

添加适当的延迟
避免高频连续操作
提供用户可见的提示

7.3 数字签名验证

加载外部DLL时应验证签名：

python复制from win32crypt import CryptQueryObject

def verify_signature(dll_path):
    hFile = win32file.CreateFile(
        dll_path,
        win32con.GENERIC_READ,
        0, None,
        win32con.OPEN_EXISTING,
        0, None
    )
    try:
        return CryptQueryObject(
            win32con.CERT_QUERY_OBJECT_FILE,
            hFile,
            win32con.CERT_QUERY_CONTENT_FLAG_ALL,
            win32con.CERT_QUERY_FORMAT_FLAG_ALL,
            0
        )
    finally:
        win32file.CloseHandle(hFile)

8. 扩展应用场景

8.1 系统托盘图标开发

python复制import win32gui_struct
import win32con

class SysTrayIcon:
    def __init__(self):
        message_map = {
            win32con.WM_DESTROY: self.on_destroy,
            win32con.WM_COMMAND: self.on_command,
            win32con.WM_USER+20: self.on_notify,
        }
        wc = win32gui.WNDCLASS()
        wc.lpszClassName = 'PythonSysTrayIcon'
        wc.style = win32con.CS_VREDRAW | win32con.CS_HREDRAW
        wc.hCursor = win32gui.LoadCursor(0, win32con.IDC_ARROW)
        wc.hbrBackground = win32con.COLOR_WINDOW
        wc.lpfnWndProc = message_map
        self.classAtom = win32gui.RegisterClass(wc)

8.2 注册表操作

python复制import win32api
import win32con

# 读取注册表
value, type = win32api.RegQueryValueEx(
    win32con.HKEY_CURRENT_USER,
    "Software\\Microsoft\\Windows\\CurrentVersion\\Run",
    "PythonApp"
)

# 写入注册表
win32api.RegSetValueEx(
    win32con.HKEY_CURRENT_USER,
    "Software\\MyApp",
    0,
    win32con.REG_SZ,
    "config_value"
)

8.3 服务控制

python复制import win32service
import win32serviceutil

class PythonService(win32serviceutil.ServiceFramework):
    _svc_name_ = "PythonService"
    _svc_display_name_ = "Python Service Example"
    
    def SvcDoRun(self):
        # 服务主逻辑
        while True:
            win32api.Sleep(5000)

if __name__ == '__main__':
    win32serviceutil.HandleCommandLine(PythonService)

9. 调试技巧与工具

9.1 Spy++替代方案

没有Visual Studio时，可以用pywin32实现窗口侦查：

python复制def get_window_info(hwnd):
    # 获取类名
    class_name = win32gui.GetClassName(hwnd)
    # 获取窗口文本
    text = win32gui.GetWindowText(hwnd)
    # 获取父窗口
    parent = win32gui.GetParent(hwnd)
    # 获取样式
    style = win32gui.GetWindowLong(hwnd, win32con.GWL_STYLE)
    return {
        'handle': hwnd,
        'class': class_name,
        'text': text,
        'parent': parent,
        'style': style
    }

9.2 消息监控

实时监控发送到指定窗口的消息：

python复制def window_proc(hwnd, msg, wParam, lParam):
    print(f"消息: {msg}, wParam: {wParam}, lParam: {lParam}")
    return win32gui.CallWindowProc(old_proc, hwnd, msg, wParam, lParam)

# 替换窗口过程
old_proc = win32gui.SetWindowLong(
    hwnd,
    win32con.GWL_WNDPROC,
    window_proc
)

9.3 内存查看

查看进程内存信息：

python复制import win32process

# 获取进程内存信息
process = win32api.GetCurrentProcess()
mem_info = win32process.GetProcessMemoryInfo(process)
print(f"工作集大小: {mem_info['WorkingSetSize']/1024/1024:.2f} MB")