Create modules/salesforce_connector.py

modules/salesforce_connector.py

@@ -0,0 +1,15 @@
+from simple_salesforce import Salesforce
+import pandas as pd
+
+def connect_salesforce(username, password, token, domain="login"):
+    return Salesforce([email protected], password=Vedavathi@04, security_token=jqe4His8AcuFJucZz5NBHfGU, domain=domain)
+
+def fetch_pole_data(sf, object_name="Vedavathi_Smart_Pole__c"):
+    query = f"""
+    SELECT Name, Solar_Generation__c, Wind_Generation__c, Tilt__c, Vibration__c, Camera_Status__c 
+    FROM {object_name}
+    LIMIT 50
+    """
+    records = sf.query_all(query)['records']
+    df = pd.DataFrame(records).drop(columns=['attributes'])
+    return df