sexta-feira, 24 de fevereiro de 2012

Executando comandos por ssh automaticamente, sem troca de chaves



Essa é bem legal.
Para enviar comando a vários servidores de um cluster ao mesmo tempo sem usar troca de chaves, existe um programa chamado cssh. O problema é que ele usa o terrminal vt100 e abre uma lista de servidores cada um em uma janela; abre as janelas, depois as organiza e a partir de então digita-se os comandos em um inputbox, que envia a saída para todas as instâncias ao mesmo tempo. O curioso e estranho  é que às vezes a saída falha em algum(ns) do(s) terminais, e então se faz necessário rodar novamente o comando nessa janela.
Para agilizar, resolvi fazer um programinha em python, para envio de comandos, mas poderia ser usado para transferência de arquivos via scp e assim evita-se a necessidade de manter chaves, o que é um problema quando se troca máquinas de um cluster e se esquece a chave na antiga máquina.
Estava com problemas com a thread, mas o Marcelo Barros mais uma vez deu-me uma luz.
Indo ao código:

#!/usr/bin/env python
#By: Djames Suhanko - Catho online

#lista de servidores

sincroniza =["server1.domain","server2.domain", "server3.domain"]

import pexpect
import sys
from threading import Thread

#usuario e senha podem ser colocados em um arquivo.ini ou estaticamente nesse arquivo

#Testa a lista de parâmetros

try:
    if sys.argv[3]:
        pass
except:
    print  "Uso: " + "script" + " <comando entre aspas> <usuario> <senha>"
    sys.exit()

#essa função inicia o envio dos comandos passados como argumentos.
def executor(comando,usuario,senha,servidor):
    a = 'ssh ' + usuario + '@' + servidor
    foo = pexpect.spawn(a)
    foo.expect('.*ssword:')
    foo.sendline(senha)
    foo.sendline('su')
    foo.expect('.*sword:')
    foo.sendline('senha_secreta')
    foo.sendline(comando + '&& exit')
    print "comando e saida para: " + servidor + "..........[OK]"
    foo.sendline('exit')
    foo.expect('.*osed.')
    foo.interact()

#cria uma lista...
tasks = []

#loop com thread da funcao executor
for i in sincroniza:
    t = Thread(target=executor,args=(sys.argv[1],sys.argv[2],sys.argv[3],i))
    t.start()
    tasks.append(t)

#aguarda a finalização de todas as tarefas.

for t in tasks:
    t.join()


Nenhum comentário:

Postar um comentário