3c7fcb3e63
Quelques petites modifications :
* Réorganisation du script sous forme d'une fonction pour pouvoir utiliser la fonction
de scan depuis un autre script
* Affichage des prises avec des VLANs bizarres mais sans machine connectée
* Gestion des erreurs de communication avec les switchs :
Si un switch ne répond pas, hptools.py ne renvoie pas d'erreur à la connection
mais rale à la première commande envoyée...
darcs-hash:20090222202703-b8e4c-45ba54d74a93fdc05d13f2c6588eb8231d7355cb.gz
142 lines
5.6 KiB
Python
Executable file
142 lines
5.6 KiB
Python
Executable file
#!/usr/bin/env python
|
|
# -*- coding: utf-8 -*-
|
|
|
|
u"""
|
|
Script permettant d'obtenir des statistiques sur les VLans actuellement distribués par les switchs.
|
|
|
|
Le script va scanner toutes les prises présentes dans le fichier d'annuaire.
|
|
S'il est impossbile de se connecter à un switch, on abandonne le batiment concerné
|
|
|
|
À la fin sont affichées quelques statistiques, dont les prises sur lesquelles on a trouvé
|
|
d'autre VLANs qu'"adherent", ainsi que les machines connectées sur ces prises.
|
|
|
|
Xavier Lagorce - 2009
|
|
"""
|
|
|
|
from sys import path
|
|
|
|
path.append('/usr/scripts/gestion')
|
|
|
|
from annuaires import chbre_prises
|
|
from affich_tools import *
|
|
from ldap_crans import crans_ldap
|
|
from ldap_crans import MachineCrans, MachineWifi, BorneWifi
|
|
from hptools import hpswitch, ConversationError
|
|
|
|
|
|
# Accès à la base ldap
|
|
db = crans_ldap()
|
|
|
|
def scan_bats(bats, annuaire=chbre_prises):
|
|
"""Fonction permettant de scanner un certain nombre de switchs
|
|
|
|
La fonction scanne les switchs enregistrés de annuaire.py se trouvant dans
|
|
les bâtiments contenus dans la liste bats.
|
|
|
|
La fonction utilise l'annuaire passé en argument au celui de annuaire.py
|
|
par défaut
|
|
|
|
La fonction renvoie un doublet (vlans_stats,data) :
|
|
* vlans_stats est un dictionnaire contenant les statistiques des VLANs
|
|
trouvés
|
|
* data contient les informations sur les machines se trouvant sur des
|
|
prises possédant d'autres VLANs que le VLAN 'adhérent'
|
|
|
|
"""
|
|
|
|
# Initialistation des données :
|
|
# nombre de prises possédant chaque VLAN
|
|
vlans_stats = dict([('adherent',0)])
|
|
# liste des résultats particuliers découverts pendant le scan
|
|
data = []
|
|
|
|
# On Commence par itérer sur chaque batiment :
|
|
for bat in chbre_prises:
|
|
prises = dict()
|
|
for chbre, prise in chbre_prises[bat].iteritems():
|
|
prises[prise] = chbre
|
|
|
|
current_switch = -1
|
|
switch_name = ' '
|
|
|
|
for prise in sorted(prises.keys()):
|
|
|
|
# Si l'on est pas connecté au switch sur lequel se trouve la prise, on en change
|
|
if current_switch != int(prise[0]) :
|
|
switch_name = u"bat" + bat + u"-" + prise[0] + u".adm.crans.org"
|
|
current_switch = int(prise[0])
|
|
print u"Connexion à " + switch_name + u" ..."
|
|
try:
|
|
switch = hpswitch(switch_name)
|
|
except ValueError:
|
|
print u"\tImpossible de se connecter à ce switch, abandon du bâtiment..."
|
|
break
|
|
else:
|
|
print u"\tConnexion établie"
|
|
|
|
# On récupère les VLANs présents sur la prise
|
|
try:
|
|
vlans = switch.vlans(prise)
|
|
except ConversationError:
|
|
print u"\tErreur de communication avec " + switch_name
|
|
print u"\tAbandon du bâtiment..."
|
|
break
|
|
|
|
if vlans:
|
|
# Si la prise ne possède que le VLAN adherent, on ne va pas plus loin
|
|
if len(vlans) == 1 and vlans[0] == u"adherent":
|
|
vlans_stats[vlans[0]] = vlans_stats[vlans[0]] + 1
|
|
else:
|
|
# Sinon, on récupère tous les VLANs
|
|
for vlan in vlans:
|
|
if not (vlan in vlans_stats):
|
|
vlans_stats[vlan] = 0
|
|
vlans_stats[vlan] = vlans_stats[vlan] + 1
|
|
# On récupère les données pour les afficher plus tard
|
|
try:
|
|
macs = switch.show_prise_mac(int(prise[1:]))
|
|
except ConversationError:
|
|
print u"\tErreur de communication avec " + switch_name
|
|
print u"\tAbandon du bâtiment..."
|
|
break
|
|
|
|
if macs:
|
|
for mac in macs:
|
|
fm = db.search("mac=%s" % mac)
|
|
if len(fm["machine"]) != 0:
|
|
m = fm["machine"][0]
|
|
if isinstance(m, MachineWifi): t = u"Wifi"
|
|
elif isinstance(m, BorneWifi): t = u"Born"
|
|
else : t = u"Fixe"
|
|
adh = m.proprietaire()
|
|
nom = adh.Nom()
|
|
data.append([bat.upper() + prise,
|
|
adh.chbre(),
|
|
adh.Nom(),
|
|
m.nom().split('.')[0],
|
|
t,
|
|
', '.join(sorted(vlans))])
|
|
# Si aucune machine n'est connectée, on ajoute quand même une
|
|
# entrée dans les résultats
|
|
else:
|
|
data.append([bat.upper() + prise,
|
|
bat.upper() + prises[prise],
|
|
'Pas de Machine',
|
|
'-',
|
|
'-',
|
|
', '.join(sorted(vlans))])
|
|
|
|
# On renvoie les résultats
|
|
return (vlans_stats,data)
|
|
|
|
# Traitement par défaut si le fichier est directement appellé comme un script
|
|
if __name__ == "__main__":
|
|
|
|
# On récupère les statistiques sur toutes les prises de l'annuaire
|
|
vlans_stats, data = scan_bats(chbre_prises.keys)
|
|
|
|
# Et on les affiche
|
|
print u"Statistiques des VLANs : ", vlans_stats
|
|
print u""
|
|
print u"Prises possédant des VLANs différent d''adherent'"
|
|
print tableau(data, titre = [u"Prise", u"Chambre", u"Adhérent", u"Machine", u"Type", u"VLANs"])
|