Python如何捕获windows窗口

Python可以通过使用Win32 API、PyAutoGUI库、OpenCV库等来捕获Windows窗口。在这些方法中，Win32 API提供了强大的功能和灵活性，但需要更多的编程经验；PyAutoGUI是一个简单易用的自动化工具，适合初学者使用；OpenCV可以进行图像处理和更高级的图像捕获操作。接下来，将详细介绍使用Win32 API进行窗口捕获的方法。

一、使用WIN32 API进行窗口捕获

Win32 API提供了对Windows系统底层操作的访问，使得捕获窗口变得可行。通过使用Win32 API，您可以精确地控制窗口捕获的过程。

1、安装PyWin32库

要使用Win32 API，需要安装PyWin32库。您可以通过以下命令安装：

pip install pywin32

2、获取窗口句柄

首先，需要获取要捕获的窗口的句柄。句柄是一个整数值，代表一个窗口的唯一标识符。可以通过窗口标题来获取：

import win32gui
def get_window_handle(window_title):
    return win32gui.FindWindow(None, window_title)

3、捕获窗口图像

一旦获得窗口句柄，就可以使用GetWindowRect函数获取窗口的位置和大小，并使用BitBlt函数捕获窗口图像：

import win32gui
import win32ui
import win32con
import win32api
def capture_window(window_handle):
    # 获取窗口位置
    left, top, right, bottom = win32gui.GetWindowRect(window_handle)
    width = right - left
    height = bottom - top
    # 获取窗口设备上下文
    hwndDC = win32gui.GetWindowDC(window_handle)
    mfcDC = win32ui.CreateDCFromHandle(hwndDC)
    saveDC = mfcDC.CreateCompatibleDC()
    # 创建位图对象
    saveBitMap = win32ui.CreateBitmap()
    saveBitMap.CreateCompatibleBitmap(mfcDC, width, height)
    saveDC.SelectObject(saveBitMap)
    # 捕获窗口
    win32gui.SetForegroundWindow(window_handle)
    saveDC.BitBlt((0, 0), (width, height), mfcDC, (0, 0), win32con.SRCCOPY)
    # 保存到文件
    saveBitMap.SaveBitmapFile(saveDC, 'captured_window.bmp')
    # 释放设备上下文
    win32gui.DeleteObject(saveBitMap.GetHandle())
    saveDC.DeleteDC()
    mfcDC.DeleteDC()
    win32gui.ReleaseDC(window_handle, hwndDC)

二、使用PYAUTOGUI库进行窗口捕获

PyAutoGUI是一个跨平台的GUI自动化库，可以轻松实现窗口截图。

1、安装PyAutoGUI库

您可以使用以下命令安装PyAutoGUI：

pip install pyautogui

2、捕获屏幕截图

PyAutoGUI提供了简单的接口来捕获屏幕截图：

import pyautogui
def capture_screenshot():
    screenshot = pyautogui.screenshot()
    screenshot.save('screenshot.png')

3、裁剪窗口截图

如果只想要特定窗口的截图，可以通过Pillow库裁剪：

from PIL import Image
def crop_window_screenshot(left, top, right, bottom):
    screenshot = pyautogui.screenshot()
    window_screenshot = screenshot.crop((left, top, right, bottom))
    window_screenshot.save('window_screenshot.png')

三、使用OPENCV库进行窗口捕获

OpenCV是一个强大的计算机视觉库，适用于更多复杂的图像处理任务。

1、安装OpenCV库

使用以下命令安装OpenCV：

pip install opencv-python

2、结合Win32 API和OpenCV

可以使用Win32 API获取窗口图像数据，然后用OpenCV处理：

import cv2
import numpy as np
import win32gui
import win32ui
import win32con
import win32api
def capture_window_with_opencv(window_handle):
    left, top, right, bottom = win32gui.GetWindowRect(window_handle)
    width = right - left
    height = bottom - top
    hwndDC = win32gui.GetWindowDC(window_handle)
    mfcDC = win32ui.CreateDCFromHandle(hwndDC)
    saveDC = mfcDC.CreateCompatibleDC()
    saveBitMap = win32ui.CreateBitmap()
    saveBitMap.CreateCompatibleBitmap(mfcDC, width, height)
    saveDC.SelectObject(saveBitMap)
    win32gui.SetForegroundWindow(window_handle)
    saveDC.BitBlt((0, 0), (width, height), mfcDC, (0, 0), win32con.SRCCOPY)
    bmpinfo = saveBitMap.GetInfo()
    bmpstr = saveBitMap.GetBitmapBits(True)
    img = np.frombuffer(bmpstr, dtype='uint8')
    img.shape = (height, width, 4)
    cv2.imwrite('opencv_window.png', cv2.cvtColor(img, cv2.COLOR_BGRA2BGR))
    win32gui.DeleteObject(saveBitMap.GetHandle())
    saveDC.DeleteDC()
    mfcDC.DeleteDC()
    win32gui.ReleaseDC(window_handle, hwndDC)