#!/usr/bin/env python3
# -*- coding:utf-8 -*-
'''
A model that use xlrd to read excel(include user info)
'''
import xlrdclass XlUserInfo(object):#initializedef __init__(self, path=''):self.path = pathself.xl = xlrd.open_workbook(self.path)# def get_ip_info(self):#     ip_sheets = self.xl.sheets()[0]#     server_name = ip_sheets.col_values(0)[1:]#     ips = ip_sheets.col_values(1)[1:]#     temp = zip(server_name, ips)#     server_ip = []#     server_ip.append(dict(temp))#     return server_ip# def get_fw_info(self):#     fw_sheets = self.xl.sheets()[1]#     fw_name = fw_sheets.col_values(0)#     fw_dir = fw_sheets.col_values(1)[1:]#     temp = zip(fw_name, fw_dir)#     server_image = []#     server_image.append(dict(temp))#     return server_image#获取sheet object，处理并读取信息def get_sheet_info(self):all_info = []info0 = []info1 = []for row in range(0,self.sheet.nrows):info = self.sheet.row_values(row)info0.append(info[0])info1.append(info[1])temp = zip(info0,info1)all_info.append(dict(temp))return all_info.pop(0)# return all_info#通过excel的sheet的名字读取信息def get_sheetinfo_by_name(self, name):self.sheet = self.xl.sheet_by_name(name)return self.get_sheet_info()#通过excel的sheet的索引读取信息def get_sheetinfo_by_index(self, index):self.sheet = self.xl.sheet_by_index(index)return self.get_sheet_info()if __name__ == '__main__':
#     实例化xl = XlUserInfo('user_info.xlsx')ips = xl.get_sheetinfo_by_name('ip')fwargs = xl.get_sheetinfo_by_name('fw')webargs = xl.get_sheetinfo_by_name('webEle')print("IPS",ips)for server_name in ips.keys():machine = server_name.split('_')[0]test_ip = ips[server_name]bmc = fwargs['BMC']official_bmc = fwargs['OfficialBMC']dsa = fwargs['DSA']print(test_ip)print(server_name)

import time
from outlog_model import OutLog
from selenium import webdriverdef login(testip,webarg):logger = OutLog(filepath='report/{}.log'.format(testip))profile = webdriver.FirefoxProfile()profile.accept_untrusted_certs = Truedriver = webdriver.Firefox(firefox_profile=profile)driver.implicitly_wait(30)driver.verificationErrors = []driver.accept_next_alert = Trueresult = 0try:driver.get('https://{}/designs/imm/index.php'.format(testip))except:logger.Write_info('[Fail to Login Web GUI] : {}'.format(testip))try:if driver.find_element_by_id(webarg["uname_id"]):result = 1else:result = 0# 'uname_id' = 'user'driver.find_element_by_id(webarg["uname_id"]).send_keys(webarg['uname'])finally:print("Wait web to be ready")time.sleep(15)# 'uname_id' = 'user'driver.find_element_by_id(webarg['uname_id']).send_keys(webarg['uname'])#pwd_id = passworddriver.find_element_by_id(webarg['pwd_id']).send_keys(webarg['pwd'])#login_button_id = btnLogin_labeldriver.find_element_by_id(webarg['login_id']).click()time.sleep(30)print("Success to login")# print("Login driver is ",driver)return driver

#!/usr/bin/env python3
# -*- coding:utf-8 -*-
'''
A model for produce outlog
'''
import time
import xlsxwriter
import logging
import os#A class to write excel
class XLLoginfo(object):def __init__(self, path=''):file_name = "Result-{}".format(time.strftime('%m-%d-%H-%M-%S',time.localtime()))self.row = 0self.xl = xlsxwriter.Workbook(path + file_name + '.xlsx')def xl_write(self, *args):col = 0for val in args:self.sheet.write_string(self.row, col, val)col += 1self.row += 1# The API for users.def Xl_init(self, sheetname, *title):self.sheet = self.xl.add_worksheet(sheetname)self.sheet.set_column('A:G',20)self.xl_write(*title)def Xl_write(self,*args):self.xl_write(*args)def Xl_close(self):self.xl.close()#A class using logging model to get log(info,warining,critical)
class OutLog(object):def __init__(self,filepath=''):self.file = filepathself.logger = logging.getLogger()self.logger.setLevel(logging.INFO)self.fh = logging.FileHandler(filepath)self.ch = logging.StreamHandler()self.formatter = logging.Formatter('%(message)s = %(name)s = %(asctime)s')self.fh.setFormatter(self.formatter)self.ch.setFormatter(self.formatter)self.logger.addHandler(self.fh)self.logger.addHandler(self.ch)def Write_info(self,info_message):self.logger.info(info_message)def Write_error(self,error_message):self.logger.error(error_message)def ReWrite_result(self):with open(self.file) as fp:old_info = []for i in fp:if i not in old_info:old_info.append(i)new_result = []for r in old_info:if r not in new_result:new_result.append(r)with open(self.file, 'w') as f:f.truncate()for h in new_result:f.writelines(h)class Totlalog(object):def __init__(self,path):self.path = pathdef eachfile(filepath):pathdir = os.listdir(filepath)print(pathdir)for file in pathdir:filename = os.path.join('%s\%s'%(filepath,file))with open(filename) as fp:print(filename,fp.readlines())if __name__ == '__main__':# A example to use XLLoginfo API# A example to use Outloglog = OutLog(filepath='report/log.txt')log.Write_error('error')log.Write_info('info')log.Write_info('info')log.Write_info('info')log.ReWrite_result()

# -*- coding: utf-8 -*-
import time
from login_model import login
from outlog_model import OutLogdef test_flash(server_name,testip,fwname,webarg):logger = OutLog(filepath='report/{}.log'.format('result'))try:driver = login(testip, webarg)driver.accept_next_alert = Trueprint('fwname:',fwname)print("Now begin to test flash {}".format(fwname))time.sleep(25)try:driver.find_element_by_id("dijit_MenuBarItem_0_text")except:driver.quit()finally:time.sleep(5)print("Web is ready now")driver.find_element_by_id("dijit_PopupMenuBarItem_2_text").click()driver.find_element_by_css_selector("#dijit_MenuItem_9_text > table > tbody > tr > td").click()driver.find_element_by_id("btnUpdateFwDlg_label").click()time.sleep(5)# .find_element_by_css_selector("input.dijitOffScreen").click()# driver.find_element_by_name("uploadedfile").clear()driver.find_element_by_name("uploadedfile").send_keys(fwname)if 'imm' in fwname:try:try:driver.find_element_by_id("updateServerFirmwareWizardbtnNext_label").click()finally:time.sleep(10)print("Now upload {} image".format(fwname))driver.find_element_by_id("updateServerFirmwareWizardbtnNext_label").click()time.sleep(200)try:driver.find_element_by_id("updateServerFirmwareWizardbtnNext_label").click()finally:time.sleep(15)print("Wait to flash {} image finished".format(fwname))driver.find_element_by_id("updateServerFirmwareWizardbtnNext_label").click()time.sleep(5)driver.find_element_by_id("updateServerFirmwareWizardbtnNext_label").click()time.sleep(180)# driver.find_element_by_xpath("(//input[@value=''])[5]").click()driver.find_element_by_id("restartIMMId_label").click()time.sleep(5)if driver.find_element_by_id("commonPopupOk_label"):result = 1logger.Write_info('[Pass] : IMM  {} {} {}'.format(server_name,testip,fwname))else:result = 0logger.Write_info('[Fail] : IMM  {} {} {}'.format(server_name, testip, fwname))print("Reatart imm wait about 5 min")driver.find_element_by_id("commonPopupOk_label").click()time.sleep(720)print("Finish test to flash IMM fw.")try:print("Close")driver.close()except:print("Quit")driver.quit()finally:return resultexcept:driver.close()result = -1logger.Write_error('[Fail] code:{} IMM  {} {} {}'.format(result,server_name, testip, fwname))return resulttime.sleep(20)elif 'uefi' in fwname:try:try:driver.find_element_by_id("updateServerFirmwareWizardbtnNext_label").click()finally:time.sleep(10)print("Now upload UEFI image")driver.find_element_by_id("updateServerFirmwareWizardbtnNext_label").click()time.sleep(20)try:driver.find_element_by_id("updateServerFirmwareWizardbtnNext_label").click()finally:time.sleep(15)driver.find_element_by_id("updateServerFirmwareWizardbtnNext_label").click()print("Wait to flash UEFI image finished,need 1 min 37 sec")time.sleep(5)driver.find_element_by_id("updateServerFirmwareWizardbtnNext_label").click()time.sleep(110)# driver.find_element_by_xpath("(//input[@value=''])[4]").click()if driver.find_element_by_id("restartOSId_label"):print("Restart OS")time.sleep(10)driver.find_element_by_id("restartOSId_label").click()else:time.sleep(10)driver.find_element_by_id("restartOSId_label").click()time.sleep(5)driver.find_element_by_id("commonPopupOk_label").click()time.sleep(5)driver.find_element_by_id("commonPopupClose_label").click()time.sleep(10)driver.find_element_by_id("dijit_MenuBarItem_0_text").click()time.sleep(5)driver.find_element_by_id("btnserverActionsListHealthSumm_label").click()time.sleep(5)driver.find_element_by_id("serverActionsListHealthSumm63_text").click()time.sleep(5)driver.find_element_by_id("commonPopupOk_label").click()time.sleep(5)print("Reatart os,wait about 5 min")time.sleep(300)print("Finish test to flash UEFI fw")if driver.find_element_by_id("commonPopupClose_label"):result = 1logger.Write_info('[Pass] : UEFI  {} {} {}'.format(server_name,testip,fwname))else:result = 0logger.Write_info('[Fail] : UEFI  {} {} {}'.format(server_name, testip, fwname))driver.find_element_by_id("commonPopupClose_label").click()try:print("Quit")driver.quit()except:print("Close")driver.close()finally:return resultexcept:driver.close()result = -1logger.Write_error('[Fail] code:{} UEFI  {} {} {}'.format(result,server_name, testip, fwname))return -1else:try:if driver.find_element_by_id("updateServerFirmwareWizardbtnNext_label"):driver.find_element_by_id("updateServerFirmwareWizardbtnNext_label").click()print("1.Now upload DSA image,need about 250s")else:time.sleep(10)print("2.Now upload DSA image,need about 250s")driver.find_element_by_id("updateServerFirmwareWizardbtnNext_label").click()time.sleep(420)if driver.find_element_by_id("updateServerFirmwareWizardbtnNext_label"):print("1.wait to flash dsa")driver.find_element_by_id("updateServerFirmwareWizardbtnNext_label").click()else:time.sleep(30)print("2.wait to flash dsa")driver.find_element_by_id("updateServerFirmwareWizardbtnNext_label").click()driver.find_element_by_id("updateServerFirmwareWizardbtnNext_label").click()print("Wait to flash DSA image finished,need 120 sec")time.sleep(5)driver.find_element_by_id("updateServerFirmwareWizardbtnNext_label").click()time.sleep(5)driver.find_element_by_id("updateServerFirmwareWizardbtnNext_label").click()time.sleep(5)driver.find_element_by_id("updateServerFirmwareWizardbtnNext_label").click()time.sleep(200)# updateServerFirmwareWizardbtnFinish_labelif driver.find_element_by_id("updateServerFirmwareWizardbtnFinish_label"):result = 1logger.Write_info('[Pass] : DSA  {} {} {}'.format(server_name,testip,fwname))driver.find_element_by_id("updateServerFirmwareWizardbtnFinish_label").click()else:result = 0logger.Write_info('[Fail] : DSA  {} {} {}'.format(server_name, testip, fwname))print("Finish test to flash DSA fw")try:print("Quit")driver.quit()except:print("Close")driver.close()finally:return resultexcept:driver.close()result = -1logger.Write_error('[Fail] code:-1  DSA  {} {} {}'.format(server_name, testip, fwname))return resultfinally:# return result
        driver.close()
if __name__ == "__main__":pass

#!/usr/bin/env python3
# -*- coding:utf-8 -*-
import threading
import time
from userdata import XlUserInfo
from flashImage import test_flash
from login_model import login
from outlog_model import OutLogdef main(ip,server_name,imm,official_imm,bios,DSA):# test_ip, serverName, bmc, official_bmc, uefi, dsalogger = OutLog(filepath='report/{}.log'.format('result'))# choose to flash immimm_result = test_flash(server_name, ip, imm, webargs)time.sleep(60)if imm_result != 1:imm_result = test_flash(server_name, ip, imm, webargs)time.sleep(60)# choose to flash uefiuefi_result = test_flash(server_name, ip, bios, webargs)if uefi_result != 1:uefi_result = test_flash(server_name, ip, bios, webargs)# choose to flash dsadsa_result = test_flash(server_name, ip, DSA, webargs)if dsa_result != 1:dsa_result = test_flash(server_name, ip, DSA, webargs)# choose to flash official immofficial_imm_result = test_flash(server_name, ip, official_imm, webargs)if official_imm_result != 1:official_imm_result = test_flash(server_name, ip, official_imm, webargs)# choose to update imm.upgrade_imm_result = test_flash(server_name, ip, imm, webargs)if upgrade_imm_result != 1:upgrade_imm_result = test_flash(server_name, ip, imm, webargs)# try:#     driver = login(test_ip,webargs)#     print(driver)#     if driver:#         driver.close()#         time.sleep(60)#         #choose to flash imm#         imm_result = test_flash(server_name,ip, imm, webargs)#         time.sleep(60)#         if imm_result != 1:#             imm_result = test_flash(server_name, ip, imm, webargs)#         time.sleep(60)#         # choose to flash uefi#         uefi_result = test_flash(server_name,ip, bios, webargs)#         if uefi_result != 1:#             uefi_result = test_flash(server_name, ip, bios, webargs)#         #choose to flash dsa#         dsa_result = test_flash(server_name, ip, DSA, webargs)#         if dsa_result != 1:#             dsa_result = test_flash(server_name, ip, DSA, webargs)#         #choose to flash official imm#         official_imm_result = test_flash(server_name, ip, official_imm, webargs)#         if official_imm_result != 1:#             official_imm_result = test_flash(server_name, ip, official_imm, webargs)#         #choose to update imm.#         upgrade_imm_result = test_flash(server_name, ip, imm, webargs)#         if upgrade_imm_result != 1:#             upgrade_imm_result = test_flash(server_name, ip, imm, webargs)#     else:#         logger.Write_error("[Fail Login into Web] : {} {}".format(server_name, ip))# except:#     logger.Write_error("[Fail Login into Web]]]]]] : {} {}".format(server_name, ip))# finally:#     time.sleep(60)#     logger.ReWrite_result()if __name__ == '__main__':now_time = time.strftime("%Y-%m-%d_%H.%M.%S", time.localtime(time.time()))#实例化一个xl对象，分别读取ip，fw，webEle的infoxl = XlUserInfo('user_info.xlsx')ips = xl.get_sheetinfo_by_name('ip')fwargs = xl.get_sheetinfo_by_name('fw')webargs = xl.get_sheetinfo_by_name('webEle')test_thread = []for serverName in ips.keys():machine = serverName.split('_')[0]test_ip = ips[serverName]bmc = fwargs['BMC']official_bmc = fwargs['OfficialBMC']dsa = fwargs['DSA']uefi = ''for uefi_name in fwargs.keys():if machine.lower() in uefi_name.lower():uefi = fwargs[uefi_name]print("*"*40)print(test_ip,bmc,official_bmc,uefi,dsa)print("*" * 40)t = threading.Thread(target=main,args=(test_ip,serverName,bmc,official_bmc,uefi,dsa))print(t)t.start()test_thread.append(t)time.sleep(2)for t in test_thread:t.join()print("Finish all Test")