轉(zhuǎn)載請(qǐng)注明以下內(nèi)容：

來(lái)源：公眾號(hào)【網(wǎng)絡(luò)技術(shù)干貨圈】

作者：圈圈

ID：wljsghq

隨著網(wǎng)絡(luò)規(guī)模的擴(kuò)大和設(shè)備數(shù)量的增加，手動(dòng)配置和管理每臺(tái)網(wǎng)絡(luò)設(shè)備變得越來(lái)越不現(xiàn)實(shí)。因此，自動(dòng)化工具和腳本變得尤為重要。Python語(yǔ)言以其簡(jiǎn)潔性和強(qiáng)大的第三方庫(kù)支持，成為了網(wǎng)絡(luò)自動(dòng)化領(lǐng)域的首選。本篇文章將詳細(xì)介紹如何使用Python批量連接華為網(wǎng)絡(luò)設(shè)備，實(shí)現(xiàn)自動(dòng)化配置和管理。

環(huán)境準(zhǔn)備

在開(kāi)始編寫(xiě)腳本之前，需要確保我們的工作環(huán)境具備以下條件：

安裝Python 3.x。

安裝paramiko庫(kù)，用于實(shí)現(xiàn)SSH連接。

安裝netmiko庫(kù)，這是一個(gè)基于paramiko的高級(jí)庫(kù)，專門用于網(wǎng)絡(luò)設(shè)備的自動(dòng)化操作。

安裝Python和相關(guān)庫(kù)

首先，確保你已經(jīng)安裝了Python 3.x。如果尚未安裝，可以從Python官方網(wǎng)站https://www.python.org/downloads下載并安裝。

然后，使用pip安裝paramiko和netmiko庫(kù)：

pipinstallparamiko
pipinstallnetmiko

基礎(chǔ)知識(shí)

在實(shí)際操作之前，我們需要了解一些基礎(chǔ)知識(shí)：

SSH協(xié)議：用于安全地遠(yuǎn)程登錄到網(wǎng)絡(luò)設(shè)備。

華為網(wǎng)絡(luò)設(shè)備的基本命令：了解一些基本的配置命令有助于編寫(xiě)自動(dòng)化腳本。

使用Netmiko連接單個(gè)設(shè)備

首先，我們來(lái)看看如何使用netmiko連接到單個(gè)華為網(wǎng)絡(luò)設(shè)備并執(zhí)行基本命令。

連接單個(gè)設(shè)備

fromnetmikoimportConnectHandler

#定義設(shè)備信息
device={
'device_type':'huawei',
'host':'192.168.1.1',
'username':'admin',
'password':'admin123',
'port':22,
}

#連接到設(shè)備
connection=ConnectHandler(**device)

#執(zhí)行命令
output=connection.send_command('displayversion')
print(output)

#斷開(kāi)連接
connection.disconnect()

在上面的代碼中，我們定義了一個(gè)包含設(shè)備信息的字典，并使用ConnectHandler類來(lái)建立連接。然后，我們使用send_command方法來(lái)發(fā)送命令并獲取輸出，最后斷開(kāi)連接。

批量連接多個(gè)設(shè)備

在實(shí)際應(yīng)用中，我們通常需要批量處理多個(gè)設(shè)備。接下來(lái)，我們將介紹如何使用Python腳本批量連接多個(gè)華為網(wǎng)絡(luò)設(shè)備。

定義設(shè)備列表

首先，我們需要定義一個(gè)設(shè)備列表，每個(gè)設(shè)備的信息以字典形式存儲(chǔ)：

devices=[
{
'device_type':'huawei',
'host':'192.168.1.1',
'username':'admin',
'password':'admin123',
'port':22,
},
{
'device_type':'huawei',
'host':'192.168.1.2',
'username':'admin',
'password':'admin123',
'port':22,
},
#可以繼續(xù)添加更多設(shè)備
]

批量連接和執(zhí)行命令

接下來(lái)，我們編寫(xiě)一個(gè)函數(shù)來(lái)批量連接這些設(shè)備并執(zhí)行命令：

defbatch_execute_commands(devices,command):
results={}
fordeviceindevices:
try:
connection=ConnectHandler(**device)
output=connection.send_command(command)
results[device['host']]=output
connection.disconnect()
exceptExceptionase:
results[device['host']]=f"Connectionfailed:{e}"
returnresults

#批量執(zhí)行命令
command='displayversion'
results=batch_execute_commands(devices,command)

#輸出結(jié)果
fordevice,outputinresults.items():
print(f"Device:{device}")
print(output)
print('-'*40)

在這個(gè)函數(shù)中，我們遍歷設(shè)備列表，逐個(gè)連接設(shè)備并執(zhí)行指定命令。結(jié)果存儲(chǔ)在一個(gè)字典中，最后輸出每個(gè)設(shè)備的結(jié)果。

高級(jí)應(yīng)用：并行連接設(shè)備

當(dāng)設(shè)備數(shù)量較多時(shí)，逐個(gè)連接和執(zhí)行命令的效率會(huì)很低。為了解決這個(gè)問(wèn)題，我們可以使用并行處理來(lái)同時(shí)連接多個(gè)設(shè)備。

使用多線程并行連接

我們可以使用Python的concurrent.futures模塊來(lái)實(shí)現(xiàn)多線程并行連接：

importconcurrent.futures
fromnetmikoimportConnectHandler

defconnect_and_execute(device,command):
try:
connection=ConnectHandler(**device)
output=connection.send_command(command)
connection.disconnect()
returndevice['host'],output
exceptExceptionase:
returndevice['host'],f"Connectionfailed:{e}"

defbatch_execute_commands_parallel(devices,command):
results={}
withconcurrent.futures.ThreadPoolExecutor(max_workers=5)asexecutor:
future_to_device={executor.submit(connect_and_execute,device,command):devicefordeviceindevices}
forfutureinconcurrent.futures.as_completed(future_to_device):
device=future_to_device[future]
try:
host,output=future.result()
results[host]=output
exceptExceptionase:
results[device['host']]=f"Executionfailed:{e}"
returnresults

#并行批量執(zhí)行命令
command='displayversion'
results=batch_execute_commands_parallel(devices,command)

#輸出結(jié)果
fordevice,outputinresults.items():
print(f"Device:{device}")
print(output)
print('-'*40)

在這個(gè)示例中，我們使用ThreadPoolExecutor來(lái)創(chuàng)建一個(gè)線程池，并行處理多個(gè)設(shè)備的連接和命令執(zhí)行。這樣可以顯著提高處理效率。

實(shí)戰(zhàn)案例：批量配置華為交換機(jī)

接下來(lái)，我們通過(guò)一個(gè)實(shí)際案例來(lái)演示如何批量配置多個(gè)華為交換機(jī)。假設(shè)我們需要配置一批交換機(jī)的基本網(wǎng)絡(luò)設(shè)置。

定義配置命令

首先，我們定義需要執(zhí)行的配置命令。假設(shè)我們要配置交換機(jī)的主機(jī)名和接口IP地址：

defgenerate_config_commands(hostname,interface,ip_address):
return[
f"system-view",
f"sysname{hostname}",
f"interface{interface}",
f"ipaddress{ip_address}",
f"quit",
f"save",
f"y",
]

批量執(zhí)行配置命令

然后，我們編寫(xiě)一個(gè)函數(shù)來(lái)批量執(zhí)行這些配置命令：

defconfigure_devices(devices,config_generator):
results={}
fordeviceindevices:
try:
connection=ConnectHandler(**device)
commands=config_generator(
hostname=f"Switch-{device['host']}",
interface="GigabitEthernet0/0/1",
ip_address=f"192.168.1.{device['host'].split('.')[-1]}/24"
)
output=connection.send_config_set(commands)
results[device['host']]=output
connection.disconnect()
exceptExceptionase:
results[device['host']]=f"Configurationfailed:{e}"
returnresults

#批量配置設(shè)備
results=configure_devices(devices,generate_config_commands)

#輸出結(jié)果
fordevice,outputinresults.items():
print(f"Device:{device}")
print(output)
print('-'*40)

在這個(gè)函數(shù)中，我們?yōu)槊颗_(tái)設(shè)備生成配置命令，并使用send_config_set方法批量執(zhí)行這些命令。配置完成后，輸出每臺(tái)設(shè)備的結(jié)果。

處理異常情況

在實(shí)際操作中，我們需要處理各種可能的異常情況。例如，設(shè)備連接失敗、命令執(zhí)行錯(cuò)誤等。我們可以在腳本中加入詳細(xì)的異常處理機(jī)制，確保腳本在出現(xiàn)問(wèn)題時(shí)能夠適當(dāng)處理并記錄錯(cuò)誤信息。

增強(qiáng)異常處理

defconfigure_devices_with_error_handling(devices,config_generator):
results={}
fordeviceindevices:
try:
connection=ConnectHandler(**device)
commands=config_generator(
hostname=f"Switch-{device['host']}",
interface="GigabitEthernet0/0/1",
ip_address=f"192.168.1.{device['host'].split('.')[-1]}/24"
)
output=connection.send_config_set(commands)
results[device['host']]=output
connection.disconnect()
exceptExceptionase:
results[device['host']]=f"Configurationfailed:{e}"
returnresults

#批量配置設(shè)備并處理異常
results=configure_devices_with_error_handling(devices,generate_config_commands)

#輸出結(jié)果
fordevice,outputinresults.items():
print(f"Device:{device}")
print(output)
print('-'*40)

在這個(gè)示例中，我們?cè)诿總€(gè)設(shè)備的配置過(guò)程中加入了異常處理。如果某個(gè)設(shè)備出現(xiàn)問(wèn)題，會(huì)捕獲異常并記錄錯(cuò)誤信息，而不會(huì)影響其他設(shè)備的配置。

日志記錄

為了更好地管理和排查問(wèn)題，我們可以在腳本中加入日志記錄功能。通過(guò)記錄詳細(xì)的日志信息，可以方便地了解腳本的運(yùn)行情況和設(shè)備的配置狀態(tài)。

使用logging模塊記錄日志

importlogging

#配置日志記錄
logging.basicConfig(filename='network_config.log',level=logging

.INFO,format='%(asctime)s-%(levelname)s-%(message)s')

defconfigure_devices_with_logging(devices,config_generator):
results={}
fordeviceindevices:
try:
connection=ConnectHandler(**device)
commands=config_generator(
hostname=f"Switch-{device['host']}",
interface="GigabitEthernet0/0/1",
ip_address=f"192.168.1.{device['host'].split('.')[-1]}/24"
)
output=connection.send_config_set(commands)
results[device['host']]=output
logging.info(f"Successfullyconfigureddevice{device['host']}")
connection.disconnect()
exceptExceptionase:
error_message=f"Configurationfailedfordevice{device['host']}:{e}"
results[device['host']]=error_message
logging.error(error_message)
returnresults

#批量配置設(shè)備并記錄日志
results=configure_devices_with_logging(devices,generate_config_commands)

#輸出結(jié)果
fordevice,outputinresults.items():
print(f"Device:{device}")
print(output)
print('-'*40)

在這個(gè)示例中，我們使用logging模塊記錄日志信息。成功配置設(shè)備時(shí)記錄INFO級(jí)別日志，配置失敗時(shí)記錄ERROR級(jí)別日志。