import logging
import base64
from odoo.addons.whatsapp.tools.whatsapp_api import WhatsAppApi

_logger = logging.getLogger(__name__)

# Guarda una referencia al método original
original_get_whatsapp_document = WhatsAppApi._get_whatsapp_document

def custom_get_whatsapp_document(self, document_id):
    _logger.info("Ejecutando versión modificada de _get_whatsapp_document")

    if self.wa_account_id.whatsapp_web_url: 
        _logger.info("Ejecutando versión modificada de _get_whatsapp_document con whatsapp web")
        result = base64.b64decode(document_id)
    else:
        result = original_get_whatsapp_document(self, document_id)

    # Aquí puedes modificar 'result' si es necesario antes de devolverlo
    return result

# Sobrescribir el método en tiempo de ejecución
WhatsAppApi._get_whatsapp_document = custom_get_whatsapp_document