Distribution EOLE

#!/usr/bin/python3.10

# -*- coding: utf-8 -*-

import getpass # pour cacher le mdp

import subprocess

import argparse # pour les options

import sys

from xmlrpc.client import Server

from xmlrpc import client as xmlrpclib # ligne à rajouter

def getserveur(user_zephir,mdp_user,serveur_address):

    class SpecialTransport(xmlrpclib.SafeTransport):

        def send_content(self, connection, request_body):

            connection.putheader("Referer", f"https://serveur_address:7080")  # URL dynamique du zephir

            connection.putheader("Origin", f"https://serveur_address:7080")  # URL dynamique du zephir

            connection.putheader("X-Requested-With", "XMLHttpRequest")

            connection.putheader("Content-Type", "text/xml")

            connection.putheader("Content-Length", str(len(request_body)))

            connection.endheaders()

            if request_body:

                connection.send(request_body)

    z = Server(f"https://{user_zephir}:{mdp_user}@{serveur_address}:7080",  # URL dynamique pour la connexion

               transport=SpecialTransport(),  # Appel de la classe de surcharge transport

               verbose=False,  # ligne à rajouter

               allow_none=True)  # ligne à rajouter

    return z

def add(args):

    # Ajouter ici le code pour ajouter l'archive

    print(f"Association effectué avec ID Zephir : {args.id_zephir} et l'UAI : {args.uai}")

    uai_file = f"{args.uai}.tar.gz"

    z = getserveur(args.user_zephir,args.mdp_user,args.serveur_address)

    a = z.aaf.add_file(args.id_zephir,args.uai)

    print(a)

def associate(args):

    # Ajouter ici le code pour ajouter l'archive

    print(f"Ajout effectué avec ID Zephir : {args.id_zephir} et UAI : {args.uai}")

    print(f"Attention, aucune notification effecutée, faire un --notify lorsqu'attendu")

def delete(args):

    # Ajouter ici le code pour ajouter l'archive

    print(f"Suppression de l'archive aaf : {args.uai}.tar.gz")

    # Surcharge des headers pour le transport

    uai_file = f"{args.uai}.tar.gz"

    z = getserveur(args.user_zephir,args.mdp_user,args.serveur_address)

    d = z.aaf.del_file(uai_file)

    print(d)

def getlist(args):

    print(f"Listing des archives pour l'ID Zephir : {args.id_zephir}")

    # Surcharge des headers pour le transport

    z = getserveur(args.user_zephir,args.mdp_user,args.serveur_address)

    g = z.aaf.get_list(args.id_zephir)

    print(g)

def notify(args):

    print(f"Synchonisation avec la base zephir : {args.user_zephir}@{args.serveur_address}")

    # Surcharge des headers pour le transport

    z = getserveur(args.user_zephir,args.mdp_user,args.serveur_address)

    uai_file = f"{args.uai}.tar.gz"

    if args.annual == True:

        a = z.aaf.notify_upload_annual(uai_file)

    else:

        a = z.aaf.notify_upload(uai_file)

    print(a)

# Fonction pour gérer les options avec argparse

if __name__ == "__main__":

    parser = argparse.ArgumentParser(description="Outil de gestion archive AAF Zephir")

    # Option associate

    subparsers = parser.add_subparsers(help='sub-command')

    add_subparser = subparsers.add_parser("add", help="Ajout d'une archive aaf au format tar.gz dans la base zephir, sans prise en compte pour action")

    add_subparser.set_defaults(func=add)

    add_subparser.add_argument('--serveur_address', type=str, required=True, help="Adresse du serveur zephir, format fqdn")

    add_subparser.add_argument('--user_zephir', type=str, required=True, help="Nom de l'utilisateur Zephir à utiliser (droit nécessaire)")

    add_subparser.add_argument('--mdp_user', type=str, required=True, help="Mot de passe pour l'utilisateur Zephir")

    add_subparser.add_argument('--id_zephir', type=int, required=True)

    add_subparser.add_argument('--uai', type=str, required=True, help="Numéro UAI et nom de l'archive.(ancien RNE)")

    associate_subparser = subparsers.add_parser("associate", help="Ajout/copie d'une archive aaf au format tar.gz, avec prise en compte par zephir pour action")

    associate_subparser.set_defaults(func=associate)

    associate_subparser.add_argument('--serveur_address', type=str, required=True, help="Adresse du serveur zephir, format fqdn")

    associate_subparser.add_argument('--user_zephir', type=str, required=True, help="Nom de l'utilisateur Zephir à utiliser (droit nécessaire)")

    associate_subparser.add_argument('--mdp_user', type=str, required=True, help="Mot de passe pour l'utilisateur Zephir")

    associate_subparser.add_argument('--id_zephir', type=int, required=True, help="Numéro identifiant le serveur cible sur zephir (enregistmrent_zephir)")

    associate_subparser.add_argument('--uai', type=str, required=True, help="Numéro UAI et nom de l'archive.(ancien RNE)")

    delete_subparser = subparsers.add_parser("delete", help="Suppression d'une archive de la base zephir")

    delete_subparser.set_defaults(func=delete)

    delete_subparser.add_argument('--serveur_address', type=str, required=True, help="Adresse du serveur zephir, format fqdn")

    delete_subparser.add_argument('--user_zephir', type=str, required=True, help="Nom de l'utilisateur Zephir à utiliser (droit nécessaire)")

    delete_subparser.add_argument('--mdp_user', type=str, required=True, help="Mot de passe pour l'utilisateur Zephir")

    delete_subparser.add_argument('--uai', type=str, required=True, help="Numéro UAI et nom de l'archive.(ancien RNE)")

    getlist_subparser = subparsers.add_parser("getlist", help="Liste les archives associées à l'id du serveur indiqué")

    getlist_subparser.set_defaults(func=getlist)

    getlist_subparser.add_argument('--serveur_address', type=str, required=True, help="Adresse du serveur zephir, format fqdn")

    getlist_subparser.add_argument('--user_zephir', type=str, required=True, help="Nom de l'utilisateur Zephir à utiliser (droit nécessaire)")

    getlist_subparser.add_argument('--mdp_user', type=str, required=True, help="Mot de passe pour l'utilisateur Zephir")

    getlist_subparser.add_argument('--id_zephir', type=int, required=True, help="Numéro identifiant le serveur cible sur zephir (enregistmrent_zephir)")

    notify_subparser = subparsers.add_parser("notify", help="Ajout/copie d'une archive aaf au format tar.gz dans la base zephir, avec prise en compte pour action")

    notify_subparser.set_defaults(func=notify)

    notify_subparser.add_argument('--serveur_address', type=str, required=True, help="Adresse du serveur zephir, format fqdn")

    notify_subparser.add_argument('--user_zephir', type=str, required=True, help="Nom de l'utilisateur Zephir à utiliser (droit nécessaire)")

    notify_subparser.add_argument('--mdp_user', type=str, required=True, help="Mot de passe pour l'utilisateur Zephir")

    notify_subparser.add_argument('--uai', type=str, required=True, help="Numéro UAI et nom de l'archive.(ancien RNE)")

    notify_subparser.add_argument('--annual', action="store_true")

    # Analyser les arguments

    args = parser.parse_args()

    args.func(args)