Update utils/database_handler.py

utils/database_handler.py

@@ -1,65 +1,71 @@
-import pandas as pd
-import os
+import requests
 
-# Set a writable path for the Excel file within the app directory
-BASE_DIR = os.path.dirname(os.path.abspath(__file__))  # Current script directory
-CUSTOMERS_FILE = os.path.join(BASE_DIR, "..workspace/database/Book1.xlsx")
+# Salesforce credentials
+SALESFORCE_URL = "https://login.salesforce.com/services/oauth2/token"
+CLIENT_ID = "3MVG9PwZx9R6_UrcDlsRUsfM9CGTVKE82QnI5Vz02b.lv3H2yjDv_wy9.nrYihWYLUD1bv08N_qCG_gxiV5y8"
+CLIENT_SECRET = "94983416FB1DF4E754B56F69982F07D94D8971523E6893B8A62D75CD8E9E3A34"
+USERNAME = "surendra@sathkrutha.ccom"
+PASSWORD = "Lavanyanaga@123"
+SECURITY_TOKEN = "GH9RG97LroDoLe6gAAOHaJBP"
 
-# Ensure the database directory exists
-if not os.path.exists(os.path.dirname(CUSTOMERS_FILE)):
-    os.makedirs(os.path.dirname(CUSTOMERS_FILE))  # Create database directory
-
-# Ensure the file exists with the required structure
-if not os.path.exists(CUSTOMERS_FILE):
-    print("Creating new Excel file as it doesn't exist.")
-    pd.DataFrame(columns=["Name", "Phone", "Email", "Password"]).to_excel(CUSTOMERS_FILE, index=False)
-
-def check_credentials(email, password):
+# Authenticate with Salesforce
+def get_salesforce_access_token():
     try:
-        df = pd.read_excel(CUSTOMERS_FILE)
+        response = requests.post(
+            SALESFORCE_URL,
+            data={
+                "grant_type": "password",
+                "client_id": CLIENT_ID,
+                "client_secret": CLIENT_SECRET,
+                "username": USERNAME,
+                "password": PASSWORD + SECURITY_TOKEN,
+            },
+        )
+        response.raise_for_status()
+        return response.json()["access_token"], response.json()["instance_url"]
     except Exception as e:
-        print("Error reading Excel file:", e)
-        return False
-
-    user = df[(df["Email"] == email) & (df["Password"] == password)]
-    return not user.empty
+        print("Error obtaining Salesforce access token:", e)
+        return None, None
 
+# Save user to Salesforce
 def save_user(name, phone, email, password):
-    print("Attempting to save user:", name, phone, email, password)
-
-    try:
-        # Read the existing data
-        df = pd.read_excel(CUSTOMERS_FILE)
-    except Exception as e:
-        print("Error reading Excel file:", e)
+    access_token, instance_url = get_salesforce_access_token()
+    if not access_token:
         return False
 
-    print("Existing data before appending:\n", df)
-
     # Check if email already exists
-    if email in df["Email"].values:
-        print("Error: Email already exists.")
-        return False
+    headers = {
+        "Authorization": f"Bearer {access_token}",
+        "Content-Type": "application/json",
+    }
+    query = f"SELECT Email__c FROM User_Login__c WHERE Email__c = '{email}'"
+    query_response = requests.get(f"{instance_url}/services/data/v53.0/query", headers=headers, params={"q": query})
+    if query_response.status_code == 200 and query_response.json()["records"]:
+        return False  # User already exists
 
-    # Add new user data
-    new_user = {
-        "Name": str(name).strip(),
-        "Phone": str(phone).strip(),
-        "Email": str(email).strip(),
-        "Password": str(password).strip()
+    # Save new user
+    data = {
+        "Name__c": name,
+        "Phone__c": phone,
+        "Email__c": email,
+        "Password__c": password,
     }
-    df = pd.concat([df, pd.DataFrame([new_user])], ignore_index=True)
+    response = requests.post(f"{instance_url}/services/data/v53.0/sobjects/User_Login__c", headers=headers, json=data)
+    return response.status_code == 201
 
-    # Save updated data back to the Excel file
-    try:
-        with pd.ExcelWriter(CUSTOMERS_FILE, engine="openpyxl", mode="w") as writer:
-            df.to_excel(writer, index=False)
-        print("User saved successfully. Updated data:\n", df)
+# Check user credentials
+def check_credentials(email, password):
+    access_token, instance_url = get_salesforce_access_token()
+    if not access_token:
+        return False
 
-        # Confirm save by re-reading the file
-        df_check = pd.read_excel(CUSTOMERS_FILE)
-        print("Data in file after saving:\n", df_check)
-        return True
-    except Exception as e:
-        print("Error while saving to Excel:", e)
-        return False
+    headers = {
+        "Authorization": f"Bearer {access_token}",
+        "Content-Type": "application/json",
+    }
+    query = f"SELECT Password__c FROM User_Login__c WHERE Email__c = '{email}'"
+    query_response = requests.get(f"{instance_url}/services/data/v53.0/query", headers=headers, params={"q": query})
+    if query_response.status_code == 200 and query_response.json()["records"]:
+        stored_password = query_response.json()["records"][0]["Password__c"]
+        return stored_password == password
+    return False