在使用python的过程中,有时会遇到这种需求:开启了很多窗口,需要把焦点锁定到某个进程的窗口上,再对该窗口进行操作。例如:我们打开了notepad.exe和wordpad.exe应用程序,现在需要把焦点定位到notepad窗口上,在notepad窗口中进行输入。
参考代码如下,其原理为:通过获取进程的pid来遍历该进程下的所有窗口,由于notepad.exe只有一个窗口,所以可得到一个窗口的句柄。再调用win32gui的SetForegroundWindow(hwnd),把该hwnd置首。
1 import win32con 2 import win32gui 3 import win32process 4 5 def get_hwnds_for_pid (pid): 6 def callback (hwnd, hwnds): 7 if win32gui.IsWindowVisible (hwnd) and win32gui.IsWindowEnabled (hwnd): 8 _, found_pid = win32process.GetWindowThreadProcessId (hwnd) 9 if found_pid == pid: 10 hwnds.append (hwnd) 11 return True 12 hwnds = [] 13 win32gui.EnumWindows (callback, hwnds) 14 return hwnds 15 16 if __name__ == '__main__': 17 import subprocess 18 import time 19 notepad = subprocess.Popen ([r"notepad.exe"]) 20 # 21 # sleep to give the window time to appear 22 # 23 time.sleep (2.0) 24 print notepad.pid 25 for hwnd in get_hwnds_for_pid (notepad.pid): 26 print hwnd, "=>", win32gui.GetWindowText (hwnd) 27 win32gui.SetForegroundWindow(hwnd)
我们使用pywinauto库,其中的HwndWrapper类中有SetFocus函数,该函数在实践过程中有些情况下会失效。可修改SetFocus函数,在其开始处先尝试执行如下代码:
1 ctypes.windll.user32.SwitchToThisWindow(self.handle,True) 2 win32functions.SetFocus(self)
如果发生异常,再继续执行原来的内容。