成语接龙.py

from selenium import webdriver
from selenium.webdriver.common.desired_capabilities import DesiredCapabilities
from selenium.webdriver.support.ui import WebDriverWait
from selenium.webdriver.support import expected_conditions as EC
from selenium.webdriver.common.by import By
from xpinyin import Pinyin
from lxml import etree
import time

p = Pinyin()
chrome_options = webdriver.ChromeOptions()  # 无头模式
chrome_options.add_argument("--headless")
desired_capabilities = DesiredCapabilities.CHROME  # 懒加载模式
desired_capabilities["pageLoadStrategy"] = "none"
driver = Chrome(chrome_options=chrome_options)
wait = WebDriverWait(driver, 3)

while True:
    inp = str(input('请输入：'))
    # if '\u4e00' <= inp[-1:] <= '\u9fff':
    #     inp=p.get_pinyin(inp[-1:], splitter='', convert='capitalize')
    # try:
    driver.get('http://chengyujielong.00cha.net/index.asp')
    wait.until(EC.presence_of_element_located((By.XPATH, '//*[@id="xiaodiv"]/div[4]/form/input[2]')))
    driver.execute_script("window.stop();")
    driver.find_elements_by_xpath('//*[@id="xiaodiv"]/div[4]/form/input[1]')[0].clear()
    driver.find_elements_by_xpath('//*[@id="xiaodiv"]/div[4]/form/input[1]')[0].send_keys(inp)
    driver.find_elements_by_xpath('//*[@id="xiaodiv"]/div[4]/form/input[2]')[0].click()
    time.sleep(1)
    html1 = etree.HTML(driver.page_source)
    result = html1.xpath('//*[@id="xiaodiv"]/div[5]/a')
    for i in result:
        if i.text != '→':
            print(i.text)
    # except:
    #     print('未查询到')