Create salesforce_integration.py

salesforce_integration.py

@@ -0,0 +1,39 @@
+from simple_salesforce import Salesforce
+
+# Salesforce Authentication
+def authenticate_salesforce():
+    return Salesforce(username='[email protected]', password='Vedavathi@04', security_token='jqe4His8AcuFJucZz5NBHfGU')
+
+# Fetch data from Salesforce custom objects (Pole, Sensor Data, Alerts)
+def fetch_salesforce_data(site_name):
+    sf = authenticate_salesforce()
+
+    # Salesforce SOQL Query to get pole data based on the site
+    query = f"""
+    SELECT Id, Name, Solar_kWh__c, Wind_kWh__c, Power_Required__c, Power_Status__c, Tilt_Angle__c,
+    Vibration__c, Camera_Status__c, Alert_Level__c, Health_Score__c, Last_Checked__c
+    FROM Pole__c
+    WHERE Site__c = '{site_name}'
+    """
+    response = sf.query(query)
+    data = response['records']
+    
+    # Return the data in a usable format
+    return [
+        {
+            'Pole ID': record['Name'],
+            'Site': site_name,
+            'Solar (kWh)': record['Solar_kWh__c'],
+            'Wind (kWh)': record['Wind_kWh__c'],
+            'Power Required (kWh)': record['Power_Required__c'],
+            'Power Status': record['Power_Status__c'],
+            'Tilt Angle (°)': record['Tilt_Angle__c'],
+            'Vibration (g)': record['Vibration__c'],
+            'Camera Status': record['Camera_Status__c'],
+            'Health Score': record['Health_Score__c'],
+            'Alert Level': record['Alert_Level__c'],
+            'Anomalies': "",  # Can be filled if necessary
+            'Last Checked': record['Last_Checked__c']
+        }
+        for record in data
+    ]