需求：将手工登录，手工刷新服务器的FW转化为Python+Selenium实现自动化操作。
1.创建用户表，实现数据与脚本分离。需要读取模块。
2.自动化刷新FW.
不说话，直接上代码：
1userdata.py
#!/usr/bin/env python3
# -*- coding:utf-8 -*-
‘‘‘
A model that use xlrd to read excel(include user info)
‘‘‘
import xlrd
class XlUserInfo(object):
#initialize
def __init__(self, path=‘‘):
self.path = path
self.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)
2.Login_model
import time
from outlog_model import OutLog
from selenium import webdriver
def login(testip,webarg):
logger = OutLog(filepath=‘report/{}.log‘.format(testip))
profile = webdriver.FirefoxProfile()
profile.accept_untrusted_certs = True
driver = webdriver.Firefox(firefox_profile=profile)
driver.implicitly_wait(30)
driver.verificationErrors = []
driver.accept_next_alert = True
result = 0
try:
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 = 1
else:
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 = password
driver.find_element_by_id(webarg[‘pwd_id‘]).send_keys(webarg[‘pwd‘])
#login_button_id = btnLogin_label
driver.find_element_by_id(webarg[‘login_id‘]).click()
time.sleep(30)
print("Success to login")
# print("Login driver is ",driver)
return driver
3.outlog model
#!/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 = 0
self.xl = xlsxwriter.Workbook(path + file_name + ‘.xlsx‘)
def xl_write(self, *args):
col = 0
for val in args:
self.sheet.write_string(self.row, col, val)
col += 1
self.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 = filepath
self.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 = path
def 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 Outlog
log = OutLog(filepath=‘report/log.txt‘)
log.Write_error(‘error‘)
log.Write_info(‘info‘)
log.Write_info(‘info‘)
log.Write_info(‘info‘)
log.ReWrite_result()
4.刷新FW，动作模块
# -*- coding: utf-8 -*-
import time
from login_model import login
from outlog_model import OutLog
def test_flash(server_name,testip,fwname,webarg):
logger = OutLog(filepath=‘report/{}.log‘.format(‘result‘))
try:
driver = login(testip, webarg)
driver.accept_next_alert = True
print(‘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 = 1
logger.Write_info(‘[Pass] : IMM {} {} {}‘.format(server_name,testip,fwname))
else:
result = 0
logger.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 result
except:
driver.close()
result = -1
logger.Write_error(‘[Fail] code:{} IMM {} {} {}‘.format(result,server_name, testip, fwname))
return result
time.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 = 1
logger.Write_info(‘[Pass] : UEFI {} {} {}‘.format(server_name,testip,fwname))
else:
result = 0
logger.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 result
except:
driver.close()
result = -1
logger.Write_error(‘[Fail] code:{} UEFI {} {} {}‘.format(result,server_name, testip, fwname))
return -1
else:
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_label
if driver.find_element_by_id("updateServerFirmwareWizardbtnFinish_label"):
result = 1
logger.Write_info(‘[Pass] : DSA {} {} {}‘.format(server_name,testip,fwname))
driver.find_element_by_id("updateServerFirmwareWizardbtnFinish_label").click()
else:
result = 0
logger.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 result
except:
driver.close()
result = -1
logger.Write_error(‘[Fail] code:-1 DSA {} {} {}‘.format(server_name, testip, fwname))
return result
finally:
# return result
driver.close()
if __name__ == "__main__":
pass
5.主模块，运行模块
#!/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 OutLog
def main(ip,server_name,imm,official_imm,bios,DSA):
# test_ip, serverName, bmc, official_bmc, uefi, dsa
logger = OutLog(filepath=‘report/{}.log‘.format(‘result‘))
# 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)
# 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的info
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‘)
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")
时间： 2024-08-07 21:20:27