AI_SQL / connectors /salesforce_connector.py
mgbam's picture
Create connectors/salesforce_connector.py
f2f4930 verified
raw
history blame
388 Bytes
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"))
def sf_query(s: str):
cur = _conn.cursor()
cur.execute(s)
cols = [d[0] for d in cur.description]
return [dict(zip(cols, r)) for r in cur.fetchall()]