#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()
