from flask import Flask, jsonify
import requests
import urllib.parse

app = Flask(__name__)

@app.route('/get-salesforce-data')
def get_salesforce_data():
    access_token = 'YOUR_ACCESS_TOKEN'
    instance_url = 'https://vedavathi3-dev-ed.my.salesforce.com'

    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 
        Pole_Monitoring__c
    LIMIT 50
    """

    encoded_query = urllib.parse.quote(query)
    url = f"{instance_url}/services/data/v59.0/query/?q={encoded_query}"

    headers = {
        'Authorization': f'Bearer {access_token}',
        'Content-Type': 'application/json'
    }

    response = requests.get(url, headers=headers)

    if response.status_code == 200:
        data = response.json()
        return jsonify(data['records'])
    else:
        return jsonify({'error': response.text}), response.status_code

if __name__ == '__main__':
    app.run(debug=True)