llm_ticket3/odoo/retrieve_ticket.py

102 lines
3.8 KiB
Python

#!/usr/bin/env python3
import os
import sys
import json
import logging
import argparse
from datetime import datetime
from .ticket_manager import TicketManager
from .auth_manager import get_auth_manager, get_output_dir
from core.utils import setup_logging, log_separator
def retrieve_ticket(ticket_code, output=None, config_file="config.json", verbose=False):
# Configurer la journalisation
log_level = logging.DEBUG if verbose else logging.INFO
setup_logging(log_level, "retrieve_ticket.log")
# Obtenir le gestionnaire d'authentification
auth_manager = get_auth_manager(config_file)
if not auth_manager:
logging.error("Impossible d'obtenir le gestionnaire d'authentification Odoo")
return None
# Définir le répertoire de sortie
output_base = output or get_output_dir(config_file)
output_dir = os.path.join(output_base, f"ticket_{ticket_code}")
# Créer le répertoire de sortie spécifique au ticket
timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
ticket_dir = os.path.join(output_dir, f"{ticket_code}_{timestamp}")
os.makedirs(ticket_dir, exist_ok=True)
logging.info(f"Extraction du ticket {ticket_code}")
log_separator()
try:
# Extraire les données du ticket
ticket_manager = TicketManager(auth_manager)
result = ticket_manager.extract_ticket_data(ticket_code, ticket_dir)
if not result:
logging.error(f"Échec de l'extraction du ticket {ticket_code}")
return None
# Afficher le résumé
log_separator()
logging.info(f"Extraction terminée avec succès")
logging.info(f"Ticket: {ticket_code}")
logging.info(f"Répertoire: {ticket_dir}")
logging.info(f"Messages traités: {result.get('messages_count', 0)}")
logging.info(f"Pièces jointes: {result.get('attachments_count', 0)}")
log_separator()
# Générer un rapport de fin
summary = {
"timestamp": timestamp,
"ticket_code": ticket_code,
"output_directory": ticket_dir,
"message_count": result.get("messages_count", 0),
"attachment_count": result.get("attachments_count", 0),
"files_created": [
os.path.basename(result.get("ticket_info", "")),
os.path.basename(result.get("ticket_summary", "")),
os.path.basename(result.get("messages_file", "")),
os.path.basename(result.get("ticket_data_file", ""))
]
}
summary_path = os.path.join(ticket_dir, "extraction_summary.json")
with open(summary_path, 'w', encoding='utf-8') as f:
json.dump(summary, f, indent=2, ensure_ascii=False)
print(f"\nExtraction du ticket {ticket_code} terminée avec succès.")
print(f"Les données ont été sauvegardées dans: {ticket_dir}")
return ticket_dir
except Exception as e:
logging.exception(f"Une erreur est survenue: {e}")
return None
def parse_arguments():
parser = argparse.ArgumentParser(description="Récupère un ticket Odoo par son code et extrait ses données.")
parser.add_argument("ticket_code", help="Code du ticket à extraire")
parser.add_argument("--output", "-o", help="Répertoire de sortie", default=None)
parser.add_argument("--config", "-c", help="Fichier de configuration", default="config.json")
parser.add_argument("--verbose", "-v", action="store_true", help="Mode verbeux")
return parser.parse_args()
def main():
args = parse_arguments()
result = retrieve_ticket(
ticket_code=args.ticket_code,
output=args.output,
config_file=args.config,
verbose=args.verbose
)
if not result:
sys.exit(1)
if __name__ == "__main__":
main()