#!/usr/bin/env python3 # -*- coding:utf-8 -*- from selenium import webdriver from selenium.webdriver.support.ui import WebDriverWait from selenium.webdriver.common.keys import Keys from selenium.webdriver.common.action_chains import ActionChains import time from userdata import get_webinfo, get_userinfo, XlUserInfo from log_module import Loginfo, Xlloginfo def get_ele_times(driver, times, func): return WebDriverWait(driver, times).until(func) def openBrower(): ''' :return: webdriver handle ''' webdriver_handle = webdriver.Firefox() return webdriver_handle # def openUrl(handle, url): # handle.get(url) # handle.maximize_window() def openUrl(handle, arg): handle.get(arg['url']) handle.maximize_window() def findElement(d, arg): ''' 1.test_id: 2.USERID 3.pwdid 4.loginid :param d: :param arg: :return: ''' if 'text_id' in arg: ele_login = get_ele_times(d, 10, lambda d: d.find_element_by_link_text(arg['text_id'])) ele_login.click() userEle = d.find_element_by_id(arg['userid']) pwdEle = d.find_element_by_id(arg['pwdid']) loginEle = d.find_element_by_id(arg['loginid']) # sign_out # signOutEle = d.find_element_by_id(arg['signOutid']) # print(signOutEle) return userEle, pwdEle, loginEle def findOutele(d, arg): try: signOutEle = d.find_element_by_id(arg['signOutid']) return signOutEle except: print("You are not login.") def sendVal(eletuple, arg): ''' :param eletuple: :param arg: :return: ''' list_key = ['uname','pwd'] i = 0 for key in list_key: eletuple[i].send_keys('') eletuple[i].clear() # print(arg[key]) eletuple[i].send_keys(arg[key]) i += 1 eletuple[2].click() def checkResult(d, err_id, arg, log): result = False time.sleep(3) try: err = d.find_element_by_id(err_id) print("Account and pwd error") # msg = "account:{},password:{}==>error:{}".format(arg['uname'],arg['pwd'], err.text) log.log_write(arg['uname'], arg['pwd'], 'Error', err.text) print(err.text) except: print("Account and pwd Right") # msg = "Account:{},password:{}==>pass!".format(arg['uname'], arg['pwd']) log.log_write(arg['uname'], arg['pwd'], 'Pass') result = True return result def logout(d, ele_dict): d.find_element_by_class_name(ele_dict['logout']).click() def login_test(ele_dict,user_list): d = openBrower() # log = Loginfo() log = Xlloginfo() log.log_init('sheet1', 'uname', 'pwd','result', 'message') openUrl(d, ele_dict) ele_tuple = findElement(d, ele_dict) # print(user_list[0]) for user_info in user_list: sendVal(ele_tuple, user_info) result = checkResult(d, ele_dict['errorid'], user_info, log) # sign_out # findOutele(d, ele_dict).click() if result: # logout logout(d, ele_dict) #login ele_tuple = findElement(d, ele_dict) else: ele_tuple = findElement(d, ele_dict) log.log_close() if __name__ == '__main__': ''' ele_dict = { 'url': 'http://www.maiziedu.com', 'uname':'maizi_test@139.com', 'pwd': 'abc123456', 'login_text':'Login', 'text_id': login_text, 'userid': 'id_account_l', 'pwdid': 'id_password_l', 'loginid': 'login_btn', 'signOutid':'sign_out', } user_list = [{'uname':account,'pwd':pwd},] ''' # file webinfo/userinfo get_webinfo(path),user_list = get ele_dict = get_webinfo(r'webinfo.txt') # user_list = get_userinfo(r'userinfo.txt') xinfo = XlUserInfo(r'userinfo.xlsx') user_list = xinfo.get_sheetinfo_by_index(0) login_test(ele_dict,user_list)
Log_Module
#!/usr/bin/env python3 # -*- coding:utf-8 -*- import time import xlsxwriter class Loginfo(object): def __init__(self, path = '', mode= 'a'): self.file_time = time.strftime('%Y-%m-%d', time.gmtime()) self.log_time = time.strftime('%H-%M-%S') fname = "{}{}".format(path, self.file_time) self.log = open("{}{}.txt".format(path, fname), mode) def log_init(self, sheetname, *title): pass def log_write(self, msg): self.log.write(" {}+{}".format(self.log_time, msg)) def log_close(self): self.log.close() class Xlloginfo(object): def __init__(self, path = ''): fname = '{}{}'.format(path,time.strftime('%Y-%m-%d', time.gmtime())) self.row = 0 self.xl = xlsxwriter.Workbook(path+fname+'.xls') self.style = self.xl.add_format({'bg_color':'red'}) def xl_write(self, *args): col = 0 style = '' if 'Error' in args: style = self.style for val in args: self.sheet.write_string(self.row, col, val) col += 1 self.row += 1 def log_init(self, sheetname, *title): self.sheet = self.xl.add_worksheet(sheetname) self.sheet.set_column('A:E', 30) self.xl_write(*title) def log_write(self, *args): self.xl_write(*args) def log_close(self): self.xl.close() if __name__ == '__main__': # log = Loginfo() # log.log_write('hahaha') # log.log_close() xinfo = Xlloginfo() xinfo.log_init('test12','uname','pwd', 'result', 'info') xinfo.log_close()
UserData
#!/usr/bin/env python3 # -*- coding:utf-8 -*- import codecs import xlrd, xlsxwriter def get_webinfo(path): web_info = {} with codecs.open(path,'r','utf-8') as config: for line in config: result = [ele.strip() for ele in line.split('=')] # print("Result,",result) web_info.update(dict([result])) return web_info def get_userinfo(path): user_info = [] with codecs.open(path,'r','utf-8') as config: for line in config: user_dict = {} result = [ele.strip() for ele in line.split(';')] for info in result: account = [ele.strip() for ele in info.split('=')] # print(account) # print("Dict",dict([account])) user_dict.update(dict([account])) user_info.append(user_dict) return user_info class XlUserInfo(object): def __init__(self, path= ''): self.xl = xlrd.open_workbook(path) def floattostr(self, val): if isinstance(val, float): val = str(int(val)) return val def get_sheet_info(self): listkey = ['uname','pwd'] infolist = [] for row in range(1, self.sheet.nrows): info = [self.floattostr(val) for val in self.sheet.row_values(row)] temp = zip(listkey, info) infolist.append(dict(temp)) return infolist def get_sheetinfo_by_name(self, name): self.sheet = self.xl.sheet_by_name(name) return self.get_sheet_info() def get_sheetinfo_by_index(self, index): self.sheet = self.xl.sheet_by_index(index) return self.get_sheet_info() if __name__ == '__main__': # userinfo = get_userinfo(r'userinfo.txt') # print(userinfo) # for user in userinfo: # print(user) # webinfo = get_webinfo(r'webinfo.txt') # print(webinfo) xinfo = XlUserInfo(r'userinfo.xlsx') info = xinfo.get_sheetinfo_by_index(0) print(info) info = xinfo.get_sheetinfo_by_name('Sheet1') print(info)