Factorisation de info_machine.

darcs-hash:20061211220244-68412-4f883702e90d2ed9d24b8c0fe41f62e035094960.gz

gestion/tools/locate-mac.py

@@ -1,18 +1,21 @@
 #! /usr/bin/env python
 # -*- coding: iso-8859-15 -*-
 
-import sys
-sys.path.append('/usr/scripts/gestion')
-from annuaires import reverse, all_switchs
-from hptools import hpswitch
+import sys, sre
 from time import sleep
 from os import system
 import threading
 
-# mise en forme d'une adresse mac 
+sys.path.append('/usr/scripts/gestion')
+from ldap_crans import crans_ldap
+import whos
+from annuaires import reverse, all_switchs
+from hptools import hpswitch
+
+# mise en forme d'une adresse mac
 def format_mac(unformated_mac):
-    return ":".join([i.zfill(2) for i in unformated_mac.split(":")]).lower()
-        
+    return str(":".join([i.zfill(2) for i in unformated_mac.split(":")]).lower())
+
 
 # classe d'interrogation des switchs
 class interroge_switch (threading.Thread) :
@@ -35,30 +38,47 @@ class interroge_switch (threading.Thread) :
             if nom.find("uplink")==-1:
                 self.reponse = ("%-10s => prise %-2s : %s" % (self.switch.encode('iso-8859-15').replace('.adm.crans.org',''), str(prise), nom))
 
-# interrogation des switchs en parallele
-def get_trace(mac):
 
-    tableau=[]
-    
+# Retourne les infos sur la machine (l'équivalent d'un whos, mais renvoie la
+# chaîne de caractères)
+def info_machine(mac):
+    s = u""
+    db = crans_ldap()
+    machines = db.search("mac=%s" % mac)['machine']
+    for m in machines:
+        r = whos.machine_details(m)
+        # On supprime les couleurs
+        r = sre.sub('\x1b\[1;([0-9]|[0-9][0-9])m', '', r)
+        s += r
+    if len(machines) == 0:
+        s = u"Recherche LDAP de la MAC %s : aucune machine trouvée\n" % mac
+    return s
+
+
+# interrogation des switchs en parallele
+def trace_machine(mac):
+    tableau = []
+
     for switch in ['backbone'] + all_switchs():
         tableau.append(interroge_switch(switch, mac))
 
     while len(threading.enumerate()) > 1 :
         sleep(1)
 
-    resultat ='tracage de %s\n' % mac
+    resultat = u'Traçage de %s...\n' % mac
 
-    tracage=''
+    tracage = u''
     for t in tableau:
         if t.reponse:
-            tracage += t.reponse +"\n"
-    if tracage=='':
-        tracage = "mac inconnue des switchs"
+            tracage += t.reponse + u"\n"
+    if tracage == u'':
+        tracage = u"Adresse MAC inconnue des switchs\n"
 
     return (resultat + tracage)
-    
+
+
 # on interroge les switchs et on fait un whos sur la mac
 if __name__ == '__main__':
     mac = format_mac(sys.argv[1])
-    print get_trace(mac)
+    print trace_machine(mac)
     print system('/usr/scripts/gestion/whos.py -a mac=%s' % mac)