DynamicMenu2 / utils /database_handler.py
nagasurendra's picture
Update utils/database_handler.py
bcdb5d8 verified
import requests
import json
# Salesforce Credentials
CONSUMER_KEY = "3MVG9PwZx9R6_UrcDlsRUsfM9CGTVKE82QnI5Vz02b.lv3H2yjDv_wy9.nrYihWYLUD1bv08N_qCG_gxiV5y8"
CONSUMER_SECRET = "94983416FB1DF4E754B56F69982F07D94D8971523E6893B8A62D75CD8E9E3E34"
USERNAME = "surendra@sathkrutha.ccom"
PASSWORD = "Lavanyanaga@123"
SECURITY_TOKEN = "GH9RG97LroDoLe6gAAOHaJBP"
# Salesforce Authentication
AUTH_URL = "https://login.salesforce.com/services/oauth2/token"
BASE_URL = None
HEADERS = None
def authenticate_salesforce():
global BASE_URL, HEADERS
data = {
"grant_type": "password",
"client_id": CONSUMER_KEY,
"client_secret": CONSUMER_SECRET,
"username": USERNAME,
"password": PASSWORD + SECURITY_TOKEN,
}
response = requests.post(AUTH_URL, data=data)
if response.status_code == 200:
auth_data = response.json()
BASE_URL = auth_data["instance_url"]
HEADERS = {
"Authorization": f"Bearer {auth_data['access_token']}",
"Content-Type": "application/json",
}
else:
raise Exception(f"Failed to authenticate with Salesforce: {response.text}")
def check_credentials(email, password):
"""Check user credentials during login."""
try:
authenticate_salesforce()
query = f"SELECT Password__c FROM User_Login__c WHERE Email__c = '{email}'"
response = requests.get(f"{BASE_URL}/services/data/v56.0/query?q={query}", headers=HEADERS)
if response.status_code == 200:
records = response.json().get("records", [])
if records and records[0]["Password__c"] == password:
return True
return False
except Exception as e:
print(f"Error checking credentials: {e}")
return False
def save_user(name, phone, email, password):
"""Save user details to Salesforce."""
try:
authenticate_salesforce()
# Check if email already exists
query = f"SELECT Id FROM User_Login__c WHERE Email__c = '{email}'"
response = requests.get(f"{BASE_URL}/services/data/v56.0/query?q={query}", headers=HEADERS)
if response.status_code == 200:
if response.json().get("totalSize", 0) > 0:
return False # Email already exists
# Create new user
user_data = {
"Name__c": name,
"Phone__c": phone,
"Email__c": email,
"Password__c": password,
}
response = requests.post(f"{BASE_URL}/services/data/v56.0/sobjects/User_Login__c/", headers=HEADERS, json=user_data)
if response.status_code == 201:
return True
else:
print(f"Error saving user: {response.text}")
return False
except Exception as e:
print(f"Error saving user: {e}")
return False