selenium之实现自动登录的实例代码

#coding=utf-8from selenium import webdriverfrom selenium.common.exceptions import NoSuchElementExceptionfrom selenium.webdriver.common.keys import Keysfrom selenium.webdriver.support.ui import WebDriverWaitimport time ,sys, requests, jsonclass Browser(object):def __init__(self,url):        self.browser = webdriver.Chrome()        self.browser.maximize_window()        self.url = url        self.browser.get(self.url)def get_element_by_id(self,element_id):        return self.browser.find_element_by_id(element_id)def get_element_by_class(self,_class):return self.browser.find_element_by_class_name(_class)def get_element_by_text(self,text):return self.browser.find_element_by_link_text(text)def get_screenshot(self,file):return self.browser.get_screenshot_as_file(file)    def waitfor(self,element_id):return WebDriverWait(self.browser,10).until(lambda b:b.find_element_by_id(element_id).is_displayed())def upload_code_check():    url = ':5678/api.php?method=upload'print url    data = {}    data.update({'username':'xxxxx'})    data.update({'password':'xxxxx'})    data.update({'codetype':'5000'})    data.update({'appid':'1'})    data.update({'appkey':'22cc5376925e9387a23cf797cb9ba745'})    data.update({'timeout':'60'})    data.update({'version':'YAPI/WEB v1.0.0'})    data.update({'showimage':'1'})    files = {'file':('dd.jpg',open('C://dd.jpg','rb'),'image/jpeg')}    res = requests.post(url,data=data,files=files)    res_json = json.loads(res.content)print res.contentreturn res_json     def get_code_check_results(url1):while True:        res1 = requests.get(url1)        res1 = json.loads(res1.content)if res1['ret'] == 0:breakreturn res1def get_image(img_url):    img = requests.get(img_url)    img_f = open('C://dd.jpg','w+b')    img_f.write(img.content)    img_f.close()return img              def main():    url = ''count = 'xxxxxxx'pwd = 'xxxxxxxx'pwd_new = 'xxxxxxx'#email_new = 'xxxxx@qq.com'email_new = 'xxxxxxxx@qq.com'browser = Browser(url)if browser.waitfor('nav-link-yourAccount'):        browser.get_element_by_id('nav-link-yourAccount').click()else:print 'connect timeout'sys.exit(1)if browser.waitfor('ap_email'):        browser.get_element_by_id('ap_email').send_keys(count)        browser.get_element_by_id('ap_password').send_keys(pwd)        browser.get_element_by_id('signInSubmit').click()try:            error = browser.get_element_by_id('auth-error-message-box')print 'login error'except Exception:passelse:print 'connect timeout'sys.exit(1)if browser.waitfor('nav-link-yourAccount'):        browser.get_element_by_id('nav-link-yourAccount').click()else:print 'connect timeout'sys.exit(1)try:        text = browser.get_element_by_text(u'登录和安全设置')        text.click()        email = browser.get_element_by_id('auth-cnep-edit-email-button')        email.click()if browser.waitfor('ap_email_new'):            browser.get_element_by_id('ap_email_new').send_keys(email_new)            browser.get_element_by_id('ap_email_new_check').send_keys(email_new)            browser.get_element_by_id('ap_password').send_keys(pwd)                        yanzhenma = browser.get_element_by_id('auth-captcha-image')            img_url = yanzhenma.get_attribute('src')            img_url = img_url.replace('&','&')print img_url            img = get_image(img_url)            res_json = upload_code_check()                       url1 = ":5678/api.php?method=result&cid=" + str(res_json['cid'])print url1            res1 = get_code_check_results(url1)print res1['text']            yzm = res1['text']print yzmif yzm:                browser.get_element_by_id('auth-captcha-guess').send_keys(str(yzm))                browser.get_element_by_id('cnep_1B_submit_button').click()if browser.waitfor('auth-success-message-box'):                browser.get_element_by_id('auth-cnep-edit-password-button').click()print 'success modify email'if browser.waitfor('ap_password_new_check'):                browser.get_element_by_id('ap_password').send_keys(pwd)                browser.get_element_by_id('ap_password_new').send_keys(pwd_new)                browser.get_element_by_id('ap_password_new_check').send_keys(pwd_new)                auth_img_url = browser.get_element_by_id('auth-captcha-image')                img_url2 = auth_img_url.get_attribute('src').replace('&','&')                img2 = get_image(img_url2)                res_json = upload_code_check()                url2 = ":5678/api.php?method=result&cid=" + str(res_json['cid'])print url2                res2 = get_code_check_results(url2)print res2['text']                yzm2 = res2['text']print yzm2if yzm2:                    browser.get_element_by_id('auth-captcha-guess').send_keys(str(yzm2))                    browser.get_element_by_id('cnep_1D_submit_button').click()             except NoSuchElementException,e:print(e)    if __name__ == "__main__":    main()