python – 单击链接时停止打开新窗口的selenium

我将我的selenium测试导出到python并使用shell运行测试.我到达某个点,我点击一个链接,然后网络驱动程序打开一个全新的Firefox窗口然后我收到一个错误,说驱动程序无法在页面上找到输入选项.我认为问题在于,当它打开新窗口时,驱动程序没有在新窗口上运行,这就是它无法找到输入选项的原因.
如何让脚本保持在同一窗口或将其焦点切换到新窗口？

谢谢！

解决方法:

我不知道如何在python中做到这一点,但应该有一个函数来切换到一个新窗口.

在Java中,我这样做：

Set<String> availableWindows = driver.getwindowHandles();
for (String windowHandle : availableWindows) {
    driver.switchTo().window(windowHandle);
    if (getTitle().contains(TITLE_TO_MATCH)){
           driver.manage().window().maximize();
            return;
    }
}

(函数想要的是driver.switchTo().window(NameOrHandle))