用于在python、selenium中切换帧的函数

我正在寻找一个可以更容易在两帧之间切换的函数。现在,每当我需要在帧之间切换时,我都会通过以下代码进行切换:

driver.switch_to.frame(driver.find_element_by_css_selector("frame[name='nav']"))

driver.switch_to.frame(driver.find_element_by_css_selector("frame[name='content']"))

我的目标是获得一个函数,该函数需要一个参数来改变导航或内容,因为其余的基本相同。

我已经尝试过的是:

def frame_switch(content_or_nav):
x = str(frame[name=str(content_or_nav)] #"frame[name='content_or_nav']"
driver.switch_to.frame(driver.find_element_by_css_selector(x))

但这给了我一个错误

 x = str(frame[name=str(content_or_nav)]
                  ^

语法错误:无效语法