Создание нескольких соединений SSH за один раз с помощью Paramiko

Нижеприведенный код запускает grep на одной машине через SSH и печатает результаты:

Как я могу grep пять машин все сразу (так что у меня нет большой задержки), чем положить все это в пять переменных и распечатать их все.

Ответы

Ответ 1

Вам нужно будет поместить вызовы в отдельные потоки (или процессы, но это будет излишним), что, в свою очередь, требует, чтобы код был в функции (что в любом случае является хорошей идеей: не имеет существенного кода в верхний уровень модуля).

Например:

import sys, os, string, threading
import paramiko

cmd = "grep -h 'king' /opt/data/horror_20100810*"

outlock = threading.Lock()

def workon(host):

    ssh = paramiko.SSHClient()
    ssh.set_missing_host_key_policy(paramiko.AutoAddPolicy())
    ssh.connect(host, username='xy', password='xy')
    stdin, stdout, stderr = ssh.exec_command(cmd)
    stdin.write('xy\n')
    stdin.flush()

    with outlock:
        print stdout.readlines()

def main():
    hosts = ['10.10.3.10', '10.10.4.12', '10.10.2.15', ] # etc
    threads = []
    for h in hosts:
        t = threading.Thread(target=workon, args=(h,))
        t.start()
        threads.append(t)
    for t in threads:
        t.join()

main()

Если у вас было более пяти хостов, я бы рекомендовал вместо этого использовать "пул потоков" и очередь рабочих единиц. Но всего за пять проще придерживаться модели "выделенного потока" (тем более, что в стандартной библиотеке нет пула потоков, поэтому вам понадобится сторонний пакет, например threadpool... или много тонкого пользовательского кода, конечно, -).

Ответ 2

Просто запустите все в цикле for и не забудьте закрыть stdin перед тем, как перейти к следующей итерации. То есть, после строки stdin.flush() добавить строку stdin.close()