Create modules/salesforce_connector.py python Copy Edit

modules/salesforce_connector.py python Copy Edit

@@ -0,0 +1,16 @@
+from simple_salesforce import Salesforce
+import pandas as pd
+
+def connect_salesforce(username, password, token, domain="login"):
+    sf = Salesforce(username=username, password=password, security_token=token, domain=domain)
+    return sf
+
+def fetch_pole_data(sf):
+    query = """
+    SELECT Name, Solar_Gen__c, Wind_Gen__c, Tilt__c, Vibration__c, Camera_Status__c 
+    FROM Smart_Pole__c
+    LIMIT 50
+    """
+    records = sf.query_all(query)['records']
+    df = pd.DataFrame(records).drop(columns=['attributes'])
+    return df