salesforce.py

import os
from simple_salesforce import Salesforce

def get_salesforce_connection():
    # Load credentials from environment variables
    sf_username = os.getenv('SF_USERNAME')
    sf_password = os.getenv('SF_PASSWORD')
    sf_token = os.getenv('SF_SECURITY_TOKEN')
    
    # Ensure the credentials are available
    if not sf_username or not sf_password or not sf_token:
        raise ValueError("Salesforce credentials are missing from environment variables.")
    
    # Connect to Salesforce
    sf = Salesforce(username=sf_username, password=sf_password, security_token=sf_token)
    print("Connected to Salesforce successfully!")
    
    return sf