consultoria-as
36dd6634e3
- Refactor whatsapp_service.py to accept bridge_url parameter - whatsapp_bp.py: remove hardcoded tenant_id=11, use g.tenant_id - whatsapp_bp.py: webhook now accepts ?tenant_id param with fallback - config_bp.py: add GET/PUT /config/whatsapp endpoints - Each tenant can now have its own Baileys bridge URL and settings
129 lines
4.5 KiB
Python
129 lines
4.5 KiB
Python
"""WhatsApp service via Baileys Bridge (self-hosted, free).
|
|
|
|
Simple REST bridge that wraps WhatsApp Web via Baileys.
|
|
Supports per-tenant configuration via bridge_url parameter.
|
|
"""
|
|
|
|
import requests
|
|
from config import WHATSAPP_BRIDGE_URL
|
|
|
|
HEADERS = {'Content-Type': 'application/json'}
|
|
|
|
|
|
def _get_url(bridge_url=None):
|
|
return bridge_url or WHATSAPP_BRIDGE_URL
|
|
|
|
|
|
def get_status(bridge_url=None):
|
|
url = _get_url(bridge_url)
|
|
try:
|
|
return requests.get(f'{url}/status', timeout=5).json()
|
|
except Exception as e:
|
|
return {'state': 'error', 'error': str(e)}
|
|
|
|
|
|
def get_qr(bridge_url=None):
|
|
url = _get_url(bridge_url)
|
|
try:
|
|
return requests.get(f'{url}/qr', timeout=5).json()
|
|
except Exception as e:
|
|
return {'state': 'error', 'error': str(e)}
|
|
|
|
|
|
def connect(bridge_url=None):
|
|
url = _get_url(bridge_url)
|
|
try:
|
|
return requests.post(f'{url}/connect', headers=HEADERS, timeout=5).json()
|
|
except Exception as e:
|
|
return {'state': 'error', 'error': str(e)}
|
|
|
|
|
|
def send_message(phone, text, bridge_url=None):
|
|
url = _get_url(bridge_url)
|
|
try:
|
|
return requests.post(f'{url}/send', headers=HEADERS, json={'phone': phone, 'message': text}, timeout=15).json()
|
|
except Exception as e:
|
|
return {'error': str(e)}
|
|
|
|
|
|
def send_quote(phone, quote_data, bridge_url=None):
|
|
lines = [f"*Cotizacion #{quote_data.get('id', '')}*", ""]
|
|
for item in quote_data.get('items', []):
|
|
lines.append(f"- {item.get('quantity', 1)}x {item.get('name', '')} ${item.get('subtotal', 0):,.2f}")
|
|
lines.append(f"\nSubtotal: ${quote_data.get('subtotal', 0):,.2f}")
|
|
lines.append(f"IVA: ${quote_data.get('tax_total', 0):,.2f}")
|
|
lines.append(f"*Total: ${quote_data.get('total', 0):,.2f}*")
|
|
return send_message(phone, "\n".join(lines), bridge_url=bridge_url)
|
|
|
|
|
|
def logout(bridge_url=None):
|
|
url = _get_url(bridge_url)
|
|
try:
|
|
return requests.post(f'{url}/logout', headers=HEADERS, timeout=5).json()
|
|
except Exception as e:
|
|
return {'error': str(e)}
|
|
|
|
|
|
def process_incoming(webhook_data):
|
|
"""Extract a normalized dict from a Baileys webhook payload.
|
|
|
|
Supports text messages, image messages, audio (voice notes), and video.
|
|
Media content comes pre-downloaded as base64 from the bridge so Python
|
|
doesn't have to re-authenticate with WhatsApp servers.
|
|
|
|
Returns:
|
|
dict with keys:
|
|
phone — numeric phone (no JID suffix)
|
|
jid — full remote JID (may be @s.whatsapp.net or @lid)
|
|
text — text content (plain text or media caption)
|
|
from_me — bool, True if we sent the message
|
|
message_id — WhatsApp message ID
|
|
media_kind — 'text' | 'image' | 'audio' | 'video'
|
|
media_base64 — base64 string if media, else None
|
|
media_mimetype — e.g. 'image/jpeg', 'audio/ogg'
|
|
is_voice_note — True for WhatsApp voice notes (audioMessage ptt)
|
|
push_name — display name from WhatsApp
|
|
"""
|
|
data = webhook_data.get('data', {})
|
|
key = data.get('key', {})
|
|
message = data.get('message', {})
|
|
|
|
# remoteJid can be phone@s.whatsapp.net or LID@lid
|
|
remote_jid = key.get('remoteJid', '')
|
|
phone = remote_jid.replace('@s.whatsapp.net', '').replace('@lid', '')
|
|
|
|
# The bridge now classifies and passes these extra fields. Fall back to
|
|
# the old parsing if they're missing (older bridge version).
|
|
media_kind = data.get('media_kind', 'text')
|
|
media_base64 = data.get('media_base64')
|
|
media_mimetype = data.get('media_mimetype')
|
|
media_caption = data.get('media_caption') or ''
|
|
is_voice_note = bool(data.get('media_ptt'))
|
|
push_name = data.get('push_name') or ''
|
|
|
|
# Text content:
|
|
# - For 'text' messages → conversation or extendedTextMessage
|
|
# - For 'image'/'video' → the caption (may be empty)
|
|
# - For 'audio' → empty (filled in later by Whisper transcription)
|
|
if media_kind == 'text':
|
|
text = (
|
|
message.get('conversation', '')
|
|
or message.get('extendedTextMessage', {}).get('text', '')
|
|
or ''
|
|
)
|
|
else:
|
|
text = media_caption
|
|
|
|
return {
|
|
'phone': phone,
|
|
'jid': remote_jid,
|
|
'text': text,
|
|
'from_me': key.get('fromMe', False),
|
|
'message_id': key.get('id', ''),
|
|
'media_kind': media_kind,
|
|
'media_base64': media_base64,
|
|
'media_mimetype': media_mimetype,
|
|
'is_voice_note': is_voice_note,
|
|
'push_name': push_name,
|
|
}
|