python 根据程序名称获取窗口句柄

原创

mob64ca12d80f3a 2024-09-21 06:26:47 ©著作权

文章标签 窗口句柄代码示例 Python 文章分类 Python 后端开发

©著作权归作者所有：来自51CTO博客作者mob64ca12d80f3a的原创作品，请联系作者获取转载授权，否则将追究法律责任

用Python根据程序名称获取窗口句柄

在开发桌面应用程序或进行自动化测试时，获取特定窗口的句柄（Handle）是非常重要的。窗口句柄是操作系统用于识别窗口的唯一标识符。本文将介绍如何根据程序名称使用Python获取窗口句柄，并提供代码示例。我们还将用Mermaid语法展示旅行图和关系图，为读者提供更加直观的理解。

环境准备

首先，我们需要确保系统已安装Python，且安装了pywin32库。pywin32提供了对Windows API的访问，从而使我们能够与Windows操作系统进行互动。可以通过以下命令安装该库：

pip install pywin32

此外，为了更好地理解，我们将使用 psutil 库来帮助我们列出正在运行的进程。使用以下命令安装：

pip install psutil

执行步骤

获取窗口句柄的步骤如下：

获取所有进程的名称。
查找与目标程序名称匹配的进程。
获取该进程的窗口句柄。

下面是详细的代码示例：

代码示例

import psutil
import win32gui
import win32process

def get_window_handle_by_process_name(process_name):
    # 1. 遍历所有进程
    for proc in psutil.process_iter(attrs=['pid', 'name']):
        try:
            # 2. 检查是否与目标程序名称匹配
            if process_name.lower() in proc.info['name'].lower():
                # 3. 使用win32process获取窗口句柄
                hwnds = []
                for hwnd in win32gui.Windows():
                    _, pid = win32process.GetWindowThreadProcessId(hwnd)
                    if pid == proc.info['pid']:
                        hwnds.append(hwnd)
                return hwnds
        except (psutil.NoSuchProcess, psutil.AccessDenied):
            continue
    return None

# 示例使用
if __name__ == "__main__":
    process_name = "notepad.exe"  # 目标进程名称
    handles = get_window_handle_by_process_name(process_name)
    
    if handles:
        print(f"找到窗口句柄: {handles}")
    else:
        print(f"未找到运行中的程序: {process_name}")

代码解析

使用 psutil.process_iter() 方法遍历当前运行的所有进程。
proc.info 提供进程的 pid 和 name。
我们使用 win32gui.Windows() 获取所有窗口，依次检查它们的 pid 是否与目标进程的 pid 相同。
一旦找到匹配项，就返回该窗口的句柄。

旅行图

通过旅行图，我们可以可视化获取窗口句柄的整个过程：

journey
    title 获取窗口句柄旅行图
    section 遍历进程
      获取所有进程和名称: 5: 呼叫
    section 匹配进程
      查找与目标名称匹配的进程: 4: 呼叫
    section 获取句柄
      获取并返回窗口句柄: 5: 呼叫

关系图

在这里我们使用关系图展示进程、窗口和窗口句柄之间的关系：

erDiagram
    PROCESS {
        int pid
        string name
    }

    WINDOW {
        int hwnd
        string title
    }

    HANDLE {
        int handle_id
        string description
    }

    PROCESS ||--o{ WINDOW : contains
    WINDOW ||--o{ HANDLE : has

在这个关系图中，我们可以看到每个进程可以拥有多个窗口，而每个窗口又可以对应多个窗口句柄。这样一来，我们就能清楚地理解各个组件之间的关系。