Update modules/salesforce_connector.py

modules/salesforce_connector.py

@@ -1,27 +1,21 @@
-from simple_salesforce import Salesforce
+import os
 import pandas as pd
+from simple_salesforce import Salesforce
 
 def connect_salesforce():
     return Salesforce(
-        username='[email protected]',
-        password='Vedavathi@04',
-        security_token='jqe4His8AcuFJucZz5NBHfGU'
+        username=os.getenv("SF_USERNAME", "[email protected]"),
+        password=os.getenv("SF_PASSWORD", "Vedavathi@04"),
+        security_token=os.getenv("SF_TOKEN", "jqe4His8AcuFJucZz5NBHfGU")
     )
 
 def fetch_pole_data_from_salesforce(sf):
     query = """
     SELECT Name, Solar_Generation__c, Wind_Generation__c, Power_Required__c,
            Camera_Status__c, Tilt__c, Vibration__c, Alert_Level__c, Power_Sufficient__c
-    FROM VIEP_Pole__c  # ← replace this with your actual API name
-query = """
-     SELECT Name, Solar_Generation__c, Wind_Generation__c, Power_Required__c,
-       Camera_Status__c, Tilt__c, Vibration__c, Alert_Level__c, Power_Sufficient__c
-    FROM VIEP_Pole__c
-     LIMIT 50
-     """
-
+    FROM YourObjectAPI__c
     LIMIT 50
-    """
+    """  # Replace 'YourObjectAPI__c' with correct object name
     result = sf.query(query)
     records = result['records']
 
@@ -36,7 +30,7 @@ query = """
             "Camera Status": r.get('Camera_Status__c', "Offline"),
             "Tilt (°)": r.get('Tilt__c', 0),
             "Vibration (g)": r.get('Vibration__c', 0),
-            "Anomalies": "N/A",  # Optional: You can add your logic
+            "Anomalies": "N/A",
             "Alert Level": r.get('Alert_Level__c', "Green"),
             "Date": pd.Timestamp.today().date()
         })