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 ([r"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函数，在其开始处先尝试执行如下代码：

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

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