selenium+python关于登录的脚本代码，使用了读取excel以及向excel中写入测试结果的方法

# encoding:utf-8

from selenium import webdriver
import unittest
import xlwt
import xlrd
import time
from xlutils.copy import copy
from selenium.webdriver.common.by import By
class Denglu(unittest.TestCase):

def setUp(self):
self.driver=webdriver.Firefox()
self.baseurl="http://localhost:8000/phpwind85"
def testGetExcel(self,excelfile):
"""
该函数主要是获取需要读取数据的excel，如果找到就返回该excel，如果找不到给出提示信息
"""
try:
data=xlrd.open_workbook(excelfile)#获取通过excel路径工作表
table=data.sheets()[0]#制定需要操作的excel的sheet
return table#返回需要操作的excel的sheet
except:
return False
def testGetRows(self):
"""
该方法主要是得到要读取数据的excel的行
"""
table=self.testGetExcel("E:\\wwg\\dengludata.xls")
if table:
rows=table.nrows
print "该文档的有%i行" %rows
return rows
else:
print("该excel文档不存在")
def testGetCols(self):
"""
该方法主要是得到要读取数据的excel的列
"""
table=self.testGetExcel("E:\\wwg\\dengludata.xls")
if table:
cols=table.ncols
return cols
else:
print("该excel文档不存在")
def testGetLoginDict(self):
"""
该方法主要是得到从excel读取的数据并整合成列表
"""
table=self.testGetExcel("E:\\wwg\\dengludata.xls")#向testGetExcel方法传递一个excel的路径的参数
logindict={}#定义一个字典
rows=self.testGetRows()#获取该excel一共有多少行
cols=self.testGetCols()#获取该excel一共有多少列
for i in range(1,rows):
for m in range(cols-1):
logindict[table.cell(i,m).value]=table.cell(i,i).value#使用字典的赋值方法，把用户名和密码作为key和value对应赋值
return list(logindict.items())#把得到的有用户名和密码的字典转换为列表并返回
def testLoginin(self):
"""
该方法主要是进行登录操作
"""
resultlist=[]#定义一个测试结果的列表
ppp=u"登录成功"
faile=u"登录失败"
br=self.driver
br.get(self.baseurl)
loginlist=self.testGetLoginDict()#通过调用方法，得到用户名和密码的列表
br.find_element_by_link_text(u"登录").click()
for k,v in loginlist:#遍历上面步骤得到的用户名和密码的列表
br.find_element_by_id("J_u_login_username").send_keys(k)#传递用户名
br.find_element_by_id("J_u_login_password").send_keys(v)#传递密码
br.find_element_by_xpath("//*[@id='J_u_login_form']/div/dl[4]/dd/button").click()
time.sleep(10)
if br.current_url=="http://localhost:8000/phpwind85/":
resultlist.append(ppp)#向测试结果列表中增加值
resultlist.append(br.current_url)
self.testWriteData(resultlist)#调用testWriteData方法往excel中输入测试结果
else:
print(faile)
resultlist.append(faile)
self.testWriteData(resultlist)
def testWriteData(self,result):
rb=xlrd.open_workbook("E:\\wwg\\qqq.xls")
wb=copy(rb)
wbk=wb.get_sheet(0)
for i in range(len(result)):
wbk.write(0,i,result[i])
wb.save("E:\\wwg\\qqq.xls")

def tearDown(self):
pass

if __name__=="__main__":
suite=unittest.TestSuite()
#    suite.addTest(Denglu("testGetExcel"))
#    suite.addTest(Denglu("testGetRows"))
#    suite.addTest(Denglu("testGetCols"))
#    suite.addTest(Denglu("testGetLoginList"))
suite.addTest(Denglu("testLoginin"))

runner=unittest.TextTestRunner()
runner.run(suite)