[Baie de disques] On remplace slon par nols là où c'est bien.

Ignore-this: 26bbaee2a355c1c1b8170d1fa58aea59

darcs-hash:20120807200036-ab199-cfb522d47031cd8ecca7c8961f430d9034df6315.gz

gestion/iscsi/nolslib.py

@@ -0,0 +1,141 @@
+#!/usr/bin/env python
+# -*- coding: utf-8 -*-
+
+# baie_lib.py
+# ----------
+# Type à taper si ça chie : Pierre-Elliott Bécue <peb@crans.org> 
+# Licence   : WTFPL
+
+'''Bibliothèque pour accéder à la baie de stockage nols, récupère les données
+formatées en XML'''
+
+import telnetlib, re
+from xml.etree.ElementTree import ElementTree, fromstring
+
+# Message envoyé par le serveur pour attendre l'appuie sur touche
+junk_regexp = re.compile("Press any key to continue \(Q to quit\)\r *\r")
+
+# Matching des fin de lignes avec \r\n
+crlf_regexp = re.compile("\r\n")
+
+username = ""
+password = ""
+
+# Récupère des identifiants
+execfile("/etc/crans/secrets/nols.py")
+
+class Nols(object):
+    '''Objet représentant la baie de stockage'''
+
+    def __init__(self, host="nols.adm.crans.org"):
+        self.tn = telnetlib.Telnet(host)
+
+        # Identification
+        self.tn.expect(["login: "])
+        self.tn.write(username + "\r\n")
+        self.tn.read_until("Password: ")
+        self.tn.write(password + "\r\n")
+
+        # Ça c'est pour attendre l'invite de commande initiale:
+        self.tn.read_until("# ")
+
+        # On veut des sorties en XML
+        self.cmd("set cli-parameters api-embed")
+
+        # On se débarasse des "appuie sur un bouton pour voir la suite"
+        self.cmd("set cli-parameters pager off")
+
+    def logout(self):
+        '''Déconnexion de la baie'''
+        self.tn.write("exit\r\n")
+        self.tn.read_all()
+        self.tn.close()
+        print ("Si vous avez effectué des modifications pensez à exécuter:\n" +
+               "/usr/scripts/gestion/iscsi/update.sh sur chacun des dom0\n")
+
+    def cmd(self, cmd, mode='text'):
+        '''Exécute une commande et renvoie le résultat. Lève
+        l'exception Error si la commande échoue'''
+
+        # Si c'est le script qui bosse, on utilise le mode XML, sinon
+        # on peut aussi lancer des commandes en mode texte
+        if mode == 'XML':
+            self.tn.write("set cli-parameters api-embed\r\n")
+            self.tn.read_until("# ")
+        else:
+            self.tn.write("set cli-parameters console\r\n")
+            self.tn.read_until("# ")
+
+        self.tn.write(cmd + "\r\n")
+        resp = ""
+        # Lecture de la réponse, c'est terminé quand on atteint un
+        # nouveau prompt:
+        resp = self.tn.read_until("# ")
+
+        # On retire les messages parasites s'il en reste par hasard
+        resp = junk_regexp.sub("", resp)
+
+        # On vire la commande qui est là et dont on veut pas
+        [_, resp] = resp.split(cmd+"\r\n", 1)
+
+        # On retire le prompt
+        [resp, _] = resp.rsplit("\r\n", 1)
+
+        # Remplace les fins de ligne dos par des fin de lignes unix
+        resp = crlf_regexp.sub("\n", resp)
+        return resp
+
+    def show(self, what):
+        '''Raccourci pour: print slon.cmd("show <what>")'''
+        print self.cmd("show " + what)
+
+    def help(self, what = ""):
+        '''Raccourci pour: print slon.cmd("help <what>")'''
+        print self.cmd("help " + what)
+
+    def volume_map(self):
+        '''Retourne le mapping lun<->nom de volume'''
+        map = {}
+        XML_map = self.cmd("show volume-map", mode="XML")
+        root = fromstring(XML_map)
+        tree = ElementTree(root)
+        Objects = tree.findall("OBJECT[@name='volume-view']")
+        for Object in Objects:
+            name = Object.findall("PROPERTY[@name='volume-name']")[0].text
+            lun = Object.findall("OBJECT/PROPERTY[@name='lun']")[0].text
+            if lun is None:
+                pass
+            else:
+                map[int(lun)] = name
+        return map
+
+    def create_volume(self, name, size, unit="GiB", vdisk="VD1"):
+        '''Crée un nouveau volume. Retourne le lun sur lequel il est
+        mappé. L'unité doit être "B", "K(i)B", "M(i)B", "G(i)B" ou T(i)B.
+        Par défault l'unité est le giga octet binaire : Gib.'''
+
+        # On récupère le mapping pour chercher un lun de libre
+        map = self.volume_map()
+        lun = 0
+        while lun in map: lun = lun + 1
+
+        # Création du volume
+        self.cmd("create volume vdisk %s size %d%s lun %d %s" % (vdisk, size, unit, lun, name))
+
+        return "Le volume %s a été créé, son numéri d'identification est %d" %(name, lun)
+
+    def delete_volume(self, name):
+        '''Supprime un volume'''
+
+        self.cmd("delete volume %s" % (name))
+
+    def expand_volume(self, name, size, unit="GiB"):
+        '''Permet d'étendre un volume, la taille par défaut est le giga octet
+        binaire (GiB), les unités possibles sont B, K(i)B, M(i)B, G(i)B, T(i)B.'''
+
+        self.cmd("expand volume size %d%s %s" % (size, unit, name))
+
+    def rename_volume(self, origin_name, new_name):
+        '''Renomme un volume.'''
+    
+        self.cmd("set volume name %s %s" % (new_name, orig_name))