【python selenium】driver.switch_to.frame(iframe)の後、元の画面に戻りたい

Question

python, seleniumでスクレイピングをしています。
あるサイトで、サイト内のあるボタンをクリックすると
iframeのページが召喚されて、GUI上では小窓が出てきます。

driver.switch_to.frame(iframe)

このコマンドを使えばiframeにスイッチして、
操作できるところまでは分かりました。

その後、この小窓内であるボタンをクリックすると、
サイト側である処理が走り、同時に小窓が消滅するのですが、
その後も元のページに戻って処理を続けたいです。

# 元ページにてiframeの画面を召喚する
driver.find_element_by_class_name('hoge').click()

# iframe画面にスイッチする
driver.switch_to.frame(driver.find_element_by_class_name('fuga'))

# iframe画面で処理を行う（処理と同時にiframe画面が消滅する）
driver.find_element_by_class_name('hogehoge').click()

# 元ページ画面にスイッチする
XXXXXXXXXXXXXXXXXX

# 元ページにて別の処理を行う
for i in range(10):
    ...

このXXXXXXXXXXXXXXXXXXはどう記述すればよいでしょうか。

Answer 1

driver.switch_to().parent_frame()

Comments

[shirai]

ご回答頂いたコマンドでググって
driver.switch_to.default_content()で行けました！
前後が重くなるのでsleepが必要でした。
ありがとうございました。