【Selenium】find_elementsを使って要素のあるなしでtry文の分岐をしたい

Question

###したいこと
検索をして、結果のあるなしで分岐をしようと考えています。
ヒット件数
0 : 検索クエリを変更
1 : クリック
2以上 : 絞り込む処理
###コード

   for title, author in zip(title_query, author_query):
    query = title+' '+author
    driver.get(url+query)

    WebDriverWait(driver, 10).until(
        EC.presence_of_all_elements_located  # 待機処理
    )

    link_list = []

    try:
        items = driver.find_elements_by_class_name('preview_data')  # ヒットするか
        if len(items) == 1:  # ヒットが一つだった場合  # hrefを取得
            links = driver.find_elements(By.CSS_SELECTOR, '.preview_container'
                                         ' > div a').get_attribute("href")
            link_list.append(links)
        elif len(items) == 0:  # ヒット0のとき例外発生
            raise Exception
        else:  # ヒット2以上
            # 絞り込み
    except Exception:  # ヒットなしのときに検索条件を変える
        traceback.print_exc()
        query = author
        driver.get(url+query)

    time.sleep(5)

###問題点
ヒット件数が一つだった場合もexceptの処理が行われています。
###環境
python3.7.4

プログラミング初心者です。もし、より良い書き方などもありましたらご教授していただければ幸いです。

Comments

[MiekoFuruhashi]

おそらく、
driver.find_elements(By.CSS_SELECTOR, '.preview_container'
' > div a').get_attribute("href")
がexception を出しているのでは？
`traceback.print_exc()`の内容をコピペしてください。

[__PORNO___]

これも、不思議に思っていたのですが、なぜかtraceback.print_exc()の出力はありません。もしかしてfor文はループが終わってから一気に、今までため込んだものを出力するような挙動でしたっけ？

[t_obara]

ヒット件数が一件だったことをどのように確認しているのですか？

[__PORNO___]

forの外にdriver.find_elements(By.CSS_SELECTOR, '.preview_container'
' > div a').get_attribute("href")を出して見たところ、AttributeError: 'list' object has no attribute 'get_attribute'となりました。多分、forでlistを回せば解決するかなと思うので、試してみます。

[__PORNO___]

items = driver.find_elements_by_class_name('preview_data')では、検索結果についているclass名 preview_dataをもつ要素の数でヒット件数を確認する構造になっているかと思います。

[__PORNO___]

おそらく解決しました。コードを少し書き換えて解決方法のほうに置いておきます。
ご助言ありがとうございました！助かりました。

Answer 1

###原因
driver.find_elements(By.CSS_SELECTOR, '.preview_container'
' > div a').get_attribute("href")が、AttributeError: 'list' object has no attribute 'get_attribute'を出していたので毎回Exceptが実行されていました。

###解決方法
for文でlinksを回しそれぞれの要素にget_attribute('href')をしました。

###コード
具体的にはtryの部分を以下のようにすることで解決しました。

try:
    items = driver.find_elements_by_class_name('preview_data')  # ヒットするか
    if len(items) == 1:  # ヒットが一つだった場合  # hrefを取得
        links = driver.find_elements(By.CSS_SELECTOR,
                                     '.preview_container > div a')
        for link in links:
            url = link.get_attribute('href')
            link_group.append(url)  # urlのリストをつくる
    elif len(items) == 0:  # ヒット0のとき例外発生
        raise Exception
    else:  # ヒット2以上
        # 絞り込み