[wifi/bornes.py] bornes.py une library pour les bornses wifi

Ignore-this: 3676cd861d964371d573bc5840443200

darcs-hash:20090630170411-bd074-4cafb8eff3294de0316de56a2fc8637b1bba8aa9.gz

wifi/bornes.py

@@ -0,0 +1,128 @@
+#!/usr/bin/env python
+# -*- coding: utf-8 -*-
+#
+# uname_bornes.py
+# ---------------
+# Copyright : (c) 2009, Jeremie Dimino <jeremie@dimino.org>
+# Licence   : BSD3
+
+"""Script qui fait un uname sur toute les bornes à la fois.
+
+La sortie est la sortie de `uname -v` sur chaque borne."""
+
+import sys, os, commands
+from optparse import OptionParser
+
+def bornes():
+    """Renvoie la liste des bornes
+
+    Moyen rapide, mais peut-être un peu crade pour récupérer la liste des bornes:
+    on regarde les machines présentes dans 138.231.148.1/24"""
+    names = []
+    for line in commands.getoutput("host -l wifi.crans.org").splitlines():
+        # Chaque ligne est de la forme:
+        # "truc.wifi.crans.org has address 138.231.148.42"
+        fields = line.split()
+        name = fields[0].split(".")[0]
+        ip = fields[3]
+        network = ip.rsplit(".", 1)[0]
+        # On ne prend que les machines du réseau 138.231.148.0/24, à part
+        # ragnarok:
+        if network == "138.231.148" and ip != "138.231.148.1":
+            names.append(name)
+    return names
+
+def ssh_exec(host, cmd):
+    """Execute une commande en ssh sur une machine et renvoie le résultat"""
+    (stdin, stdout, stderr) = os.popen3("ssh -T -x -o BatchMode=yes -o ConnectTimeout=5 %(host)s exec %(cmd)s" % locals())
+    stdin.close()
+    stderr.close()
+    return stdout
+
+def munin(config, cmd, process=(lambda x: x)):
+    """plugin munin"""
+
+    if 'autoconf' in sys.argv:
+        print 'yes'
+        sys.exit(0)
+
+    names = bornes()
+
+    if 'config' in sys.argv:
+        print config
+        for borne in names:
+            print "%s.label %s"% (borne, borne)
+        sys.exit(0)
+
+    outputs = {}
+    for name in names:
+        outputs[name] = ssh_exec(name+".wifi", cmd)
+
+    for name, res in outputs.iteritems():
+        value = process(res.read()).split('\n', 1)[0].strip()
+        print '%s.value %s' % (name, value)
+
+def main():
+    parser = OptionParser(usage=usage)
+
+    parser.add_option('-v', '--uname', help=u"renvoie le uname -v",
+                      action='store_const', const='uname -v', dest='cmd', default='uname -v')
+    parser.add_option('-u', '--uptime', help=u"renvoie l'uptime",
+                      action='store_const', const='uptime', dest='cmd')
+    (options, args) = parser.parse_args()
+
+    names = bornes()
+    outputs = {}
+    for name in names:
+        outputs[name] = ssh_exec(name+".wifi", options.cmd)
+
+    # On attend que tous les fils qu'on a lancé terminent, avec une petite
+    # animation si la sortie est un terminal:
+    count = 0
+    total = len(names)
+    show_anim = sys.stderr.isatty()
+    message = "Réception des résultats: %d/%d"
+    if show_anim: sys.stderr.write(message % (count, total))
+    while count < total:
+        try: os.wait()
+        except OSError: break
+        count += 1
+    if show_anim: sys.stderr.write("\r" + message % (count, total))
+    if show_anim: sys.stderr.write("\n")
+
+    # On lit tout les résultats:
+    results = {}
+    for name, output in outputs.iteritems():
+        results[name] = output.read().strip()
+        output.close()
+
+    results = results.items()
+    results.sort(key=lambda x: x[0])
+
+    for name, result in results:
+        if not result:
+            print "%-15s failure" % (name + ":")
+
+    for name, result in results:
+        if result:
+            print "%-15s %s" % (name + ":", result)
+
+
+if __name__ == "__main__":
+    usage = u"""usage:       %prog [OPTION]
+Récupère la liste des bornes et récupère le résultat d'une commande sur
+toutes les bornes"""
+    if len(sys.argv[0].split('_', 1)) > 1:
+        plugin = sys.argv[0].split('_', 1)[0]
+        if plugin == 'uptime':
+            config = """graph_title Uptime Bornes
+graph_args --base 1000 -l 0
+graph_vlabel uptime in days"""
+            def process(uptime):
+                if uptime: return str(float (uptime.split()[1]) / (24 * 3600))
+                else: return '0'
+            munin(config, 'cat /proc/uptime', process=process)
+        else: raise NotImplementedError
+
+    else:
+        main()