电子活页4-48文件testpandemo3.py的完整代码

文件testpandemo3.py的完整代码fromseleniumimportwebdriverfromselenium.webdriver.support.uiimportWebDriverWaitfrommon.action_chainsimportActionChainsimporttimeimportxlrdimportxlsxwriterfrommon.byimportByclassget_web_info(object):def__init__(self,index,path=''):self.xl=xlrd.open_workbook(path)self.sheet=self.xl.sheet_by_index(index)defget_sheet_info(self):info_list={}forrowinrange(self.sheet.nrows):result=[valforvalinself.sheet.row_values(row)]temp=dict([result])info_list.update(temp)returninfo_listclassget_user_info(object):"""docstringforget_user_info"""def__init__(self,index,path=''):self.xl=xlrd.open_workbook(path)self.sheet=self.xl.sheet_by_index(index)defget_sheet_info(self):info_list=[]info_dict={}listkey=['account','password']forrowinrange(1,self.sheet.nrows):result=[valforvalinself.sheet.row_values(row)]info_dict=dict(zip(listkey,result))info_list.append(info_dict)returninfo_listclassset_login_result(object):"""docstringforset_login_result"""def__init__(self,sheetname):fname=time.strftime("%Y-%m-%d",time.gmtime())self.row=1self.xl=xlsxwriter.Workbook(fname+'.xlsx')self.sheet=self.xl.add_worksheet(sheetname)self.style=self.xl.add_format({'bg_color':'#DC143C'})self.otherstyle=self.xl.add_format({'bg_color':'green'})self.sheet.set_column('A:D',40)defxl_title(self,*title):row=0col=0forvalintitle:self.sheet.write_string(row,col,val)col+=1defxl_write(self,*args):col=0style=''if'Error'inargs:style=self.styleif'Pass'inargs:style=self.otherstyleforvalinargs:self.sheet.write_string(self.row,col,val,style)col+=1self.row+=1defxl_close(self):self.xl.close()defget_ele_times(driver,times,func):returnWebDriverWait(driver,times,1).until(func)defopenBrowser():browser=webdriver.Chrome()returnbrowserdefopenUrl(driver,url):driver.get(url)time.sleep(1.5)driver.maximize_window()deffindElements(driver,arg):if'way_id'inarg:ele_login_way=get_ele_times(driver,5,lambdax:x.find_element(By.CLASS_NAME,arg['way_id']))ele_login_way.click()#driver.find_element(By.CLASS_NAME,arg['way_id']).click()time.sleep(5)useEle=driver.find_element(By.ID,arg['account_id'])pwdEle=driver.find_element(By.ID,arg['pwd_id'])loginEle=driver.find_element(By.ID,arg['confirm_id'])returnuseEle,pwdEle,loginEledefsendVals(eletuple,arg):listkey=['account','password']i=0forkeyinlistkey:#eletuple[i].send_keys('')eletuple[i].clear()eletuple[i].send_keys(arg[key])i+=1time.sleep(2)eletuple[2].click()deflogout(driver,web_info):driver.switch_to.window(driver.window_handles[0])time.sleep(5)arrow=driver.find_element(By.XPATH,web_info['arrow_xpath'])ActionChains(driver).move_to_element(arrow).perform()#设置元素为可见js1='document.getElementsByClassName("wp-s-header-user__body-userinfo")[0].style.display="block";'driver.execute_script(js1)time.sleep(2)driver.find_element(By.LINK_TEXT,web_info['logout_text']).click()#设置元素为可见js2='document.getElementsByClassName("u-dialog__wrapper")[0].style.display="block";'driver.execute_script(js2)time.sleep(5)driver.find_element(By.XPATH,web_info['confirm_xpath']).click()defcheckResults(driver,errid,arg,log):result=Falsetime.sleep(2)try:err=driver.find_element(By.ID,errid)print("用户名或密码有误!")log.xl_write(arg['account'],arg['password'],'Error',err.text)except:print("用户名与密码正确!")log.xl_write(arg['account'],arg['password'],'Pass','账号密码正确')result=Truereturnresultdeflogin_test(web_info,user_info):browser=openBrowser()openUrl(browser,web_info['url'])log=set_login_result('sheet1')log.xl_title('account','password','result','massage')ele_tuple=findElements(browser,web_info)forarginuser_info:sendVals(ele_tuple,arg)result=checkResults(browser,web_info['error_id'],arg,log)#result=1ifresult:time.sleep(3)logout(browser,web_info)ele_tuple=findElements(browser,web_info)log.xl_close()if__name__=='