【Python】【pywin32】【指定窗口截图】

背景

最近对python pywin32模块感兴趣，且想实现一个能够截取指定窗口图片的小功能。在各大论坛遨游一番后，有所收获，记录下来帮主其它有需要的同学。

涉及到的模块，请自行安装。

源码

import os
import random
import time
# import subprocess
import cv2
import numpy as np
import win32api, win32gui, win32ui, win32con
from PIL import Image,ImageGrab


# parent为父窗口句柄id
def get_child_windows(parent):

    hwndChildList = []
    win32gui.EnumChildWindows(parent, lambda hwnd, param: param.append(hwnd),  hwndChildList)
    return hwndChildList

# 通过窗口句柄截取当前句柄图片 返回cv2格式的Mat数据
def window_capture(hwnd, picture_name=None):
    x1, y1, x2, y2 = win32gui.GetWindowRect(hwnd)  # 获取当前窗口大小
    hwndDC = win32gui.GetWindowDC(hwnd)  # 通过应用窗口句柄获得窗口DC
    mfcDC = win32ui.CreateDCFromHandle(hwndDC)  # 通过hwndDC获得mfcDC(注意主窗口用的是win32gui库，操作位图截图是用win32ui库)
    neicunDC = mfcDC.CreateCompatibleDC()  # 创建兼容DC，实际在内存开辟空间（ 将位图BitBlt至屏幕缓冲区（内存），而不是将屏幕缓冲区替换成自己的位图。同时解决绘图闪烁等问题）
    savebitmap = win32ui.CreateBitmap()  # 创建位图
    width = x2 - x1
    height = y2 - y1
    savebitmap.CreateCompatibleBitmap(mfcDC, width, height)  # 设置位图的大小以及内容
    neicunDC.SelectObject(savebitmap)  # 将位图放置在兼容DC，即 将位图数据放置在刚开辟的内存里
    neicunDC.BitBlt((0, 0), (width, height), mfcDC, (0, 0), win32con.SRCCOPY)  # 截取位图部分，并将截图保存在剪贴板
    if picture_name is not None:
        savebitmap.SaveBitmapFile(neicunDC, picture_name)  # 将截图数据从剪贴板中取出，并保存为bmp图片
    img_buf = savebitmap.GetBitmapBits(True)

    img = np.frombuffer(img_buf, dtype="uint8")
    img.shape = (height, width, 4)
    # mat_img = cv2.cvtColor(img, cv2.COLOR_RGB2RGBA)  # 转换RGB顺序

    # cv2.imshow('baofeng', img)
    # cv2.waitKey()

    # 释放内存
    win32gui.DeleteObject(savebitmap.GetHandle())
    neicunDC.DeleteDC()
    mfcDC.DeleteDC()
    win32gui.ReleaseDC(hwnd, hwndDC)

    return img


if __name__ == '__main__':
	# 获取应用句柄
    hwnd = win32gui.FindWindow(None, '雷电模拟器-0')
    sub_hwnd = get_child_windows(hwnd)
    print(sub_hwnd)

    hld = sub_hwnd[0]
    print(hld)
    print(win32gui.GetWindowRect(hld))  # 获取窗口位置

    img = window_capture(hld)

    cv2.imshow('demo', img)
    cv2.waitKey()

参考链接

https://blog.csdn.net/freeking101/article/details/88231952
https://dandelioncloud.cn/article/details/1408092175489789954/