python - 通过 IP 地址列表执行 SSH 命令(使用 Paramiko)

Question

第一次发帖，因为我对 Python 非常陌生，但我遇到了一个需要帮助的问题 - 如果我的代码看起来有点初级，请原谅我。

使用 Paramiko，我构建了一个用于商业用途的自定义 SSH 客户端 - 最终游戏是让它通过 IP 地址列表(从文本文件解析)执行给定的命令(以重新启 Action 为示例)顺序。

对于 Paramiko 部分:

import paramiko
import sys
import os
import time

port = 22

def ssh_comm(ip, user, password, cmd):
     try:
        client = paramiko.SSHClient()
        client.set_missing_host_key_policy(paramiko.AutoAddPolicy())
        client.connect(ip, username='username', password='password')
        ssh_session = client.get_transport().open_session()
        if ssh_session.active:
           ssh_session.exec_command(cmd),
           buffer = (ssh_session.recv(4096)).decode("utf-8")
           print (buffer)
     except paramiko.AuthenticationException:
        ssh_comm2(ip, user, password, cmd) **#TRIES NEXT SET OF CREDENTIALS**

def cmd_loop_ssh():
        while True:
                cmd = input('Command: ')
                ssh_comm(ip, user, password, cmd)

ip = input('IP: ')
cmd_loop_ssh()

这对于在一台机器上执行命令来说效果很好，但我需要它在多台机器上执行 - 我可以从格式化文本文件中解析我需要的 IP，如下所示:

iplist = '\IP TEXT FILE LOCATION.txt\'
    with open(iplist, 'r') as fp:
        line = fp.readline()
        while line:
            print ('{}'.format(line.strip()))
            line = fp.readline()

再次，非常对此感到陌生 - 任何帮助将不胜感激!

Answer 1

首先，您可以更清晰地读取您的 IP 地址文件，如下所示:

with open(iplist) as fp:
    ips = [line.strip() for line in fp]

然后我认为您需要做的就是将该 block 放入 cmd_loop_ssh 中，在 while 循环之前，然后替换

ssh_comm(ip, user, password, cmd)

与

for ip in ips:
    ssh_comm(ip, user, password, cmd)

您可能还想添加退出命令:

cmd = input('Command: ')
if cmd == 'quit':
    break
else:
    for ip in ips:
        ssh_comm(ip, user, password, cmd)