Python - Netmiko 從 2 列讀取

Question

我有以下代碼讀取帶有主機名列表的 CSV，並運行 2 個命令。

我需要更改它，以便它收到的 CSV 文件有 2 列，一列包含主機名，另一列包含要插入該路由器的相應命令。

主機名	命令
CPE_1111	sh ip 國際br
CPE_2222	sh運行
ETC	(...)

(...)

nodenum=1
f=open('routers.csv', 'r') #File with Hostnames
c=f.read()
file_as_list = c.splitlines()


with open('Output.txt','w') as f: #File with output
    
    logf = open("error.csv", "a") #Logfile
    loga = csv.writer(logf)
    loga.writerow(["Hostname"])

    for i in file_as_list :
        print ("Node", nodenum, "...Checking IP Address...", i)
        try:
            Connection = netmiko.ConnectHandler(ip=i, device_type="cisco_ios" , username=raw_input("Enter your Username:"), password=getpass.getpass(), verbose=False)
        except:
            try:
                print("Cannot connect via SSH. Trying Telnet")
                Connection = netmiko.ConnectHandler(ip=i, device_type="cisco_ios_telnet" , username=raw_input("Enter your Username:"), password=getpass.getpass(), verbose=False)
                
            except:
                    print("SSH and Telnet Failed")
                    print("")
                    now = str(datetime.now().strftime("%Y-%m-%d %H:%M:%S"))
                    loga.writerow([i])
                    nodenum = nodenum+1
                    continue
          
        hostname = (Connection.send_command("show run | include hostname"))
        cellular = (Connection.send_command("sh ip int brief"))
        Connection.disconnect

(...)

Answer 1

您的答案取決於如何讀取 csv。 您可以使用csv.DictReader()讀取每一行並將其轉換為字典。

import csv

with open(file="routers.csv", mode="rt") as f:
    next(f)
    lst = csv.DictReader(f=f, fieldnames=["ip", "cmd"])
    ips_cmds = list(lst)

for ip_cmd in ips_cmds:
    print("Hostname:", ip_cmd["ip"])
    print("Command:", ip_cmd["cmd"], end="\n\n")

# Hostname: CPE_1111
# Command: show ip interface brief

# Hostname: CPE_2222
# Command: show running-config

然后在連接到每個路由器的for loop中，您可以 select 從fieldnames中指定的鍵中獲取所需的值。

conn = ConnectHandler(
    device_type="cisco_ios",
    ip=ip_cmd["ip"],
    username=input("Username: "),
    password=getpass(prompt="Password: "),
    secret=getpass(prompt="Enable Secret: "),
    fast_cli=False,
)


hostname = conn.send_command(command_string=ip_cmd["cmd"])

不要忘記為要執行的disconnect() function 添加括號。

conn.disconnect()