Python使用mss获取窗口图片的方法

pip install mss pywin32

import win32gui
from mss import mss

# 获取窗口句柄
hwnd = win32gui.FindWindow(None, "记事本")  # 替换为实际窗口标题
if hwnd == 0:
    raise Exception("未找到目标窗口")

# 获取窗口客户区坐标（排除标题栏/边框）
left, top, right, bot = win32gui.GetClientRect(hwnd)
win32gui.ClientToScreen(hwnd, (left, top))  # 转换为屏幕绝对坐标

with mss() as sct:
    # 定义截图区域（精确到客户区）
    monitor = {
        "left": left,
        "top": top,
        "width": right - left,
        "height": bot - top
    }
    
    # 捕获屏幕区域
    sct_img = sct.grab(monitor)
    
    # 保存为PNG（自动处理BGRA转RGB）
    sct_img = sct_img.convert("RGB")
    sct_img.save("window_mss.png")

from ctypes import windll
dpi = windll.user32.GetDpiForWindow(hwnd)
scale = dpi / 96.0  # 96为基准DPI
adjusted_width = int((right - left) * scale)

monitor_info = {}
ctypes.windll.user32.GetMonitorInfoA(ctypes.windll.user32.MonitorFromWindow(hwnd, 1), ctypes.byref(monitor_info))
print("显示器工作区:", monitor_info["rcWork"])

import win32gui
from mss import mss
import ctypes

# 获取窗口句柄
hwnd = win32gui.FindWindow(None, "记事本")
if hwnd == 0:
    raise Exception("未找到记事本窗口")

# 获取客户区坐标
left, top, right, bot = win32gui.GetClientRect(hwnd)
win32gui.ClientToScreen(hwnd, (left, top))

# 处理DPI缩放
dpi = ctypes.windll.user32.GetDpiForWindow(hwnd)
scale = dpi / 96.0
adjusted_left = int(left * scale)
adjusted_top = int(top * scale)
width = int((right - left) * scale)
height = int((bot - top) * scale)

# 使用mss截图
with mss() as sct:
    monitor = {
        "left": adjusted_left,
        "top": adjusted_top,
        "width": width,
        "height": height
    }
    sct_img = sct.grab(monitor)
    sct_img = sct_img.convert("RGB")  # 转换颜色格式
    sct_img.save("corrected_window.png")

print("截图已保存，尺寸：{}x{}".format(width, height))

# macOS示例（通过Quartz获取窗口ID）
from Quartz import CGWindowListCopyWindowInfo
windows = CGWindowListCopyWindowInfo(kCGWindowListOptionAll)
target_window = [w for w in windows if "记事本" in w.get("kCGWindowName", "")][0]
bounds = target_window["kCGWindowBounds"]