selenium python 百度网盘的登陆与退出

# -*- coding:utf-8 -*-
from selenium import webdriver
from selenium.webdriver.common.action_chains import ActionChains
from selenium.webdriver.support.ui import WebDriverWait

import time

driver = webdriver.Firefox(executable_path='/Users/huiliang/Downloads/geckodriver')
driver.get("http://pan.baidu.com/")
driver.implicitly_wait(10)

#登陆百度网盘
driver.find_element_by_xpath('//*[@id="login-middle"]/div/div[6]/div[2]/a').click()
driver.find_element_by_id('TANGRAM__PSP_4__userName').clear()
driver.find_element_by_id('TANGRAM__PSP_4__userName').send_keys('xxx')
driver.find_element_by_id('TANGRAM__PSP_4__password').clear()
driver.find_element_by_id('TANGRAM__PSP_4__password').send_keys('ooo')
driver.find_element_by_id('TANGRAM__PSP_4__submit').click()

#确认是否登陆成功
test = driver.find_element_by_xpath('//*[@id="dynamicLayout_0"]/div/div/dl/dd[2]/span/span[2]').text
print test
if test == 'xxx':
    print u'登陆成功'

#退出百度网盘
ele = driver.find_element_by_xpath('//*[@id="dynamicLayout_0"]/div/div/dl/dd[2]/span/span[1]/i')
ActionChains(driver).move_to_element(ele).perform()
sub_ele = driver.find_element_by_link_text(u'退出')
sub_ele.click()
ele_out = driver.find_element_by_id('_disk_id_4')
ele_out.click()


time.sleep(5)
driver.quit()