Rename connectors/salesforce_connector.py to connectors/hubspot_connector.py

connectors/hubspot_connector.py

@@ -0,0 +1,25 @@
+### FILE: connectors/hubspot_connector.py
+
+import os
+import json
+from hubspot import HubSpot
+from hubspot.crm.contacts import ApiException as HSContactsError
+
+# Initialize HubSpot client
+HUBSPOT_TOKEN = os.getenv("HUBSPOT_TOKEN")
+hs_client = HubSpot(access_token=HUBSPOT_TOKEN)
+
+def query_hubspot(object_type: str, limit: int = 100) -> str:
+    """
+    Fetch up to `limit` objects of type `contacts`, `companies`, or `deals`.
+    Example: object_type="contacts"
+    """
+    try:
+        api = getattr(hs_client.crm, object_type)
+        page = api.basic_api.get_page(limit=limit)
+        items = [r.to_dict() for r in page.results]
+        return json.dumps(items)
+    except HSContactsError as he:
+        return json.dumps({"error": he.body})
+    except Exception as e:
+        return json.dumps({"error": str(e)})

connectors/salesforce_connector.py

@@ -1,22 +0,0 @@
-"""
-Salesforce connector using CData JDBC bridge.
-Set env vars:
-  SF_USER, SF_PASS, SF_TOKEN
-"""
-
-import os, json
-from cdata.salesforce import connect
-
-_conn = connect(
-    User          = os.getenv("SF_USER"),
-    Password      = os.getenv("SF_PASS"),
-    SecurityToken = os.getenv("SF_TOKEN"),
-)
-
-def sf_query(sql: str) -> str:
-    """Execute SQL against Salesforce and return JSON rows."""
-    cur = _conn.cursor()
-    cur.execute(sql)
-    cols = [d[0] for d in cur.description]
-    rows = [dict(zip(cols, r)) for r in cur.fetchall()]
-    return json.dumps(rows)