Python 关于使用三个iFrame并来回切换的建议

我目前正在从事一个项目，在这个项目中，我必须在每一步之后在多个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");