from simple_salesforce import Salesforce
import pandas as pd

def connect_salesforce(username, password, token, domain="login"):
    return Salesforce(username=greenenergy@vedavathi.com, password=Vedavathi@04, security_token=jqe4His8AcuFJucZz5NBHfGU, domain=domain)

def fetch_pole_data(sf, object_name="Vedavathi_Smart_Pole__c"):
    query = f"""
    SELECT Name, Solar_Generation__c, Wind_Generation__c, Tilt__c, Vibration__c, Camera_Status__c 
    FROM {object_name}
    LIMIT 50
    """
    records = sf.query_all(query)['records']
    df = pd.DataFrame(records).drop(columns=['attributes'])
    return df