Python锁定焦点到指定进程窗口的参考方法

在使用python的过程中，有时会遇到这种需求：开启了很多窗口，需要把焦点锁定到某个进程的窗口上，再对该窗口进行操作。例如：我们打开了notepad.exe和wordpad.exe应用程序，现在需要把焦点定位到notepad窗口上，在notepad窗口中进行输入。

参考代码如下，其原理为：通过获取进程的pid来遍历该进程下的所有窗口，由于notepad.exe只有一个窗口，所以可得到一个窗口的句柄。再调用win32gui的SetForegroundWindow(hwnd)，把该hwnd置首。

import win32con

import win32gui

import win32process

def get_hwnds_for_pid (pid):

    def callback (hwnd, hwnds):

        if win32gui.IsWindowVisible (hwnd) and win32gui.IsWindowEnabled (hwnd):

            _, found_pid = win32process.GetWindowThreadProcessId (hwnd)

            if found_pid == pid:

                hwnds.append (hwnd)

            return True

    hwnds = []

    win32gui.EnumWindows (callback, hwnds)

    return hwnds

if __name__ == '__main__':

    import subprocess

    import time

    notepad = subprocess.Popen (["  notepad.exe"])

    #

    # sleep to give the window time to appear

    #

    time.sleep (2.0)

    print (notepad.pid)

    for hwnd in get_hwnds_for_pid (notepad.pid):

        print (hwnd, "=>", win32gui.GetWindowText (hwnd))

        win32gui.SetForegroundWindow(hwnd)

我们使用pywinauto库，其中的HwndWrapper类中有SetFocus函数，该函数在实践过程中有些情况下会失效。可修改SetFocus函数，在其开始处先尝试执行如下代码：

1ctypes.windll.user32.SwitchToThisWindow(self.handle,True)2win32functions.SetFocus(self)

如果发生异常，再继续执行原来的内容。