crans/scripts
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions
scripts/gestion/annuaires_pg.py
Daniel STAN 8f452efe65 [annuaires_pg] throw NameError si conn impossible 
C'est entre autre l'exception récupérée dans lc_ldap (pour l'attribut chambre),
qui permet de fallback en connexion de test. Ce n'est pas la manière idéale
de faire une connexion de test, mais c'est mieux que rien.
2013-07-24 13:21:36 +02:00

317 lines
10 KiB
Python
Executable file
Raw Blame History

#!/usr/bin/python
# -*- coding: utf-8 -*-
import psycopg2
from functools import wraps
conn = None
def _need_conn(f):
"""Décorateur à appliquer aux fonctions nécessitant une connexion pgsql"""
@wraps(f)
def first_connect(*args, **kwargs):
global conn
if conn == None:
if __name__ == 'annuaires_pg_test':
host='localhost'
else:
host='pgsql.adm.crans.org'
# "connecting …"
try:
conn = psycopg2.connect(user='crans', database='switchs', host=host)
except psycopg2.OperationalError:
# Les scripts appelant annuaires_pg n'ont pas à connaître le
# backend pgsql. On utilise donc une exception plus standard
raise NameError
return f(*args, **kwargs)
return first_connect
bat_switchs = ["a", "b", "c", "g", "h", "i", "j", "m", "o", "p"]
bat_manuels = []
@_need_conn
def chbre_prises(batiment, chambre = None):
"""Correspondance chambre -> prise"""
batiment = batiment.lower()
if chambre:
chambre = chambre.lower()
cur = conn.cursor()
cur.execute("SELECT prise_crans FROM prises WHERE (batiment, chambre) = (%s, %s)", (batiment, chambre))
try:
return "%03d" % cur.fetchone()[0]
except TypeError:
raise ValueError("Chambre inexistante bat %r, chbre %r" % (batiment, chambre))
else:
cur = conn.cursor()
cur.execute("SELECT chambre, prise_crans FROM prises WHERE batiment = %s", batiment)
ret = {}
for chambre, prise_crans in cur.fetchall():
ret[chambre] = "%03d" % prise_crans
if not ret:
raise ValueError("Batiment %s inexistant" % batiment)
return ret
@_need_conn
def chbre_commentaire(batiment, chambre):
""" Renvoie le commentaire associé à la chambre """
global conn
batiment = batiment.lower()
cur = conn.cursor()
cur.execute("SELECT commentaire FROM prises WHERE (batiment, chambre) = (%s,%s)", (batiment, chambre))
try:
return cur.fetchone()[0]
except TypeError:
raise ValueError("Chambre inexistante bat %r, chbre %r" % (batiment, chambre))
@_need_conn
def reverse(batiment, prise = None):
"""Correspondance prise -> chambre"""
batiment = batiment.lower()
if prise:
cur = conn.cursor()
cur.execute("SELECT chambre FROM prises WHERE (batiment, prise_crans) = (%s, %s)", (batiment, int(prise)))
try:
return [chbre for (chbre,) in cur.fetchall()]
except TypeError:
raise ValueError("Prise %s inexistante" % prise)
else:
cur = conn.cursor()
cur.execute("SELECT chambre, prise_crans FROM prises WHERE batiment = %s", batiment)
ret = {}
for chambre, prise_crans in cur.fetchall():
try:
ret["%03d" % prise_crans].append(chambre)
except KeyError:
ret["%03d" % prise_crans] = [chambre]
if not ret:
raise ValueError("Batiment %s inexistant" % batiment)
return ret
@_need_conn
def is_crans(batiment, chambre):
"""Chambre cablee au Cr@ns ?"""
batiment = batiment.lower()
chambre = chambre.lower()
cur = conn.cursor()
cur.execute("SELECT crans FROM prises WHERE (batiment, chambre) = (%s, %s)", (batiment, chambre))
return cur.fetchone()[0]
@_need_conn
def is_connected(batiment, chambre):
"""Cablage physique effectue ?"""
batiment = batiment.lower()
chambre = chambre.lower()
cur = conn.cursor()
cur.execute("SELECT cablage_effectue FROM prises WHERE (batiment, chambre) = (%s, %s)", (batiment, chambre))
return cur.fetchone()[0]
@_need_conn
def crous_to_crans(batiment, chambre):
"""Passage d'une chambre de CROUS a Cr@ns"""
batiment = batiment.lower()
chambre = chambre.lower()
if is_crans(batiment, chambre):
return
cur = conn.cursor()
cur.execute("UPDATE prises SET (crans, crous, cablage_effectue) = (TRUE, FALSE, not cablage_effectue) WHERE (batiment, chambre) = (%s, %s)", (batiment, chambre))
conn.commit()
cur.close()
# Prises d'uplink, de machines du crans / Prises d'utilité CRANS
uplink_prises={ 'a' :
{ 49 : 'uplink->bata-4', 50 : 'libre-service',
149 : 'uplink->bata-4', 150 : 'libre-service',
225 : 'uplink->bata-4', 226 : 'libre-service',
325 : 'uplink->bata-4', 326 : 'libre-service',
401 : 'uplink->bata-0', 402 : 'uplink->bata-1',
403 : 'uplink->bata-2', 404 : 'uplink->bata-3',
424 : 'uplink->backbone' },
'b' :
{ 49 : 'uplink->batb-4', 50 : 'libre-service',
149 : 'uplink->batb-4', 150 : 'libre-service',
249 : 'uplink->batb-4', 250 : 'libre-service', # 249 morte ?! (olasd 21/01/2010)
349 : 'uplink->batb-4', 350 : 'libre-service',
401 : 'uplink->batb-0', 402 : 'uplink->batb-1',
403 : 'uplink->batb-2', 404 : 'uplink->batb-3',
405 : 'uplink->backbone' },
'c' :
{ 49 : 'uplink->batc-3', 50 : 'libre-service',
149 : 'uplink->batc-3', 150 : 'libre-service',
225 : 'uplink->batc-3', 226 : 'libre-service',
301 : 'uplink->batc-0', 302 : 'uplink->batc-1',
303 : 'uplink->batc-2', 324 : 'uplink->backbone' },
'g' :
{
21 : 'libre-service', 22 : 'uplink->backbone',
23 : 'libre-service', 24 : 'uplink->batg-8',
149 : 'uplink->batg-8', 150 : 'libre-service',
249 : 'uplink->batg-8', 250 : 'uplink->batg-3',
325 : 'uplink->batg-8', 326 : 'libre-service',
449 : 'uplink->batg-9', 450 : 'uplink->batg-5',
549 : 'uplink->batg-9', 550 : 'uplink->batg-6',
649 : 'uplink->batg-9', 650 : 'uplink->batg-5',
725 : 'uplink->batg-9', 726 : 'libre-service',
801 : 'uplink->batg-1', 802 : 'uplink->batg-2',
803 : 'uplink->batg-3', 821 : 'uplink->batg-0',
823 : 'uplink->batg-9',
901 : 'uplink->batg-4', 902 : 'uplink->batg-5',
903 : 'uplink->batg-6', 904 : 'uplink->batg-7',
921 : 'uplink->batg-8',
},
'h' :
{ 49 : 'uplink->bath-3', 50 : 'libre-service',
149 : 'uplink->bath-3', 150 : 'libre-service',
225 : 'uplink->bath-3', 226 : 'libre-service',
301 : 'uplink->bath-0', 302 : 'uplink->bath-1',
303 : 'uplink->bath-2', 324 : 'uplink->backbone' },
'i' :
{ 49 : 'uplink->bati-3', 50 : 'libre-service',
149 : 'uplink->bati-3', 150 : 'libre-service',
301 : 'uplink->bati-0', 302 : 'uplink->bati-1',
324 : 'uplink->backbone' },
'j' :
{ 49 : 'uplink->batj-3', 50 : 'libre-service',
149 : 'uplink->batj-3', 150 : 'libre-service', # XXX: 150 semble morte
225 : 'uplink->batj-3', 226 : 'libre-service',
321 : 'uplink->backbone',
301 : 'uplink->batj-0', 303 : 'uplink->batj-1',
305 : 'uplink->batj-2', 307 : 'uplink->multiprise',
},
'k' : {
25 : 'uplink->backbone',
},
'm' :
{
49 : 'libre-service', 50 : 'uplink->batm-7',
149 : 'libre-service', 150 : 'uplink->batm-7',
249 : 'libre-service', 250 : 'uplink->batm-7',
349 : 'libre-service', 350 : 'uplink->batm-7',
449 : 'libre-service', 450 : 'uplink->batm-7',
549 : 'libre-service', 550 : 'uplink->batm-7',
649 : 'libre-service', 650 : 'uplink->batm-7',
724 : 'libre-service', 723 : 'libre-service',
722 : 'libre-service',
721 : 'uplink->backbone', 720 : 'uplink->batm-0',
719 : 'uplink->batm-1', 718 : 'uplink->batm-2',
717 : 'uplink->batm-3', 716 : 'uplink->batm-4',
715 : 'uplink->batm-5', 714 : 'uplink->batm-6',
},
'p' :
{
49 : 'uplink->batp-4',
149: 'uplink->batp-4',
249: 'uplink->batp-4',
350: 'uplink->batp-4',
# On ne génère pas la conf de batp-4 automatiquement, mais ses uplinks
# peuvent être utiles à connaître
401: 'uplink->batp0', 402: 'uplink->batp-1',
403: 'uplink->batp2', 403: 'uplink->batp-3',
405: 'libre-service', 406: 'uplink->bato-1',
},
'o' :
{
25 : 'uplink->bato-1', 26 : 'libre-service',
101 : 'uplink->bato-0', 121: 'uplink->NRD',
122: 'uplink->backbone', 123: 'uplink->backbone (unused)',
124: 'uplink->batp-0'
} ,
'backbone' : #For your consideration
{
#A: 12eth+12fibre, B: 24 eth
'A1': 'malloc-0', 'B1': 'bata',
'A2': 'komaz-ens', 'B2': 'multiprise-wifi',
'A3': 'sable', 'B3': 'dyson',
'A4': 'komaz', 'B4': 'fy',
'A5': 'zbee' , 'B5': 'switch-ilo',
'B6': 'vigile 0B',
'B7': 'kdell',
'B8': 'batb',
'B9': '2b',
'B10': 'fz',
'B11': 'gordon',
'B12': 'nols2',
'A13': 'batm',
'A14': 'batp', 'B14': 'zamok',
'A15': 'batc', 'B15': 'charybde',
'A16': 'bato',
'A17': 'bath',
'A18': 'bati', 'B18': 'nols',
'A19': 'batj',
'A20': 'batg',
'A21': 'batk',
'B21': 'osm1',
'B22': 'osm1-ilo',
'B23': 'osm2',
'B24': 'osm2-ilo',
},
}
_SPECIAL_SWITCHES=['backbone.adm.crans.org',
'multiprise-v6.adm.crans.org',
'batk-0.crans.org',
'batp-4.adm.crans.org',
'minigiga.adm.crans.org',
]
def all_switchs(bat=None, hide=_SPECIAL_SWITCHES):
"""Retourne la liste des switchs pour un batiment.
Si bat est donné, seulement pour le bâtiment demandé, sinon pour
tous les bâtiments. bat peut être une liste aussi. Le backbone n'est
pas pris en compte. La convention est batx-y sauf si y=0 et on a donc
simplement batx"""
if bat == None:
bat = bat_switchs
if type(bat) not in [ tuple, list ] :
bat = [bat]
switchs = []
for b in bat:
indexes = set(n/100 for n in uplink_prises[b])
for i in indexes:
hostname = "bat%s-%s.adm.crans.org" % (b, i)
if hostname not in hide:
switchs.append(hostname)
# on ajoute quand-même le backbone et/ou multiprise-v6 si demandé
switchs += set(_SPECIAL_SWITCHES).difference(hide)
switchs.sort()
return switchs
# Locaux clubs : lecture dans chbre_prises et ajout des locaux dans les bats non
# manageables
def locaux_clubs() :
""" Retourne le dictionaire des locaux club : {bat: [locaux]} """
# Corespondance chbre -> nom du local club
locaux_clubs = { 'Bcl0' : 'Kfet' ,
'Bcl1' : 'Med',
'Pcl0' : 'Bds' ,
'Mcl0' : 'Shape',
'Mcl1' : 'Krobot',
'EXT' : 'EXT' }
# Ajout des locaux d'étage A, B et C
for b in 'ABC' :
for i in range(2,7) :
locaux_clubs['%scl%i' % ( b, i)] = '%i@%s' % (i, b)
# Ajout de ceux des H, I et J
for b in 'HIJ' :
for i in range(1,5) :
locaux_clubs['%scl%i' % ( b, i)] = '%i@%s' % (i, b)
# Supression du 2@B et 4@J
locaux_clubs.pop('Bcl2')
locaux_clubs.pop('Jcl4')
return locaux_clubs
Reference in a new issue View git blame Copy permalink