Selenium Python iFrameがある時の処理方法

Seleniumを実行しているときiframeの扱いに迷った

iframe内の要素をクリックしたかったのだが、なぜか要素を取得できなかった。
そんな時の備忘録。

iframe内の要素をクリックしたい場合は、フレームを切り替える必要がある。

from selenium import webdriver

#ドライバー作成
driver = webdriver.Chrome(executable_path='chromedriver.exe')

#ここでフレーム要素を取得
frameElement = driver.find_element_by_id("dialog")

#ここでフレームを切り替える
driver.switch_to_frame(frameElement) 

##何らかの処理

#親フレームに戻す
driver.switch_to.default_content()