實驗環境：

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

在本次實驗中，主要使用了如下三個模塊：

openpyxl，對excel文件進行操作。

gevent，使用協程，同時對五臺設備進行操作。

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

python代碼：

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

monkey.patch_all()

#設置模板為華為的vrp操作系統
driver=get_network_driver('huawei_vrp')
#定義5臺設備的IP地址，如果IP過多的話，可以通過數據庫或者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+'連接成功')

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

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

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

#根據pprint(interfaces_ip_info)的打印信息，進行處理，提取接口名稱、IP地址、MASK保存到當前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'))

#得到插入數據的行數，n+2的原因是因為range是左開右閉，并且excel的第一行也已經寫入了數據類型（A1、B1、C1）
row_numbers=[n+2forninrange(len(interfaces_list))]

#對excel當前sheet中指定的cell進行賦值
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)

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

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

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


if__name__=='__main__':
#調用Openpyxl創建Workbook對象，用來創建一個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'))

#同時執行5個任務
tasks=[gevent.spawn(collect_basic_info,ip,yellowFill,thin_border)foripindevices_ip]
all_result=gevent.joinall(tasks)

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

pycharm運行結果：

查看Excel上的最終結果：

審核編輯：劉清