from selenium.webdriver import Chrome
from selenium.webdriver.common.by import By
from selenium.webdriver.common.keys import Keys
import time

with Chrome() as driver:
    # 開啟搜狗網站
    driver.get('https://www.sogou.com/')
    # 獲取輸入框
    # 用標籤ID "query"獲取輸入框
    input_keys = driver.find_element(By.ID, '
 
query')
    # 輸入框輸入搜尋內容'筆記本'
    input_keys.send_keys('筆記本')
    # 延遲兩秒
    time.sleep(2)
    # 按下回車鍵搜尋
    input_keys.send_keys(Keys.ENTER)
    # 延遲兩秒
    time.sleep(2)
    # 用XPath表示式獲取包含推廣公司的dom節點. |在XPath表示式中是或的意思
    # 包含推廣公司的dom節點有cite標籤和類名class為"cite"的a標籤
    list_extension = driver.find_elements(By.XPATH, '
 
//cite | //a[@class="cite"]')
    # 遍歷,將查詢到的公司放入列表中
    list_company = []
    for item in list_extension:
        list_company.append(item.text)

    print(list_company)