有关处理三个三个iframe并来回切换的建议

Question

我目前正在开展一个项目，我必须在每个步骤之后切换多个iframe，并且我很难维护可重用的功能。每个其他步骤都可以有不同的Iframe。

我尝试使用try和除了iframe，但我认为这不是一个好主意。

所以在点击任何按钮之前我都有这样的东西。注意 - 可能存在没有Iframe的情况。

try:
    self.switchToFrame(*MainPageLocatars.FRAMEONE)
    self.switchToFrame(*MainPageLocatars.SUBLISTFRAME)
except:
    pass

try:
    self.switchToFrame(*MainPageLocatars.FRAMEONE)
except:
    pass

有可能改变它吗？并在函数中定义它，只需调用函数，它将变为正确的框架？

WebElement frame = driver.findElement(By.Id("frame1"));
 //  or
WebElement frame = driver.findElement(By.Id("frame2"));
// or 
WebElement frame = driver.findElement(By.Id("frame3"))

//Now we will switch focus,
driver.switchTo().frame("frame");