python与selenium自动化基础-测试多账户

代码

from selenium import webdriver
from selenium.webdriver.common.action_chains import ActionChains
import time
from log_module import Loginfo
from userdata import get_webinfo,get_userinfo
def openBrower():
    webdriver_handle = webdriver.Firefox()
    return webdriver_handle
def openUrl(handle,url):
    handle.get(url)
def findElement(d,arg):
    ele_login = d.find_element_by_link_text(arg['login_text'])
    ele_login.click()
    time.sleep(5)
    userEle = d.find_element_by_name(arg['userid'])
    pwdEle = d.find_element_by_name(arg['pwdid'])
    loginEle = d.find_element_by_css_selector('input[type="button"]')
    return userEle,pwdEle,loginEle
def sendVals(eletuple,arg):
    listkey = ['uname','pwd']
    i = 0
    for key in listkey:
        eletuple[i].clear()
        eletuple[i].send_keys(arg[key])
        i+=1
    eletuple[2].click()
def checkResult(d,xpath1,arg,log):
    result = False
    try:
        err = d.find_element_by_xpath(xpath1)
        print("Account and Pwd Error!")
        msg = '%s %s:error:%s
'%(arg['uname'],arg['pwd'],err.text)
        log.log_write(msg)
    except:
        print("Account and Pwd Right!")
        msg = '%s %s:pass
' % (arg['uname'],arg['pwd'])
        log.log_write(msg)
        result = True
    return result
def logout(d,ele_dict):
    time.sleep(5)
    ele = d.find_element_by_id(ele_dict['usermenu'])
    ActionChains(d).move_to_element(ele).perform()
    time.sleep(5)
    d.find_element_by_link_text(ele_dict['logout']).click()
def login_test(ele_dict,user_list):
    d = openBrower()
    log = Loginfo()
    openUrl(d,ele_dict['url'])
    ele_tuple = findElement(d,ele_dict)
    i=0
    for arg in user_list:
        print(i)
        sendVals(ele_tuple, arg)
        time.sleep(5)
        result = checkResult(d,ele_dict['errorid'],arg,log)
        if result:
            logout(d,ele_dict)
            time.sleep(3)
            ele_tuple = findElement(d,ele_dict)
        i+=1
    log.close()
if __name__ == '__main__':
    ele_dict = get_webinfo(r'G:yayapycharmwebinfo')
    user_list = get_userinfo(r'G:yayapycharmuserinfo')
    login_test(ele_dict,user_list)

webinfo

url = https://www.imooc.com/
login_text = 登录
userid = email
pwdid = password
errorid=/html/body/div[9]/div[2]/div/form/div[1]/p
usermenu=header-avator
logout=安全退出

userinfo

uname=xxxxx pwd=yyyyy
uname=1 pwd=1
uname=2 pwd=2

userdata.py

def get_webinfo(path):
    web_info = {}
    config = open(path,encoding='UTF-8')
    for line in config:
        result = [ele.strip() for ele in line.split('=')]
        web_info.update(dict([result]))
    return web_info

def get_userinfo(path):
    user_info = []
    config = open(path,encoding='UTF-8')
    for line in config:
        user_dict = {}
        result = [ele.strip() for ele in line.split(' ')]
        for r in result:
            account = [ele.strip() for ele in r.split('=')]
            user_dict.update(dict([account]))
        user_info.append(user_dict)
    return user_info
if __name__ == '__main__':
    webinfo = get_webinfo(r'G:yayapycharmwebinfo')
    for key in webinfo:
        print(key,webinfo[key])
    userinfo = get_userinfo(r'G:yayapycharmuserinfo')
    for l in userinfo:
        print(l)

log_moudle.py

import time
class Loginfo(object):
    def __init__(self,path='',mode='w'):
        fname = path+time.strftime('%Y-%m-%d',time.gmtime())
        self.log = open(path+fname+'.txt',mode,encoding='UTF-8')
    def log_write(self,msg):
        self.log.write(msg)
    def close(self):
        self.log.close()
if __name__ == '__main__':
    log = Loginfo()
    log.log_write('Test Loginfo 测试')
    log.close()

完