feat: Build complete POC for Butler model (client, matrix, daemon)

connector-superoffice/inject_demo_data.py

@@ -0,0 +1,108 @@
+import os
+import sqlite3
+import requests
+from dotenv import load_dotenv
+
+load_dotenv(override=True)
+
+# --- CONFIGURATION ---
+DB_FILE = "marketing_matrix.db"
+SO_CLIENT_ID = os.getenv("SO_CLIENT_ID") or os.getenv("SO_SOD")
+SO_CLIENT_SECRET = os.getenv("SO_CLIENT_SECRET")
+SO_REFRESH_TOKEN = os.getenv("SO_REFRESH_TOKEN")
+BASE_URL = "https://app-sod.superoffice.com/Cust55774/api/v1"
+
+# --- SUPEROFFICE UDF ProgIds (from your discovery) ---
+PROG_ID_CONTACT_CHALLENGE = "SuperOffice:6"
+PROG_ID_PERSON_SUBJECT = "SuperOffice:5"
+PROG_ID_PERSON_INTRO = "SuperOffice:6"
+PROG_ID_PERSON_PROOF = "SuperOffice:7"
+
+# --- TEST DATA ---
+TEST_PERSON_ID = 1
+TEST_CONTACT_ID = 2
+TEST_VERTICAL_ID = 24 # Healthcare - Hospital
+TEST_ROLE_ID = 19     # Operativer Entscheider
+
+def get_token():
+    url = "https://sod.superoffice.com/login/common/oauth/tokens"
+    data = {"grant_type": "refresh_token", "client_id": SO_CLIENT_ID, "client_secret": SO_CLIENT_SECRET, "refresh_token": SO_REFRESH_TOKEN, "redirect_uri": "http://localhost"}
+    try:
+        resp = requests.post(url, data=data)
+        if resp.status_code == 200:
+            return resp.json().get("access_token")
+        else:
+            print(f"Token Error: {resp.text}")
+            return None
+    except Exception as e:
+        print(f"Connection Error: {e}")
+        return None
+
+def get_text_from_matrix(vertical_id, role_id):
+    conn = sqlite3.connect(DB_FILE)
+    c = conn.cursor()
+    c.execute("SELECT subject, intro, social_proof FROM text_blocks WHERE vertical_id = ? AND role_id = ?", (vertical_id, role_id))
+    row = c.fetchone()
+    conn.close()
+    return row if row else (None, None, None)
+
+def update_udfs(entity, entity_id, payload, token):
+    url = f"{BASE_URL}/{entity}/{entity_id}"
+    headers = {"Authorization": f"Bearer {token}", "Content-Type": "application/json", "Accept": "application/json"}
+    
+    # SuperOffice expects the full JSON body, not just the UDF part for PUT
+    # First, GET the existing entity
+    get_resp = requests.get(url, headers=headers)
+    if get_resp.status_code != 200:
+        print(f"❌ ERROR fetching {entity} {entity_id}: {get_resp.text}")
+        return False
+
+    existing_data = get_resp.json()
+    
+    # Merge UDFs
+    if "UserDefinedFields" not in existing_data:
+        existing_data["UserDefinedFields"] = {}
+    existing_data["UserDefinedFields"].update(payload)
+    
+    print(f"Updating {entity} {entity_id} with new UDFs...")
+    put_resp = requests.put(url, headers=headers, json=existing_data)
+    
+    if put_resp.status_code == 200:
+        print(f"✅ SUCCESS: Updated {entity} {entity_id}")
+        return True
+    else:
+        print(f"❌ ERROR updating {entity} {entity_id}: {put_resp.status_code} - {put_resp.text}")
+        return False
+
+if __name__ == "__main__":
+    print("--- Starting SuperOffice Injection Test ---")
+    
+    # 1. Get Text from local DB
+    subject, intro, proof = get_text_from_matrix(TEST_VERTICAL_ID, TEST_ROLE_ID)
+    if not subject:
+        print("❌ ERROR: Could not find matching text in local DB. Aborting.")
+        exit()
+        
+    print(f"Found texts for V_ID:{TEST_VERTICAL_ID}, R_ID:{TEST_ROLE_ID}")
+
+    # 2. Get API Token
+    access_token = get_token()
+    if not access_token:
+        print("❌ ERROR: Could not get SuperOffice token. Aborting.")
+        exit()
+        
+    # 3. Prepare Payloads
+    contact_payload = {
+        PROG_ID_CONTACT_CHALLENGE: intro # Using intro for challenge in this demo
+    }
+    person_payload = {
+        PROG_ID_PERSON_SUBJECT: subject,
+        PROG_ID_PERSON_INTRO: intro,
+        PROG_ID_PERSON_PROOF: proof
+    }
+
+    # 4. Inject data
+    update_udfs("Contact", TEST_CONTACT_ID, contact_payload, access_token)
+    update_udfs("Person", TEST_PERSON_ID, person_payload, access_token)
+    
+    print("\n--- Test complete ---")