no such elementが出たときにページが移行できるプログラム

Question

現在、seleniumを使ってページの移行について勉強しています。

以下、エラー文です。
selenium.common.exceptions.NoSuchElementException: Message: no such element: Unable to locate element:

この原因はわかっていますが、一方でこれを利用したif文を作成したいです。
つまり、

if(NoSuchElementExceptionの場合):
別のページに移行する

というようなプログラムです。

このようにするにはどのようにすればいいでしょうか？

Answer 1

こんな感じです。

from selenium.common.exceptions import NoSuchElementException

try:
    # 要素を探すコード
except NoSuchElementException:
    # 別のページに移行するコード

Answer 2

私も初心者ですが、try/exceptを利用することが多いです。

# 例外処理用のlibraryをimport
from selenium.common.exceptions import NoSuchElementException

try:
 実行したい処理

except NoSuchElementException:
 上記が実行できなかった場合の処理
 break

# if文を利用した場合の例：
# 特定の要素の有無で条件分岐

if len(driver.find_elements_by_class_name('hoge')) > 0:
    print('Yes')
else:
    print('No')

回答になっていなかったら申し訳ございません。。。。。