Distribution EOLE » Outils Eole » arv

#!/usr/bin/python

# -*- coding: UTF-8 -*-

# Karim Ayari

# DSI - Rectorat de Lyon

# Met à jour les info serveurs Zéphir (id et version) dans arv

from arv.config import amon_module_list, sphynx_module_list

from arv.lib.logger import logger

from arv.lib.usezephir import Zephir

from arv.db.node import get_nodes

from arv.db.initialize import initialize_database, commit_database

from getpass import getpass

from commands import getstatusoutput

### Initialisation de la base arv

initialize_database(create=False)

###

### Connexion à Zéphir

user=raw_input("Nom d'utilisateur Zéphir : ")

password=getpass("Mot de passe Zéphir pour l'utilisateur "+user+": ")

zephir = Zephir(user=user, password=password)

###

for node in get_nodes():

    if zephir != None:

        uai=node.uai

        name=node.name

        id_zephir=node.id_zephir

        server = zephir.get_etab_server(uai, name, id_zephir)

        if server:

            if id_zephir is not None:

                print("Update uai and name from Zéphir")

                logger.debug("Update uai and name from Zéphir")

                uai = server['rne']

                libelle = server['libelle']

                if type(uai) == unicode:

                    node.uai = str(unicode.encode(uai, "utf-8")).decode('utf-8')

                else:

                    node.uai = uai

                if type(libelle) == unicode:

                    node.name = str(unicode.encode(libelle, 'utf-8')).decode('utf-8')

                else:

                    node.name = libelle

            else:

                 print("Fill id_zephir and eole_version")

                 logger.debug("Fill id_zephir and eole_version")

                 node.id_zephir = int(server['id'])

                 module_eole_version = zephir.dict_eole_version(amon_module_list + sphynx_module_list)

                 node.eole_version = module_eole_version[server['module_actuel']]

                 if type(node.eole_version) == unicode:

                     node.eole_version = str(unicode.encode(node.eole_version, 'utf-8')).decode('utf-8')

        else:

            print("Node does not exist in Zéphir")

            logger.debug("Node does not exist in Zéphir")

    else:

        print("No Zéphir connection")

        logger.debug("No Zéphir connection")

#enregistrement des modifications et redémarrage d'ARV

print "enregistrement et redémarrage d'ARV..."

commit_database()

ret,val=getstatusoutput("/etc/init.d/arv restart")

###