scripts/gestion/ressuscite_lc.py

80 lines
3.3 KiB
Python
Executable file

#!/bin/bash /usr/scripts/python.sh
# -*- mode: python; coding: utf-8 -*-
################# ATTENTION #####################
# À ne pas laisser executable par n'importe qui,#
# permet de récupérer un shell vers lc_ldap #
#################################################
import sys
import pprint
from affich_tools import coul, prompt
from IPython.frontend.terminal.embed import InteractiveShellEmbed
import contextlib
import lc_ldap.shortcuts
import lc_ldap.attributs
import lc_ldap.crans_utils
import lc_ldap.printing
def reset_ip(machine):
# Ne marche plus parce que _find_id pose un lock puis machine.__setitems__ aussi, il faut voir pour faire autrement
machine['rid']=unicode(conn._find_id('rid', lc_ldap.crans_utils.find_rid_plage(machine['rid'][0].value)[0], lockId=machine.lockId))
machine['ipHostNumber']=lc_ldap.attributs.ipHostNumber.default
machine['ip6HostNumber']=lc_ldap.attributs.ip6HostNumber.default
def reset_mid(machine, proprio=None):
# Ne marche plus parce que _find_id pose un lock puis machine.__setitems__ aussi, il faut voir pour faire autrement
machine['mid']=unicode(conn._find_id('mid', lockId=machine.lockId))
if proprio:
machine.dn="mid=%s,aid=%s,%s" % (machine['mid'][0], proprio['aid'][0], machine.dn.split(',',2)[2])
else:
print coul("Pas de proprio de fourni, on suppose que l'aid du dn %s est valide" % machine.dn.split(',',2)[1], "jaune")
machine.dn="mid=%s,%s" % (machine['mid'][0], machine.dn.split(',',1)[1])
def display():
global coffins
cmp=0
for i in coffins:
print coul(u"Élement %s de coffins :" % cmp, 'gras')
i.display()
cmp+=1
class Edit(Exception):
pass
if __name__ == '__main__':
if len(sys.argv) >2:
conn=lc_ldap.shortcuts.lc_ldap_admin()
try:
coffins=conn.gravedig(sys.argv[1], sys.argv[2])
if not coffins:
print "Nothing to do"
elif len(coffins)==1:
print lc_ldap.printing.sprint(coffins[0])
ret=prompt("Ressuciter ?[y,n,e]", "N")
if ret.lower() == 'y':
with coffins[0] as machine:
machine.create()
if ret.lower() == 'e':
raise Edit("Édition manuelle")
else:
raise lc_ldap.attributs.UniquenessError("Plus d'un objet machtant %s" % sys.argv[2])
except (lc_ldap.attributs.UniquenessError, Edit) as error:
msg=coul("%s, démarrage d'un shell." % error, 'rouge')
msg+="""
Les objets ressucitables sont dans la variable \033[1;50mcoffins\033[1;0m.
Une connexion admin à la base ldap dans la variable \033[1;50mconn\033[1;0m.
Pour afficher les objects de \033[1;50mcoffins\033[1;0m, vous pouvez utiliser la fonction \033[1;50mdisplay()\033[1;0m.
Pour ressuciter les machines, il y a en plus les fonctions :
* reset_mid(machine, proprio=None)
* reset_ip(machine)
"""
ipshell = InteractiveShellEmbed.instance(banner1 = unicode(msg, 'utf-8'))
with contextlib.nested(*coffins) as coffins:
ipshell()
sys.exit(0)
except ValueError as error:
print coul("%s" % error, "rouge")
else:
sys.stderr.write("Usage: %s {cimetière} {filtre}\nLe cimetière est dans /home/cimetiere_lc/\n" % sys.argv[0])