Systme de secours.

darcs-hash:20050826160639-41617-310b0023890ac12103143c0cfea9d9f615510539.gz

secours/secours.py

@@ -0,0 +1,148 @@
+#! /usr/bin/env python
+# -*- coding: iso-8859-15 -*-
+
+import sys, sre, os, commands
+from socket import gethostname
+hostname = gethostname().split(".")[0]
+path = os.path.dirname(sys.argv[0])
+
+### Fichiers à modifier, chaine indiquant un commentaire dans ceux-ci
+### et commandes à excécuter après édition
+if hostname in ('zamok', 'rouge') :
+    fichiers = { '/etc/bind/named.conf.options' : '//' ,
+                 '/etc/postfix/main.cf' : '#' }
+    cmds = [ '/etc/init.d/postfix restart' , '/etc/init.d/bind9 reload' ]
+elif hostname == 'sila' :
+    fichiers = { '/etc/bind/named.conf.options' : '//' ,
+		 '/etc/squid/squid.conf' : '#' }
+    cmds = [ '/etc/init.d/squid reload' , '/etc/init.d/bind9 reload' ]
+elif hostname == 'bleu' :
+    fichiers = { '/etc/postfix/main.cf' : '#' }
+    cmds = [ '/etc/init.d/postfix restart' ]
+else :
+    print "Script sans effet sur cette machine."
+    sys.exit(1)
+
+# Machines à pinguer pour tester la connexion
+hosts = ( '216.239.57.104', '213.228.0.42', '217.12.3.11')
+
+####################
+# Fonctions utiles #
+####################
+
+def normal_ok() :
+    """ Teste le connexion normale à l'aide de pings
+    Retourne False si il y a un problème sur la connexion normale
+    Retourne True si la connexion normale est fonctionelle
+    """
+    pings = commands.getoutput('/usr/sbin/fping %s' % ' '.join(hosts))
+    print pings
+    if pings.count('is unreachable') == len(hosts) :
+        return False
+    else :
+        return True
+        
+def edit(file,comment,etat) :
+    """ Edite le fichier fourni en commentant (mode normal) 
+    ou décommentant (mode secours) les lignes signalées :
+    * celles se terminant avec #POUR SECOURS 
+    * ou les n lignes (<10)suivant une qui commande par #POUR SECOURS-n
+    (si le -n est omis une seule ligne est affectée)"""
+    
+    signal = '#POUR SECOURS'
+    l = len(signal)
+    
+    fd = open(file)
+    line = fd.readline()
+    new = ''
+    reste = 0                     # Nombre de lignes restant à traiter
+    while line :
+	l = line.rstrip()
+        mo = sre.match('^(.*)'+signal+'(|-.)$',l)
+        if (mo and len(mo.group(1)) > 1) or reste:
+	    # Ligne pour secours
+	    if not sre.match('^' + comment,l) and etat == 'normal':
+		# On est actuellement configuré en secours
+		# Il faut passer en normal
+		new += comment + line
+	    elif sre.match('^' + comment,l) and etat == 'secours' :
+		# On est actuellement configuré en normal
+		# Il faut passer en secours
+		new += line.replace(comment,'',1)
+	    else :
+		# Rien à faire, on est bien configuré
+		new += line
+            if reste :
+                reste -= 1
+        elif mo and len(mo.group(1)) == 0:
+            # On a une ligne avec secours uniquement, c'est les n 
+	    # prochaines lignes qui font foi
+	    try : reste = int(mo.group(2)[1:])
+	    except : reste = 1
+            new += line
+	else :
+	    # Ligne normale
+	    new += line
+
+	line = fd.readline()
+
+    fd.close()
+    
+    # Ecriture de la nouvelle version
+    open(file,'w').write(new)
+
+if __name__ == '__main__' :
+    # Etat dans lequel on devrai être
+    etat_maitre, mode = map(str.strip,open(path+'/etat_maitre').readlines())
+    nouvel_etat = etat_maitre
+    try:
+	etat_actuel = open('%s/etat_%s' % (path,hostname)).readline().strip()
+    except :
+	etat_actuel = None
+	
+    if len(sys.argv) == 2 :
+        if sys.argv[1] == 'secours'  :
+            print "Connexion de secours forcée."
+	    nouvel_etat = 'secours'
+            mode = 'manuel'
+	    open(path+'/etat_maitre','w').write('%s\n%s'% (nouvel_etat,mode))
+
+        elif sys.argv[1] == 'normal' :
+            print "Connexion normale forcée."
+	    nouvel_etat = 'normal'
+            mode = 'manuel'
+	    open(path+'/etat_maitre','w').write('%s\n%s'% (nouvel_etat,mode))
+	    
+	elif sys.argv[1] in ( 'auto', 'test' ) :
+            if sys.argv[1] == 'auto' :
+		mode = 'auto'
+		
+	    if mode != 'auto' :
+		print 'Mode manuel, passer en mode auto pour tester'
+		sys.exit(1)
+		
+	    # On choisi le mode de connexion
+	    if normal_ok() : nouvel_etat = 'normal'
+	    else : nouvel_etat = 'secours'
+		    
+	    if nouvel_etat != etat_maitre or sys.argv[1] == 'auto' :
+		open(path + '/etat_maitre','w').write('%s\n%s'% (nouvel_etat,mode))
+		
+    print "Mode %s" % mode
+	
+    if nouvel_etat == etat_actuel :
+	print etat_actuel
+    else :
+        # Il faut changer
+	sys.stderr.write("Passage en mode %s\n" % nouvel_etat) # Ecriture sur stderr pour le cron
+        for f, c in fichiers.items() :
+            try:
+                print 'Edition de %s' % f
+                edit(f,c,nouvel_etat)
+            except:
+                import traceback
+                traceback.print_exc()
+	open('%s/etat_%s' % (path,hostname),'w').write(nouvel_etat)
+        for c in cmds :
+	    os.system(c)
+