176 lines
6.3 KiB
Python
176 lines
6.3 KiB
Python
#!/usr/bin/env python
|
|
# -*- coding: utf-8 -*-
|
|
#
|
|
# CRANS_UTILS.PY-- Utils for Cr@ns gestion
|
|
#
|
|
# Copyright (c) 2010, Cr@ns <roots@crans.org>
|
|
# Author: Antoine Durand-Gasselin <adg@crans.org>
|
|
# All rights reserved.
|
|
#
|
|
# Redistribution and use in source and binary forms, with or without
|
|
# modification, are permitted provided that the following conditions are met:
|
|
# * Redistributions of source code must retain the above copyright
|
|
# notice, this list of conditions and the following disclaimer.
|
|
# * Redistributions in binary form must reproduce the above copyright
|
|
# notice, this list of conditions and the following disclaimer in the
|
|
# documentation and/or other materials provided with the distribution.
|
|
# * Neither the name of the Cr@ns nor the names of its contributors may
|
|
# be used to endorse or promote products derived from this software
|
|
# without specific prior written permission.
|
|
#
|
|
# THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND
|
|
# ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
|
|
# WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
|
|
# DISCLAIMED. IN NO EVENT SHALL <COPYRIGHT HOLDER> BE LIABLE FOR ANY
|
|
# DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
|
|
# (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
|
|
# LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
|
|
# ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
|
|
# (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
|
|
# SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
|
|
|
|
import calendar, netaddr, re, time, smtplib, sys
|
|
sys.path.append('/usr/scripts/gestion')
|
|
import config
|
|
from unicodedata import normalize
|
|
|
|
def ip4_of_rid(rid):
|
|
"""Convertit un rid en son IP associée"""
|
|
for net, plage in config.rid.items():
|
|
if rid >= plage[0] and rid <= plage[1]:
|
|
break
|
|
else:
|
|
raise ValueError("Rid dans aucune plage: %d" % rid)
|
|
|
|
if net == 'special':
|
|
try:
|
|
return netaddr.IPAddress(config.rid_machines_speciales[rid])
|
|
except KeyError:
|
|
return ValueError(u"Machine speciale inconnue: %d" % rid)
|
|
|
|
return netaddr.IPAddress(netaddr.IPNetwork(config.NETs[net][0]).first + rid - plage[0])
|
|
|
|
def prefixev6_of_rid(rid):
|
|
"""
|
|
L'ip de sous-réseau privé d'une machine. L'adhérent en fait ce qu'il veut, mais c'est la machine
|
|
associée au rid qui est responsable du traffic.
|
|
|
|
Cette fonction retourne l'ip de début de ce sous-réseau.
|
|
"""
|
|
for net, plage in config.rid.items():
|
|
if rid >= plage[0] and rid <= plage[1]:
|
|
break
|
|
else:
|
|
raise ValueError("Rid dans aucune plage: %d" % rid)
|
|
|
|
# fil-v6 ou wifi-v6, we don't care
|
|
return netaddr.IPAddress(netaddr.IPNetwork(config.prefix['fil-v6'][0]).first + 2**64*rid)
|
|
|
|
def ip6_of_mac(mac, rid):
|
|
"""
|
|
Retourne la bonne ipv6 de la machine en fonction de sa mac et de son rid.
|
|
"""
|
|
|
|
for net, plage in config.rid.items():
|
|
if rid >= plage[0] and rid <= plage[1]:
|
|
break
|
|
else:
|
|
raise ValueError("Rid dans aucune plage: %d" % rid)
|
|
|
|
print net
|
|
|
|
# En théorie, format_mac est inutile, car on ne devrait avoir
|
|
# que des mac formatées.
|
|
mac = format_mac(mac).replace(':', '')
|
|
|
|
# hex retourne un str, donc on concatène, suivant la RFC
|
|
euid64v6 = hex(int(mac[:2], 16)^0b00000010) + mac[2:6] + 'fffe' + mac[6:12]
|
|
|
|
# fil-v6 ou wifi-v6, we don't care
|
|
return netaddr.IPAddress(netaddr.IPNetwork(config.prefix[net][0]).first + int(euid64v6, 16))
|
|
|
|
def strip_accents(a):
|
|
""" Supression des accents de la chaîne fournie"""
|
|
res = normalize('NFKD', a).encode('ASCII', 'ignore')
|
|
return unicode(res)
|
|
|
|
def strip_spaces(a):
|
|
""" Suppression des espaces et des apostrophes"""
|
|
return a.replace(u' ', u'_').replace(u"'", u'')
|
|
|
|
def mailexist(mail):
|
|
"""Vérifie si une adresse mail existe ou non grace à la commande
|
|
vrfy du serveur mail """
|
|
|
|
mail = mail.split('@', 1)[0]
|
|
# try:
|
|
s = smtplib.SMTP('smtp.adm.crans.org')
|
|
s.putcmd("vrfy", mail)
|
|
r = s.getreply()[0] in [250, 252]
|
|
s.close()
|
|
# except:
|
|
# raise ValueError(u'Serveur de mail injoignable')
|
|
|
|
return r
|
|
|
|
def format_ldap_time(tm):
|
|
u"""Formatage des dates provenant de la base LDAP
|
|
Transforme la date YYYYMMDDHHMMSS.XXXXXXZ (UTC)
|
|
en date DD/MM/YY HH:MM (local)"""
|
|
tm_st = time.strptime(tm.split('.')[0], "%Y%m%d%H%M%S") # struct_time UTC
|
|
timestamp = calendar.timegm(tm_st)
|
|
tm_st = time.localtime(timestamp) # struct_time locale
|
|
return time.strftime("%d/%m/%Y %H:%M", tm_st)
|
|
|
|
def format_mac(mac):
|
|
u""" Formatage des adresses mac
|
|
Transforme une adresse pour obtenir la forme xx:xx:xx:xx:xx:xx
|
|
Retourne la mac formatée.
|
|
"""
|
|
mac = netaddr.EUI(mac)
|
|
if not mac:
|
|
raise ValueError(u"MAC nulle interdite\nIl doit être possible de modifier l'adresse de la carte.")
|
|
return str(mac).replace('-', ':')
|
|
|
|
def format_tel(tel):
|
|
u"""Formatage des numéros de téléphone
|
|
Transforme un numéro de téléphone pour ne contenir que des chiffres
|
|
(00ii... pour les numéros internationaux)
|
|
Retourne le numéro formaté.
|
|
"""
|
|
tel_f = tel.strip()
|
|
if tel_f.startswith(u"+"):
|
|
tel_f = u"00" + tel_f[1:]
|
|
if u"(0)" in tel_f:
|
|
tel_f = tel_f.replace(u"(0)", u"")
|
|
tel_f = re.sub(r'\D', '', tel_f)
|
|
return tel_f
|
|
|
|
def validate_name(value, more_chars=""):
|
|
"""Valide un nom: ie un unicode qui contient lettres, espaces et
|
|
apostrophes, et éventuellement des caractères additionnels"""
|
|
if re.match("^[A-Za-z0-9]([-' %s]?[A-Za-z0-9])*$" % more_chars,
|
|
normalize('NFKD', value).encode('ASCII', 'ignore')):
|
|
return value
|
|
else:
|
|
raise ValueError("Nom invalide ('%s')" % value)
|
|
|
|
def ldap_sanitize(s):
|
|
""" Échappe les caractères spéciaux ldap.
|
|
(Si vous avez une méthode plus propre pour coder ça, je suis preneur)
|
|
Todo: rajouter conversion pour caractère utf-8 multioctets ?
|
|
(cf http://blog.dzhuvinov.com/?p=585 )
|
|
"""
|
|
replace = {
|
|
'*': '\\2a',
|
|
'(': '\\28',
|
|
')': '\\29',
|
|
'\\': '\\5c',
|
|
'\x00': '\\00'}
|
|
def conv(c):
|
|
try: return replace[c]
|
|
except KeyError: return c
|
|
return "".join([conv(c) for c in s])
|
|
|
|
|
|
|