Create salesforce_integration.py

salesforce_integration.py

@@ -0,0 +1,22 @@
+
+
+from simple_salesforce import Salesforce
+import pandas as pd
+
+def fetch_poles():
+    sf = Salesforce(
+        username="[email protected]",
+        password="Vedavathi@04",
+        security_token="jqe4His8AcuFJucZz5NBHfGU"
+    )
+
+    query = """
+    SELECT Name, Site__c, Solar_Generation__c, Wind_Generation__c,
+           Power_Required__c, Power_Sufficient__c, Camera_Status__c,
+           Alert_Level__c
+    FROM Pole__c
+    LIMIT 50
+    """
+    results = sf.query_all(query)
+    df = pd.DataFrame(results['records']).drop(columns='attributes')
+    return df