selenium+python完整的百度搜索操作

import time
from selenium import webdriver
driver=webdriver.Chrome()
#隐藏式等待
driver.implicitly_wait(10)
driver.get('https://baidu.com/')
#截取百度按钮
btn1=driver.find_element_by_id('su')
btn1.screenshot(r'd:\baidu.png')
input1=driver.find_element_by_id('kw')
#输入qq邮箱
input1.send_keys('qq邮箱')
btn=driver.find_element_by_id('su')
btn.click()
#获取百度QQ邮箱结果第一条
res=driver.find_element_by_id('1').find_element_by_tag_name('a')
print(res.text)
# 保存开始界面的handles
baidu_handle = driver.current_window_handle
if '登录QQ邮箱' in res.text:
    print('pass')
    res.click()
    #截图
    #driver.get_screenshot_as_file(r'd:\123.png')
    #切换到webdriver到新的界面...................
    for handle in driver.window_handles:
        driver.switch_to.window(handle)
        #判断页面标题
        if '登录QQ邮箱' in driver.title:
            break
else:
    print('fail')
    print(res.text)
    #切回百度页面
    driver.switch_to.window(baidu_handle)
    time.sleep(3)
    driver.refresh()