Update main.py

main.py

@@ -1,23 +1,255 @@
+# file: app.py
+# pip install fastapi uvicorn requests supabase==2.5.1 brotli
+
 import os
+import time
+import json
 import requests
-from fastapi import FastAPI, Request
+from typing import Tuple, Optional, Dict, Any
 from datetime import datetime, timezone, timedelta
+from fastapi import FastAPI, Request
+from supabase import create_client, Client
 
 app = FastAPI()
 
-# Load Supabase environment variables
+# ==========[ ODOO / Cloudflare – ثوابت للاختبار (زي ما اتفقنا) ]==========
+BASE = "https://odoo-demo.binrushd.care"
+
+CF_ID     = "0491b36d7dcabce5b04f1a53f347bb4e.access"
+CF_SECRET = "22152cb41b62393e159daaff7dce433006c3744c5850e6adc15fa3544bb5eb09"
+
+LOGIN    = "binrushd.automation@gmail.com"
+PASSWORD = "BR2025"
+DB       = "Live_August_25"
+
+# Defaults للّيد في Odoo (عدّل الارقام حسب بيئتك)
+ODOO_DEFAULT_USER_ID  = 2
+ODOO_DEFAULT_STAGE_ID = 1
+# لو عندك مصدر "TikTok" في Odoo وحابب تربطه ID مباشر، حطّ الـ ID هنا
+ODOO_SOURCE_ID_TIKTOK = None  # مثال: 4
+
+# ==========[ Supabase – من env زي كودك الأصلي ]==========
 SUPABASE_URL = os.getenv("SUPABASE_URL")
 SUPABASE_API_KEY = os.getenv("SUPABASE_API_KEY")
 
+# جدول الليدز اللي انت بتكتب فيه من TikTok
+SUPABASE_LEADS_TABLE = "binrushdleads"
+
+# جدول حفظ التوكن والسيشن
+TOKENS_TABLE = "api_tokens"
+TOKEN_ROW_ID = "odoo_demo"
+
+# =========[ Supabase Client ]=========
+supabase: Client = create_client(SUPABASE_URL, SUPABASE_API_KEY)
+
+# =========[ Helpers (مش بنغيّر فيها) ]=========
+def _std_headers() -> Dict[str, str]:
+    return {
+        "User-Agent":      "PostmanRuntime/7.48.0",
+        "Accept":          "application/json",
+        "Accept-Encoding": "gzip, deflate",
+        "Connection":      "keep-alive",
+    }
+
+def _api_headers() -> Dict[str, str]:
+    return {
+        "CF-Access-Client-Id":     CF_ID,
+        "CF-Access-Client-Secret": CF_SECRET,
+    }
+
+def _save_tokens_to_supabase(access_token: str, session_id: Optional[str]) -> None:
+    payload = {
+        "id": TOKEN_ROW_ID,
+        "access_token": access_token,
+        "session_id": session_id or "",
+        "updated_at": int(time.time()),
+    }
+    try:
+        res = supabase.table(TOKENS_TABLE).upsert(payload, on_conflict="id").execute()
+        print("📝 upsert attempted:", getattr(res, "data", None))
+    except Exception as e:
+        print("❌ upsert failed:", e)
+
+    # اقرأه تاني فورًا للتأكد
+    try:
+        check = (
+            supabase.table(TOKENS_TABLE)
+            .select("access_token, session_id, updated_at")
+            .eq("id", TOKEN_ROW_ID)
+            .execute()
+        )
+        print("🔎 readback:", getattr(check, "data", None))
+    except Exception as e:
+        print("⚠️ readback failed:", e)
+
+def _load_tokens_from_supabase() -> Tuple[Optional[str], Optional[str]]:
+    try:
+        res = (
+            supabase.table(TOKENS_TABLE)
+            .select("access_token, session_id")
+            .eq("id", TOKEN_ROW_ID)
+            .execute()
+        )
+        data = None
+        if hasattr(res, "data") and res.data:
+            if isinstance(res.data, list) and len(res.data) > 0:
+                data = res.data[0]
+            elif isinstance(res.data, dict):
+                data = res.data
+        if not data:
+            print("⚠️ No token row found in Supabase.")
+            return None, None
+        return data.get("access_token"), (data.get("session_id") or None)
+    except Exception as e:
+        print("⚠️ Error loading tokens from Supabase:", e)
+        return None, None
+
+def _maybe_parse_brotli(resp: requests.Response) -> Optional[Any]:
+    enc = (resp.headers.get("Content-Encoding") or "").lower()
+    if "br" in enc:
+        try:
+            import brotli
+            text = brotli.decompress(resp.content).decode("utf-8", errors="replace")
+            return json.loads(text)
+        except Exception:
+            return None
+    try:
+        return resp.json()
+    except ValueError:
+        return None
+
+# =========[ Odoo Auth ]=========
+def get_token_and_session() -> Tuple[str, Optional[str]]:
+    s = requests.Session()
+    headers = {
+        **_std_headers(),
+        **_api_headers(),
+        "login":    LOGIN,
+        "password": PASSWORD,
+        "db":       DB,
+    }
+    r = s.get(f"{BASE}/api/auth/token", headers=headers, timeout=30)
+    r.raise_for_status()
+    data = r.json()
+
+    access_token = data.get("access_token")
+    session_id   = r.cookies.get("session_id")
+    if not access_token:
+        raise RuntimeError(f"No access_token in response: {data}")
+
+    _save_tokens_to_supabase(access_token, session_id)
+    return access_token, session_id
+
+def _build_auth_headers(access_token: str, session_id: Optional[str], extra_headers: Dict[str, str]) -> Dict[str, str]:
+    hdr = {**_std_headers(), **_api_headers(), **extra_headers}
+    hdr["access-token"] = access_token
+    if session_id:
+        hdr["Cookie"] = f"session_id={session_id}"
+    hdr.setdefault("Content-Type", "application/json")
+    return hdr
+
+def call_endpoint(
+    path: str,
+    extra_headers: Optional[Dict[str, str]] = None,
+    method: str = "GET",
+    retry_on_expired: bool = True,
+    json_body: Optional[Dict[str, Any]] = None,
+) -> Optional[requests.Response]:
+    """
+    يعمل الطلب باستخدام التوكن والسيشن من Supabase.
+    لو الرد HTML (login) أو JSON فيه access_token error، نجدد ونكرر مرة واحدة.
+    """
+    access_token, session_id = _load_tokens_from_supabase()
+    if not access_token:
+        access_token, session_id = get_token_and_session()
+
+    url = f"{BASE}{path}"
+
+    def _do(token, sid):
+        headers = _build_auth_headers(token, sid, extra_headers or {})
+        try:
+            return requests.request(method, url, headers=headers, json=json_body, timeout=30)
+        except requests.RequestException as e:
+            print(f"⚠️ Network error during request: {e}")
+            return None
+
+    r = _do(access_token, session_id)
+    if r is None:
+        return None
+
+    ct = (r.headers.get("Content-Type") or "").lower()
+    if ("text/html" in ct) or (r.text.strip().startswith("<!DOCTYPE html")) or ("csrf_token" in r.text):
+        print("⚠️ Detected HTML login page — renewing token/session...")
+        if retry_on_expired:
+            new_token, new_sid = get_token_and_session()
+            r = _do(new_token, new_sid)
+            return r
+
+    data = _maybe_parse_brotli(r)
+    if isinstance(data, dict):
+        msg = (data.get("message") or "").lower()
+        if data.get("response") == "access_token" or "token seems to have expired" in msg or "expired" in msg:
+            print("⚠️ Detected expired token in JSON — renewing...")
+            if retry_on_expired:
+                new_token, new_sid = get_token_and_session()
+                r = _do(new_token, new_sid)
+                return r
+
+    return r
+
+# =========[ بناء ليد Odoo من TikTok ]=========
+def build_odoo_lead_from_tiktok(
+    campaign_id: Optional[str],
+    campaign_name: Optional[str],
+    adset_id: Optional[str],
+    adset_name: Optional[str],
+    ad_id: Optional[str],
+    ad_name: Optional[str],
+    field_map: Dict[str, Any],
+    formatted_create_time: Optional[str],
+) -> Dict[str, Any]:
+    lead = {
+        "name": field_map.get("name") or field_map.get("full_name") or "TikTok Lead",
+        "type": "opportunity",
+        "kanban_state": "grey",
+        "stage_id": ODOO_DEFAULT_STAGE_ID,
+        "user_id": ODOO_DEFAULT_USER_ID,
+        # بيانات الإعلان/المصدر
+        "adgroup_name": adset_name,
+        "adset_name": adset_name,
+        "ad_name": ad_name,
+        "lead_timestamp": formatted_create_time,
+        # تواصل
+        "email_from": field_map.get("email"),
+        "phone": field_map.get("phone_number"),
+        "city_ar": field_map.get("city"),
+    }
+    if ODOO_SOURCE_ID_TIKTOK:
+        lead["source_id"] = ODOO_SOURCE_ID_TIKTOK
+    return lead
+
+def create_crm_lead_on_odoo(lead: Dict[str, Any]) -> None:
+    r = call_endpoint("/api/crm.lead", method="POST", json_body=lead)
+    if r is None:
+        print("❌ Odoo request failed (no response).")
+        return
+
+    print("ODDO Status:", r.status_code, "| Content-Type:", r.headers.get("Content-Type"),
+          "| Content-Encoding:", r.headers.get("Content-Encoding"))
+    data = _maybe_parse_brotli(r)
+    if isinstance(data, (dict, list)):
+        print("Odoo JSON:", json.dumps(data, ensure_ascii=False, indent=2))
+    else:
+        print("Odoo RAW:", (r.text or "")[:400])
+
+# =========[ Webhook – نفس منطقك + إضافة إنشاء ليد في Odoo ]=========
 @app.post("/webhook")
 async def receive_webhook(request: Request):
     payload = await request.json()
-    
     print("📥 Webhook received:")
 
     gmt_plus_3 = timezone(timedelta(hours=3))
 
-    # Top-level fields
     print(f"Request ID: {payload.get('request_id')}")
     print(f"Object: {payload.get('object')}")
     print(f"Time (Unix): {payload.get('time')}")
@@ -28,7 +260,6 @@ async def receive_webhook(request: Request):
         print("Time (Formatted): Invalid timestamp")
     print("----")
 
-    # Loop through entries
     for entry in payload.get('entry', []):
         lead_id = entry.get('id')
         campaign_id = entry.get('campaign_id')
@@ -54,17 +285,17 @@ async def receive_webhook(request: Request):
             formatted_create_time = None
             print("Create Time: Invalid")
 
-        # Extract lead fields
-        field_map = {}
+        # fields من TikTok
+        field_map: Dict[str, Any] = {}
         for change in entry.get('changes', []):
-            field = change.get('field').lower()
+            field = (change.get('field') or "").lower()
             value = change.get('value')
             field_map[field] = value
             print(f"{field}: {value}")
 
         print("----")
 
-        # Prepare data for Supabase insertion
+        # 1) Supabase insert (زي كودك)
         insert_data = {
             "platform": "tiktok",
             "campaign_id": campaign_id,
@@ -87,11 +318,10 @@ async def receive_webhook(request: Request):
         }
 
         supa_resp = requests.post(
-            f"{SUPABASE_URL}/rest/v1/binrushdleads",
+            f"{SUPABASE_URL}/rest/v1/{SUPABASE_LEADS_TABLE}",
             headers=headers,
             json=insert_data
         )
-
         print("💾 Supabase insert status:", supa_resp.status_code)
         if supa_resp.status_code == 201:
             print("✅ Lead inserted successfully.")
@@ -101,4 +331,11 @@ async def receive_webhook(request: Request):
             except Exception:
                 print("❌ Supabase Response:", supa_resp.text)
 
-    return {"status": "ok"}
+        # 2) بناء وإرسال الليد لـ Odoo
+        odoo_lead = build_odoo_lead_from_tiktok(
+            campaign_id, campaign_name, adset_id, adset_name, ad_id, ad_name,
+            field_map, formatted_create_time
+        )
+        create_crm_lead_on_odoo(odoo_lead)
+
+    return {"status": "ok"}