Update salesforce_integration.py

salesforce_integration.py

@@ -1,22 +1,15 @@
-
-
-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
+    SELECT Name, Solar_Generation__c, Wind_Generation__c, Tilt__c, Vibration__c, 
+           Camera_Status__c, Longitude__c, Latitude__c, Last_Check_Time__c
+    FROM Smart_Pole__c
     LIMIT 50
     """
-    results = sf.query_all(query)
-    df = pd.DataFrame(results['records']).drop(columns='attributes')
-    return df
+    response = requests.get(
+        f"{INSTANCE_URL}/services/data/v59.0/query/",
+        headers={"Authorization": f"Bearer {ACCESS_TOKEN}"},
+        params={"q": query}
+    )
+    data = response.json()
+    return pd.DataFrame([record for record in data["records"]])
+