Selenium+python complete Baidu search operation

import time
from selenium import webdriver
driver=webdriver.Chrome()
# Hidden waiting 
driver.implicitly_wait(10)
driver.get('https://baidu.com/')
# Intercept Baidu button 
btn1=driver.find_element_by_id('su')
btn1.screenshot(r'd:\baidu.png')
input1=driver.find_element_by_id('kw')
# Input qq mailbox 
input1.send_keys('qq mailbox ')
btn=driver.find_element_by_id('su')
btn.click()
# Get Baidu QQ The first email result 
res=driver.find_element_by_id('1').find_element_by_tag_name('a')
print(res.text)
# Save the start interface handles
baidu_handle = driver.current_window_handle
if ' Sign in QQ mailbox ' in res.text:
print('pass')
res.click()
# Screenshot 
#driver.get_screenshot_as_file(r'd:\123.png')
# Switch to webdriver To a new interface ...................
for handle in driver.window_handles:
driver.switch_to.window(handle)
# Determine page title 
if ' Sign in QQ mailbox ' in driver.title:
break
else:
print('fail')
print(res.text)
# Switch back to Baidu page 
driver.switch_to.window(baidu_handle)
time.sleep(3)
driver.refresh()