Update mcp_server.py

mcp_server.py

@@ -1,17 +1,17 @@
 """
 mcp_server.py – MCP tool server exposing:
-  • query_database   – SQLite sample
-  • query_salesforce – Salesforce via CData JDBC
-Add more connectors under connectors/ and register the same way.
+  • query_database   – built-in SQLite demo
+  • query_hubspot    – HubSpot CRM via CData Python connector
 """
 
 from mcp.server.fastmcp import FastMCP
-import sqlite3, json
-from connectors.salesforce_connector import sf_query   # ← new connector
+import sqlite3, json, os
+from cdata.hubspot import connect  # pip install cdata-hubspot
 
+# ─── 1. FastMCP instance ────────────────────────────────────────────────
 mcp = FastMCP("EnterpriseData")
 
-# ─── in-memory SQLite sample ─────────────────────────────────────────────
+# ─── 2. In-memory SQLite sample ─────────────────────────────────────────
 conn = sqlite3.connect(":memory:", check_same_thread=False)
 cur  = conn.cursor()
 
@@ -35,10 +35,10 @@ cur.executemany(
 )
 conn.commit()
 
-# ─── SQLite tool ─────────────────────────────────────────────────────────
+# ─── 3. SQLite tool ─────────────────────────────────────────────────────
 @mcp.tool()
 def query_database(sql: str) -> str:
-    """Run arbitrary SQL against the Customers SQLite table."""
+    """Run SQL against the in-memory Customers table and return JSON rows."""
     try:
         cur.execute(sql)
         cols = [d[0] for d in cur.description or []]
@@ -47,11 +47,30 @@ def query_database(sql: str) -> str:
     except Exception as exc:
         return json.dumps({"error": str(exc)})
 
-# ─── Salesforce tool (premium) ───────────────────────────────────────────
+# ─── 4. HubSpot connector setup ─────────────────────────────────────────
+# Uses CData’s HubSpot Python connector :contentReference[oaicite:4]{index=4}
+HUBSPOT_TOKEN = os.getenv("HUBSPOT_TOKEN")
+hs_conn = connect(PrivateAppToken=HUBSPOT_TOKEN)
+
+# ─── 5. HubSpot tool ────────────────────────────────────────────────────
 @mcp.tool()
-def query_salesforce(sql: str) -> str:
-    """Run SOQL-as-SQL against Salesforce objects (requires Pro plan)."""
-    return sf_query(sql)
+def query_hubspot(sql: str) -> str:
+    """
+    Execute SQL/ODBC-style queries on HubSpot objects (Contacts, Deals, etc.).
+    Example:
+      SELECT firstname, lastname, email
+      FROM contacts
+      WHERE lifecyclestage = 'lead';
+    """
+    try:
+        cur_hs = hs_conn.cursor()
+        cur_hs.execute(sql)
+        cols = [d[0] for d in cur_hs.description or []]
+        rows = [dict(zip(cols, r)) for r in cur_hs.fetchall()]
+        return json.dumps(rows)
+    except Exception as exc:
+        return json.dumps({"error": str(exc)})
 
+# ─── 6. Run the server ───────────────────────────────────────────────────
 if __name__ == "__main__":
     mcp.run(transport="stdio")