llm_ticket3/test_extraction.py
2025-04-16 09:59:27 +02:00

34 lines
1.1 KiB
Python

#!/usr/bin/env python3
# -*- coding: utf-8 -*-
from odoo.attachment_manager import AttachmentManager
from odoo.auth_manager import AuthManager
import json
import os
def main():
# Initialiser le gestionnaire d'authentification avec des valeurs factices
# puisque nous n'allons pas vraiment nous connecter à Odoo pour ce test
auth = AuthManager(url='https://odoo.cbao.fr', db='dummy', username='dummy', api_key='dummy')
# Chemin vers le dossier du ticket
ticket_folder = 'output/ticket_T11143/T11143_20250416_094512'
# Charger les données des messages
with open(os.path.join(ticket_folder, 'all_messages.json'), 'r') as f:
messages_data = json.load(f)
# Créer le gestionnaire de pièces jointes
am = AttachmentManager(auth)
# Extraire les images manquantes
extracted_images = am.extract_missing_images(messages_data, ticket_folder)
print(f"Nombre d'images extraites: {len(extracted_images)}")
for img in extracted_images:
print(f"Image extraite: {img.get('name', 'Sans nom')} - {img.get('url', '')}")
print("Extraction terminée")
if __name__ == "__main__":
main()