[Scripts] On rajoute un champ rid qui fait le lien avec les plages d'ip, et on rend le mid strictement croissant.

Ignore-this: 199e9ff5f09e1fe600c1066179f4e47b
Ce patch est un test, il ne restera en prod que si ça fonctionne. L'idée est qu'on souhaiterait conserver les vieilles machines comme les vieux adhérents, sauf demande explicite de suppression, par ailleurs, l'association mid <=> ip est très utile pour pas mal de choses. Pour la conserver, on crée un identifiant rid, qui supplante le mid, qui est lui choisi comme l'aid ou le fid, en incrémentant.

Ce patch vise à implémenter cela. S'il génère des bugs, il subira un rollback.

darcs-hash:20130123021650-b6762-347428d75f066f7f4821ca067d8c9bb6a4396bf5.gz

gestion/ridtools.py

@@ -0,0 +1,151 @@
+#!/usr/bin/env python
+# -*- coding: utf-8 -*-
+#
+# RIDTOOLS.PY -- Gestion de la conversion rid <-> IP
+#
+# Copyright (C) 2010 Olivier Iffrig
+#           (c) 2010 Nicolas Dandrimont
+# Authors: Olivier Iffrig <iffrig@crans.org>
+#          Nicolas Dandrimont <olasd@crans.org>
+# Adapté par Pierre-Elliott Bécue pour cause de changement de schéma.
+#
+# This program is free software: you can redistribute it and/or modify
+# it under the terms of the GNU General Public License as published by
+# the Free Software Foundation, either version 3 of the License, or
+# (at your option) any later version.
+#
+# This program is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+# GNU General Public License for more details.
+#
+# You should have received a copy of the GNU General Public License
+# along with this program.  If not, see <http://www.gnu.org/licenses/>.
+
+import netaddr
+
+import config
+
+class Rid(object):
+    """
+    Permet de décortiquer un rid et d'obtenir les IP correspondantes
+
+    Variables d'instance :
+     - rid : le rid
+     - type : type de rid (cf. config.rid et [1])
+     - reste : le "reste" du rid (permettant de calculer l'IP par exemple)
+     - ipv4_dispo : une IPv4 existe pour ce rid
+     - priv : cette machine ne doit pas être accessible de l'extérieur
+
+    [1] http://wiki.crans.org/CransTechnique/PlanAdressage#Machines
+    """
+
+    def __init__(self, rid=None, ipv4=None):
+        self.rid = None
+        self.reste = False
+        self.type = None
+        self.ipv4_dispo = False
+        self.priv = False
+
+        if rid is not None:
+            self.__parse(rid)
+        elif ipv4 is not None:
+            self.__from_ipv4(ipv4)
+        else:
+            raise ValueError("Un des champs rid et ipv4 doit être défini")
+
+    def __parse(self, rid):
+        """Peuple les champs de l'instance par rapport au rid"""
+        self.rid = rid
+
+        for tp, (start, end) in config.rid.iteritems():
+            if start <= rid <= end:
+                self.type = tp
+                break
+        else:
+            raise ValueError("rid inconnu : %d" % rid)
+
+        self.ipv4_dispo = (rid & (1 << 15)) == 0 and self.type != 'special' or self.rid in config.rid_machines_speciales
+
+        self.priv = (rid & (1 << 14)) != 0
+
+        if self.type == 'personnel-ens':
+            self.reste = rid & 0xff
+        else:
+            self.reste = rid & 0x7ff
+
+    def __from_ipv4(self, ip):
+        """Peuple les champs à partir de l'ipv4 donnée"""
+        if not isinstance(ip, netaddr.IPAddress):
+            ip = netaddr.IPAddress(ip)
+        try:
+            ip.ipv4()
+        except netaddr.AddrConversionError:
+            raise ValueError("Ceci n'est pas une IPv4... %s" % ip)
+
+        self.ipv4_dispo = True
+        self.priv = ip.is_private()
+
+        for tp in ['fil', 'wifi', 'adm', 'gratuit', 'personnel-ens']:
+            for net in config.NETs[tp]:
+                if ip in netaddr.IPNetwork(net):
+                    self.type = tp
+                    break
+            if self.type:
+                break
+        else:
+            for rid, ip_speciale in config.rid_machines_speciales.items():
+                if ip == netaddr.IPAddress(ip_speciale):
+                    self.rid = rid
+                    self.type = "special"
+                    break
+            else:
+                raise ValueError("%s dans aucun des réseaux gérés par le Cr@ns..." % ip)
+
+        if not self.rid:
+            self.rid = config.rid[self.type][0] + ip.value - netaddr.IPNetwork(config.NETs[self.type][0]).value
+
+        if self.rid > config.rid[self.type][1]:
+            raise ValueError("%s trop élevée pour le réseau '%s'" % (ip, self.type))
+
+        if self.type == 'personnel-ens':
+            self.reste = self.rid & 0xff
+        else:
+            self.reste = self.rid & 0x7ff
+        return self.rid
+
+    def ipv4(self):
+        """
+        Génère l'IPv4 associée à la machine
+        """
+        if not self.ipv4_dispo:
+            raise ValueError("Pas d'adresse IPv4 disponible pour la machine %r" % self)
+
+        if not hasattr(self, '__ipv4'):
+            if self.type != "special":
+                net = netaddr.IPNetwork(config.NETs[self.type][0])
+                self.__ipv4 = netaddr.IPAddress(net.first + self.reste)
+            else:
+                self.__ipv4 = netaddr.IPAddress(config.rid_machines_speciales[self.rid])
+
+        return self.__ipv4
+
+    def ipv6_network(self):
+        """
+        Génère le réseau /64 IPv6 associé à la machine
+        """
+        if self.priv:
+            raise ValueError("Pas de prefix ipv6 disponible pour cette machine")
+
+        global_net = netaddr.IPNetwork(config.prefix["subnet"][0])
+        global_net.prefixlen = 64
+        return global_net.next(self.rid)
+
+    def __repr__(self):
+        if self.ipv4_dispo:
+            return "<Rid(rid=%s, ipv4='%s')>" % (self.rid, self.ipv4())
+        else:
+            return "<Rid(%s)>" % self.rid
+    
+    def __int__(self):
+        return self.rid