SRC : Simultaneous Remote Command

J'ai donc développé un petit logiciel en Python pour envoyer une commande (ou série de commandes) à un ensemble de serveur. Le logiciel envoie cette commande grâce à ssh. Par exemple, cela permet de faire les mise-à-jour de sécurité de serveurs Debian en envoyant "apt-get update;apt-get upgrade".

Le script

Le script utilise des thread pour envoyer à chaque serveur la commande à exécuter. La sortie de cette commande est enregistrée dans un fichier qui se nomme <nom_du_server>.log

 #!/usr/bin/env python
 
 import os
 from threading import Thread
 
 
 class MyThread(Thread):
     def __init__(self,serverName,command):
                 Thread.__init__(self)
                 self.serverName=serverName
                 self.command=command
     
    def run(self):
                cmd_ssh = "ssh root@"+self.serverName+ " '"+self.command+"'"
                 logfile = open(self.serverName+".log",'w')
                 for line in os.popen(cmd_ssh).readlines():
                         logfile.write(line)
                 logfile.close()
 
 cmdFile = open("cmd")
 cmd = cmdFile.readline()
 cmdFile.close()
 
 servers = open("servers")
 server = servers.readline()
 while server:
         thrd = MyThread(server[:-1], cmd[:-1])
         thrd.start()
         print "Thread for "+server[:-1]+" server started"
         server = servers.readline()
 servers.close() 

Utilisation

Pour utiliser le script, téléchargez le ici et créer les fichiers suivants :

• servers : contenant sur chaque ligne le nom ou l'adresse du serveur
• cmd : la commande à envoyer sur une ligne (si vous voulez envoyer plusieurs commandes, il faut les acrire sur une seule ligne et les séparer par un ; ).

Pour que cela fonctionne, il faut utiliser ssh avec une authentification par certificat et que le script n'est pas à "saisir" la passphrase (soit il n'y en a pas, soit il faut utiliser ssh-agent).
Il ne reste plus qu'à lancer la commande src.py et attendre la fin des commandes sur chaque serveur