怎么使用python提取華為交換機(jī)的接口IP信息保存到excel中呢？

實(shí)驗(yàn)環(huán)境：

如圖所示，SW1-SW5上分別已經(jīng)配置了SSH，使用外部PC可以SSH到其中。

在本次實(shí)驗(yàn)中，主要使用了如下三個(gè)模塊：

openpyxl，對excel文件進(jìn)行操作。

gevent，使用協(xié)程，同時(shí)對五臺設(shè)備進(jìn)行操作。

napalm-huawei-vrp，SSH到交換機(jī)，并提取格式化信息。

python代碼：

fromnapalmimportget_network_driver
fromopenpyxlimportWorkbook
fromopenpyxl.stylesimportColor,PatternFill,Font,Border,Side
importgevent
fromgeventimportmonkey
frompprintimportpprint

monkey.patch_all()

#設(shè)置模板為華為的vrp操作系統(tǒng)
driver=get_network_driver('huawei_vrp')
#定義5臺設(shè)備的IP地址，如果IP過多的話，可以通過數(shù)據(jù)庫或者txt文件保存并讀取
devices_ip=['192.168.0.11','192.168.0.22','192.168.0.33','192.168.0.44','192.168.0.55']


defcollect_basic_info(ip,fill,border):
print("start:",ip)
try:
SW=driver(ip,'prin','Huawei@123')
SW.open()
print(ip+'連接成功')

#獲取設(shè)備基礎(chǔ)信息
basic_info=SW.get_facts()
#pprint(basic_info)
#提取其中的hostname信息
hostname=basic_info.get('hostname')

#獲取設(shè)備接口三層信息
interfaces_ip_info=SW.get_interfaces_ip()
#pprint(interfaces_ip_info)
#返回設(shè)備名稱和接口IP信息

#在excel文件中，創(chuàng)建以hostname命名的sheet
ws=wb.create_sheet(hostname)
#設(shè)置此sheet中記錄數(shù)據(jù)的類別
ws['A1']='Interfaces'
ws['B1']='IP'
ws['C1']='Mask'
#設(shè)置背景顏色
ws['A1'].fill=fill
ws['B1'].fill=fill
ws['C1'].fill=fill
#創(chuàng)建要寫入數(shù)據(jù)的列表
interfaces_list=[]
ip_addr_list=[]
mask_list=[]

#根據(jù)pprint(interfaces_ip_info)的打印信息，進(jìn)行處理，提取接口名稱、IP地址、MASK保存到當(dāng)前sheet中
forkey,valuesininterfaces_ip_info.items():
interfaces_list.append(key)

forip,maskinvalues['ipv4'].items():
ip_addr_list.append(ip)
mask_list.append(mask.get('prefix_length'))

#得到插入數(shù)據(jù)的行數(shù)，n+2的原因是因?yàn)閞ange是左開右閉，并且excel的第一行也已經(jīng)寫入了數(shù)據(jù)類型（A1、B1、C1）
row_numbers=[n+2forninrange(len(interfaces_list))]

#對excel當(dāng)前sheet中指定的cell進(jìn)行賦值
forinterface,rowinzip(interfaces_list,row_numbers):
ws.cell(row=row,column=1,value=interface)
forip_addr,rowinzip(ip_addr_list,row_numbers):
ws.cell(row=row,column=2,value=ip_addr)
formask,rowinzip(mask_list,row_numbers):
ws.cell(row=row,column=3,value=mask)

#調(diào)整列的寬度（自適應(yīng)數(shù)據(jù)長度）,保證excel文件的美觀性,首先創(chuàng)建一個(gè)名為dims的空字典
dims={}
#ws1.rows返回值的類型為生成器generator，其中包含每一排和每一列有交集的所有單元格
#（每一排中至少有一個(gè)單元格為非空），比如(A1,B1),(A2,B2),(A3,B3)
forrowinws.rows:
#遍歷每一排元組里的每一個(gè)元素（即單元格A1,B1,A2,B2,A3,B3）
forcellinrow:
#為每一個(gè)單元添加邊框
cell.border=border
#如果單元格內(nèi)容為非空，則用max()比較每一列下最長的字符，比如從A1和A2,A3相比較，B1和B2,B3相比較
ifcell.value:
#cell.column_letter返回的值是單元格所在的列的名稱，其數(shù)據(jù)類型為字符串，比如A1,A2,A3返回'A'，B1,B2,B3則返回'B'。
#第一次故意用dims.get(cell.column,0)返回一個(gè)0，因?yàn)榇藭r(shí)dims字典下還沒有cell.column這個(gè)鍵名，
#字典的的get()函數(shù)在鍵名缺失的情況下會返回第二個(gè)我們給定的參數(shù)，即這里的0。
dims[cell.column_letter]=max((dims.get(cell.column_letter,0),len(str(cell.value))))

#遍歷字典里的鍵值對，以每一排寬度最長的單元格作為自動調(diào)整單元格長度的標(biāo)準(zhǔn)，
#長度+1以確保列的寬度超過最長單元格的寬度
forcol,valueindims.items():
ws.column_dimensions[col].width=value+1

exceptExceptionase:
print(e)
print("end:",ip)


if__name__=='__main__':
#調(diào)用Openpyxl創(chuàng)建Workbook對象，用來創(chuàng)建一個(gè)workbook，并將它賦值給變量wb
wb=Workbook()
#定義表格中的背景顏色和樣式
yellowFill=PatternFill(start_color='FFFF00',end_color='FFFF00',fill_type='solid')
thin_border=Border(left=Side(style='thin'),right=Side(style='thin'),top=Side(style='thin'),
bottom=Side(style='thin'))

#同時(shí)執(zhí)行5個(gè)任務(wù)
tasks=[gevent.spawn(collect_basic_info,ip,yellowFill,thin_border)foripindevices_ip]
all_result=gevent.joinall(tasks)

#刪除掉默認(rèn)的Sheet
ws=wb["Sheet"]
wb.remove(ws)
#查看存在哪些表
print(wb.sheetnames)
#保存excel信息
wb.save('huawei_interface_ip_info.xlsx')

pycharm運(yùn)行結(jié)果：

查看Excel上的最終結(jié)果：

審核編輯：劉清