# -*- coding: utf-8 -*-

 from selenium import webdriver

 from openpyxl import load_workbook

 from time import sleep

 class mylogintest():

     '''打开百度测试'''

     def openBD(self):

         '''打开百度'''

         print("begin")

         book = load_workbook("mylogintest.xlsx")  # 默认可读写，若有需要可以指定write_only和read_only为True

         b = book.active

         datadict = {} #创建一个字典变量

         maxC = b.max_column#获取最大列

         maxR = b.max_row#获取最大行

         for i in range(1, maxC + 1):#range默认从0开始，到后面参数的-1结束，而openpyxl都是从第一行第一列开始的，所以参数为1，maxC+1；意思就是遍历第一列到最后一列，

             datadict.setdefault(b.cell(1, i).value)#将第一行中的所有列即2标题作为键放到字典datadict里面

             # print(datadict)

         for i in range(2, maxR + 1):#除去第一行标题，从第二行遍历到最后一行

             for j in range(1, maxC + 1):#从第一列遍历到最后一列

                 datadict[b.cell(1, j).value] = b.cell(i, j).value#把第二行的内容作为键对应的值放入字典datadict里面

             try:

                 self.driver = webdriver.Firefox()

                 self.driver.get(datadict["url"])#从对应键里取出对应的URL值，这里的键就是表格的第一行标题，所以EXCEL的标题要按照规定来写

                 self.driver.find_element_by_xpath("/html/body/div[1]/div/div[1]/div[3]/a[1]/img").click()

                 if datadict["username"]!=None:

                     username = self.driver.find_element_by_name("username").send_keys(datadict["username"])#当用户名不为空时，把值放到username文本框中

                 if datadict["password"] != None:

                     password = self.driver.find_element_by_name("password").send_keys(datadict["password"])#当密码不为空时，把值放到password文本框中,要判断一下否则会出错

                 self.driver.find_element_by_id("loginbtn").click()#点击登录

                 if datadict["checkmeg"]!=None:#当校验值不为空时

                     msg = self.driver.find_element_by_xpath("/html/body/div[8]/div").text#比对校验值

                     if msg == datadict['checkmeg']:#如果校验值比对成功，将校验值写入结果

                         b.cell(i, maxC, msg)

                     else:

                         b.cell(i, maxC, '失败')#否则将失败写入结果中

                 else:#当校验值不为空时

                 # print(self.driver.switch_to.alert.text)

                     cur_url = self.driver.current_url#判断当前url

                     if cur_url == datadict['checkurl']:#如果当前url是校验的url

                         b.cell(i, maxC, '登录成功')  # 就将测试登录成功结果写入表格中

                     else:

                         b.cell(i, maxC, '登录失败')  # 就将测试登陆失败结果写入表格中

                 sleep(1)

                 self.driver.quit()

             except:

                 b.cell(i, maxC, 'error')#当程序出现错误时，将错误结果写入表格中

         book.save("mylogintest.xlsx")#最后记得关闭

         print("end")

 mylogintest().openBD()