python - 出于暂存目的循环遍历 CSV 文件

Question

大家早上好

这次我不是提出一个特定的代码问题，而是提出一个“我是否走在正确的轨道上”的问题。

我想编写一个用于暂存网络设备的脚本。我将得到一个包含两列的 cvs 列表:主机名 + 硬件类型，例如:

Hostname,Type
TestAP-1,AP250
TestAP-2,AP250
TestAP-3,AP250
TestSW-1,SW100
TestSW-2,SW100
TestAP-4,AP250

真正的列表显然会更长。

到目前为止，我的想法是解析调用硬件类型脚本的参数以获取主机名列表。然后连接到硬件(IP 地址将始终是相同的默认 IP 或通过串行控制台)，向硬件发送一些命令(设置主机名、系统日志服务器 ip 等)，再次从机器读取值(使用 show 命令) )并将结果写入另一个 csv 文件。

类似于:

staging.py AP250
get a list with all hostnames where the HW Type is AP250
loop
connect (ssh or serial) to the hardware
commands:
  set hostname 'hostname'
  set syslog ip (fixed ip address, the same for all devices)
  set whatever need to be set as well
the do a show on these settings and parse the value
write the value to a output csv file (create or append)

我走在正确的道路上吗？现在，我的脚本如下所示:

import csv
import argparse

parser = argparse.ArgumentParser()
parser.add_argument('Type', type=str,
                    help='REQUIRED: Please enter the Hardware Type (AP250, SW100, etc)')

args = parser.parse_args()

Manager = "manager.domain.com"
MACAddress = "aa:bb:cc:dd:ee:ff"

with open("input.csv", 'r', newline='') as f:

    reader = csv.DictReader(f)
    for row in reader:
        if row['Type'] == args.Type:

            print(row['Hostname']) # This will later on be replaced by some paramiko code
            h = (row['Hostname'])
            t = (row['Type'])

            outdict = {}
            print (outdict)

            """
            with open('output.csv', 'w', newline='') as o:
                writer = csv.DictWriter(o, delimiter=',', fieldnames=["Hostname", "Type", "Hivemanager", "MACAddress"])
                writer.writeheader()
                newrow = (h + "," + t + "," + Manager + "," + MACAddress)
                writer.writerow(newrow)
                #writer.writerows(['Hostname'], ['Type'], Manager)

            print (open('output.csv', 'r').read())
            """
        input("Press Enter to continue")

我不确定的地方:

• 使用词典是实现此目的的正确方法吗？
• 是否循环遍历主机名列表并以正确的方式执行该循环中的所有操作？

我不期望完全工作的脚本作为答案。相反，一些可以让我获得一些想法的提示或链接会很棒。

非常感谢。

Answer 1

至于你的第一个问题，使用字典是否正确，答案是双重的。首先，如果您的目标是 csv 阅读器，那么绝对是。这是正确的做法。您以 CSV 格式获取输入数据，因此没有明智的方法来解决 csv.DictReader 问题。其次，输出部分；这是您要存储返回值的位置。为此，您不一定需要字典，在我看来，列表会更容易。您计划使用 outdict 来实现此目的，对吗？但这又是个人喜好的问题，字典也可以，只是对数据的访问有点不同。底线是:是的，对于第一个问题，您的思路是正确的。

除了循环主机名并在每个主机名的基础上执行所有操作之外，还有什么(明智的)替代方案？这最大限度地减少了您必须对各个主机执行的登录操作，如果您在脚本中间失去网络连接，您只需修复一台主机，而不是所有主机。

希望这对您有帮助。