selenum自动化实现查看小说内容

from selenium import webdriver
import time

url = 'https://www.xbiquge.la/'

# header = {
#     'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/98.0.4758.102 Safari/537.36'
# }
read = input('请输入需要查看的小说:')
path = 'chromedriver.exe'
drive = webdriver.Chrome(path)
drive.get(url)

text = drive.find_element_by_id('wd')
text.send_keys(read)
time.sleep(2)
button = drive.find_element_by_id('sss')

button.click()
time.sleep(2)

# button = drive.find_element_by_link_text('盗墓笔记')
# button.click()
# time.sleep(3)
url = drive.find_element_by_xpath('//div[@id="content"]//a').get_attribute('href')

drive.get(url)

url = drive.find_element_by_xpath('//div[@id="list"]//a').get_attribute('href')
drive.get(url)
time.sleep(3)
drive.close()
print('阅读完毕')