Update connectors/salesforce_connector.py

connectors/salesforce_connector.py

@@ -1,11 +1,22 @@
-from cdata.salesforce import connect  # pip install cdata-salesforce
-import os
-_conn = connect(User=os.getenv("SF_USER"),
-                Password=os.getenv("SF_PASS"),
-                SecurityToken=os.getenv("SF_TOKEN"))
+"""
+Salesforce connector using CData JDBC bridge.
+Set env vars:
+  SF_USER, SF_PASS, SF_TOKEN
+"""
 
-def sf_query(s: str):
+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(s)
+    cur.execute(sql)
     cols = [d[0] for d in cur.description]
-    return [dict(zip(cols, r)) for r in cur.fetchall()]
+    rows = [dict(zip(cols, r)) for r in cur.fetchall()]
+    return json.dumps(rows)