161 lines
5.3 KiB
Python
161 lines
5.3 KiB
Python
#!/usr/bin/env python
|
|
# -*- coding: utf-8 -*-
|
|
|
|
# nolslib.py
|
|
# Ce script est inspiré de slonlib.py, développé par
|
|
# Jérémie Dimino en 2008.
|
|
# Développé pour communiquer en telnet avec une baie MSA 2012a (modèle à
|
|
# vérifier), il a été adapté pour communiquer avec une MSA P2000, qui sait
|
|
# retourner des sorties compatibles avec des api.
|
|
# ----------
|
|
# Auteur : Pierre-Elliott Bécue <peb@crans.org>
|
|
# Licence : GPLv3
|
|
|
|
'''Bibliothèque pour accéder à la baie de stockage nols, récupère les données
|
|
formatées en XML'''
|
|
|
|
import telnetlib
|
|
import 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 NolsError(Exception):
|
|
def __init__(self, msg):
|
|
Exception.__init__(self, msg)
|
|
|
|
|
|
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)
|
|
if resp.lower().startswith("error"):
|
|
raise NolsError(resp.replace("Error: ", ""))
|
|
|
|
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)
|
|
|
|
# XML c'est trobyien
|
|
Objects = tree.findall("OBJECT[@name='volume-view']")
|
|
|
|
for Object in Objects:
|
|
name = None
|
|
lun = None
|
|
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="slon1"):
|
|
'''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 = 1
|
|
while lun in map: lun = lun + 1
|
|
|
|
# Création du volume
|
|
result = self.cmd("create volume vdisk %s size %d%s lun %d %s" % (vdisk, size, unit, lun, name))
|
|
|
|
print "Le volume %s a été créé, son numéro 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, origin_name))
|