scripts/gestion/gen_confs/generate.py

#! /usr/bin/env python
# -*- coding: iso-8859-15 -*-

# Copyright (C) Frédéric Pauget
# Licence : GPLv2

"""Ce script permet de lancer la reconfiguration des divers services 

Usage: %(prog)s options
Les options possibles sont :
\t%(options)s
Les options avec = doivent être suivies d'un argument. Plusieurs 
arguments peuvent être founis pour une même option, les séparer par &
"""

import sys, signal, os, getopt

sys.path.append('/usr/scripts/gestion')

from ldap_crans import crans_ldap, crans, hostname, invite
from lock import *
from affich_tools import anim, cprint, OK
from time import localtime, strftime, time, sleep
import config
from inspect import getargspec

# On vérifie que l'on est root
if os.getuid() != 0:
    sys.stderr.write("Il faut être root\n")
    sys.exit(1)
    
signal.signal(signal.SIGINT,signal.SIG_IGN) # Pas de Ctrl-C

db = crans_ldap()
make_lock('auto_generate','Big lock',nowait=1)

class base_reconfigure :
    __service_develop = {
        'macip' : [ 'rouge-macip', 'zamok-macip', 'sila-macip' , 'komaz-macip' ] ,
        'droits': [ 'rouge-droits', 'nectaris-droits' ] ,
        'blacklist_upload' : [ 'bl_squid_upload' , 'komaz-blacklist'] ,
        'blacklist_autodisc' : [ 'komaz-blacklist'] }
    
    def __init__(self,to_do=[]) :
        if not to_do :
            if debug : print 'Lecture services à redémarrer dans la base LDAP'
            auto = 1
            to_do = {}
            # Création de la liste de ce qu'il y a à faire 
            for serv in db.services_to_restart() :
                # Services spéciaux portant sur plusieurs machines
                to_add = self.__service_develop.get(serv.nom,[])
                if to_add :
                    for nom in to_add :
                        for t in serv.start :
                            db.services_to_restart(nom,serv.args,t)
                            if time() > t :
                                to_do[nom] = serv.args
                    db.services_to_restart('-' + serv.nom)
                else :
                     for t in serv.start :
                         if time() > t :
                             to_do[serv.nom] = serv.args
                             break
        else :
            auto = 0
            if debug : print 'Services à redémarrer imposés (non lecture de la base LDAP)'

        for serv,args in to_do.items() :
            # Au cas ou le service porte sur plusieurs machines
            service = serv.replace('%s-'%hostname,'')
            if hasattr(self,service) :
                # Le service est à reconfigurer sur cette machine
                db.services_to_restart('-%s' % serv)
                try:
                    m = getattr(self,service)
                    if len(getargspec(m)[0])>1 :
                        m(args)
                    else :
                        m()
                except:
                    if auto : db.services_to_restart(serv,args)
                    sys.stderr.write('%s :'%serv)
                    import traceback
                    traceback.print_exc()
        
        if debug :
            reste = db.services_to_restart()
            if reste : 
                print "Reste à faire :"
                for s in reste :
                    print '\t%s' % s
            else :
                print "Plus rien à faire"
            
    def _machines(self) :
        """ Retourne les machines de la base étant 'à jour' """
        return db.all_machines(graphic=True)

    def _do(self,service,machines=None) :
        # Reconfiguration des services
        service.debug = debug
        if machines :
            service.machines = machines
        service.reconfigure()

    def macip(self,ips) :
        import firewall
        cprint("Mise à jour correspondance MAC-IP",'gras')
        eval("firewall.firewall_%s()" % hostname).mac_ip_maj(ips)
        
class rouge(base_reconfigure) :
    def autostatus(self) :
        from autostatus import autostatus
        self._do(autostatus())

    def dns(self) : 
        from gen_confs.bind import dns
        self._do(dns(),self._machines())
        
    def dhcp(self) :
        from gen_confs.dhcpd import dhcp
        self._do(dhcp(),self._machines())

    def droits(self) :
        from gen_confs.droits import droits_ldap
        self._do(droits_ldap())

    def ML_ens(self,mails) :
        from adherents import ML_ens
        self._do(ML_ens(mails))
        
    def mail_modif(self,trucs) :
        """ trucs est une liste de recherches à effectuer dans la base
        l'affichage des résultats formera le corps du mail """
        from supervison import mail
        self._do(mail(trucs))
                
    def switch(self,chambres) :
        from gen_confs.switchs import switch
        self._do(switch(chambres))

    def surveillance_exemptions(self) :
        from gen_confs.surveillance import exemptions
        self._do(exemptions())

    def surveillance_machines(self) :
        from gen_confs.surveillance import machines
        self._do(machines(),self._machines())

class zamok(base_reconfigure) :
    # Tout est dans le parent
    pass

class vert(base_reconfigure) :
    def home(self,args) :
        from adherents import home
        self._do(home(args))

    def del_user(self,args) :
        from adherents import del_user
        self._do(del_user(args))

class komaz(base_reconfigure) :
    def __fw(self) :
        if not hasattr(self,'__real_fw') :
            from firewall import firewall_komaz
            self.__real_fw = firewall_komaz()
        return self.__real_fw
    
    def macip(self,ips) :
        cprint("Mise à jour correspondance MAC-IP",'gras')
        self.__fw().mac_ip_maj(ips)

    def ports(self,ips) :
        self.__fw().port_maj(ips)

    def blacklist(self) :
        self.__fw().blacklist()

class sila(base_reconfigure) :
    def bl_carte_etudiant(self) :
        from gen_confs.squid import squid_carte
        self._do(squid_carte())
        
    def blacklist_virus(self) :
        from gen_confs.squid import squid_virus
        self._do(squid_virus())

    def blacklist_warez(self) :
        from gen_confs.squid import squid_warez
        self._do(squid_warez())
           
    def bl_squid_upload(self) : 
        from gen_confs.squid import squid_upload
        self._do(squid_upload())

    def bl_chbre_invalide(self) :
        from gen_confs.squid import squid_chbre
        self._do(squid_chbre())

class nectaris(base_reconfigure) :
    __restart_wifi_update = False

    def macip(self) :
        # Surcharge car rien à faire ici
        pass

    def __wifi(self) :
        # On arrete wifi-update
        os.system("kill $(cat /var/run/wifi-update.pid)")
        sleep(1)
        self.__restart_wifi_update = True

    def conf_wifi_ng(self) :
        from gen_confs.wifi_ng import conf_wifi_ng
        self._do(conf_wifi_ng())
        
    def conf_wifi(self) :
        from gen_confs.wifi import conf_wifi, bornes_wifi
        self.__wifi()
        self._do(conf_wifi())
        # On va aussi reconfigurer toutes les bornes
        self._do(bornes_wifi())
            
    def bornes_wifi(self,bornes) :
        from gen_confs.wifi import bornes_wifi
        self.__wifi()
        self._do(bornes_wifi(bornes))

    def dhcp(self) :
        from gen_confs.dhcpd import dhcp
        self._do(dhcp(),self._machines())
            
    def droits(self) :
        from gen_confs.droits import droits_openbsd
        self._do(droits_openbsd())
        
    def __del__(self) :
        if self.__restart_wifi_update :
            os.system("cd / ;  systrace -d /etc/systrace/wifi-update -A /usr/local/bin/twistd --syslog -o -y /usr/scripts/gestion/wifi-update.py --pidfile=/var/run/wifi-update.pid")

class vert(base_reconfigure) :
    def home(self,args) :
        from adherents import home
        self._do(home(args))
    
    def del_user(self,args) :
        from adherents import del_user
        self._do(del_user(args))
        
    def mail_bienvenue(self,mails) :
        from adherents import mail_bienvenue
        self._do(mail_bienvenue(mails))

                
signal.signal(signal.SIGINT,signal.SIG_DFL) # Comportement normal de Ctrl-C
remove_lock('auto_generate')

if __name__ == '__main__' :
    classe = eval(hostname)
    
    args_autorises = ['quiet', 'remove=', 'add=', 'list' ,'help', 'reconnect']
    # Ajout aussi des arguments spécifiques à la machine
    for nom in dir(classe) :
        if nom[0]!='_' :
            if len(getargspec(getattr(classe,nom))[0])>1 :
                nom += '='
            args_autorises.append(nom)
            

    ##### Options fournies ?
    try:
        if len(sys.argv) > 1 :
            options, arg = getopt.getopt(sys.argv[1:], 'h', args_autorises)
        else :
            options, arg = ( [],'')
    except getopt.error, msg :
        sys.stderr.write('%s\n' % msg)
        sys.exit(255)

    debug = 1 # défaut
    to_do = {}

    for opt, val in options :
        if opt == '--quiet' :
            debug = 0
        
        elif opt == '--remove' :
            for serv in val.split('&') :
                print 'Supression de %s' % serv
                db.services_to_restart('--%s' % serv)
            sys.exit(0)
        
        elif opt == '--list' :
            print 'Services à redémarrer :'
            for s in db.services_to_restart() :
                print '\t%s' % s
            sys.exit(0)
        
        elif opt == '--reconnect' :
            # Personnes à reconnecter
            print 'Recheche personnes en fin de sanction.'
            hier = strftime('%d/%m/%Y %H:%M'.split()[0],localtime(time()-60*60*24))
            c = db.search('blacklist=*,%s*' % hier)
            services = []
            for a_reco in c['adherent'] + c['machine'] + c['club'] :
                for bl in a_reco.blacklist() :
                    fin, sanction = bl.split(',')[1:3]
                    if fin.split()[0] == hier and sanction not in services :
                        services.append(sanction)
            for s in services :
                print "Ajout de blacklist_%s pour reconfiguration" % s
                db.services_to_restart('blacklist_%s' % s.encode())
            sys.exit(0)
        
        elif opt=='--add' :
            # Ajout d'un item dans les services à redémarrer
            for serv in val.split('&') :
                if serv.find(',')!=-1 :
                    serv, arg = serv.split(',',1)
                    arg = arg.split(',')
                else :
                    arg = []
                
                print 'Ajout de %s (%s)' % ( serv, arg )
                db.services_to_restart(serv, arg ) 
            sys.exit(0)
                
        elif opt =='-h' or opt == '--help' :
            print __doc__ % { 'prog' : sys.argv[0].split('/')[-1] , 'options' : '\n\t'.join(args_autorises) }
            sys.exit(0)
        
        elif len(opt)>2 and opt[:2]=='--' :
            to_do[opt[2:]] = val.split('&')
        
        
    # On fait ce qu'il y a à faire
    classe(to_do)