Update app.py

app.py

@@ -1,29 +1,26 @@
 from flask import Flask, render_template, redirect, request, url_for
 from simple_salesforce import Salesforce
-from dotenv import load_dotenv
-import os
-
-# Load environment variables from .env file
-load_dotenv()
 
 app = Flask(__name__)
 
-# Get Salesforce credentials from environment variables
-SF_USERNAME = os.getenv('SF_USERNAME')
-SF_PASSWORD = os.getenv('SF_PASSWORD')
-SF_SECURITY_TOKEN = os.getenv('SF_SECURITY_TOKEN')
-SF_DOMAIN = os.getenv('SF_DOMAIN')
-
-# Salesforce connection
-try:
-    sf = Salesforce(username=SF_USERNAME, 
-                    password=SF_PASSWORD, 
-                    security_token=SF_SECURITY_TOKEN, 
-                    domain=SF_DOMAIN)
-    print("Salesforce connection successful!")
-except Exception as e:
-    print(f"Salesforce connection failed: {e}")
-    sf = None
+# Connect to Salesforce (replace with your Salesforce credentials)
+sf = Salesforce(username='your_username', password='your_password', security_token='your_token')
+
+# Function to update reward points in Salesforce
+def update_rewards_points(email, points):
+    # Query Salesforce to get the user by email
+    user = sf.query(f"SELECT Id, Reward_Points__c FROM Customer_Login__c WHERE Email__c = '{email}'")
+    
+    if user['totalSize'] == 1:
+        # Get the user ID and current reward points
+        user_id = user['records'][0]['Id']
+        current_points = user['records'][0]['Reward_Points__c']
+        
+        # Calculate new reward points
+        new_points = current_points + int(points)
+        
+        # Update reward points in Salesforce
+        sf.Customer_Login__c.update(user_id, {'Reward_Points__c': new_points})
 
 # Route for login page
 @app.route('/')
@@ -31,87 +28,43 @@ def login():
     return render_template('login.html')
 
 # Route to process login
-@app.route('/auth', methods=['POST'])
-def auth():
+@app.route('/login', methods=['POST'])
+def process_login():
     email = request.form['email']
     password = request.form['password']
-
-    if not sf:
-        return "Salesforce connection failed. Please check credentials and try again."
-
-    try:
-        # Query Salesforce for user authentication
-        query = f"SELECT Id, Reward_Points__c FROM Customer_Login__c WHERE Email__c = '{email}' AND Password__c = '{password}'"
-        result = sf.query(query)
-
-        if result['totalSize'] == 0:
-            return "Invalid Login Details"
-
-        customer = result['records'][0]
-        reward_points = customer['Reward_Points__c']
-
-        # Redirect to rewards page
-        return redirect(url_for('rewards', customer_id=customer['Id'], points=reward_points))
-    except Exception as e:
-        return f"Error during authentication: {e}"
+    
+    # Query Salesforce to check if the user exists
+    user = sf.query(f"SELECT Id, Email__c, Reward_Points__c FROM Customer_Login__c WHERE Email__c = '{email}'")
+    
+    if user['totalSize'] == 1:
+        # Redirect to rewards page after successful login
+        return redirect(url_for('rewards', email=email))
+    else:
+        # Return to login page with error message if user not found
+        return render_template('login.html', error="Invalid credentials")
 
 # Route to display rewards page
-@app.route('/rewards/<customer_id>')
-def rewards(customer_id):
-    try:
-        customer = sf.Customer_Login__c.get(customer_id)
-        points = customer['Reward_Points__c']
-
-        # Render the rewards page
-        return render_template('rewards.html', points=points, customer_id=customer_id)
-    except Exception as e:
-        return f"Error fetching rewards: {e}"
+@app.route('/rewards/<email>')
+def rewards(email):
+    user = sf.query(f"SELECT Id, Reward_Points__c FROM Customer_Login__c WHERE Email__c = '{email}'")
+    
+    if user['totalSize'] == 1:
+        reward_points = user['records'][0]['Reward_Points__c']
+        return render_template('rewards_above.html', reward_points=reward_points, email=email)
+    else:
+        # If the user is not found, redirect to login page
+        return redirect(url_for('login'))
 
 # Route to apply rewards
 @app.route('/apply_rewards', methods=['POST'])
 def apply_rewards():
-    customer_id = request.form['customer_id']
-    bill_amount = float(request.form['bill_amount'])
-    apply_rewards = request.form.get('apply_rewards')
-
-    try:
-        customer = sf.Customer_Login__c.get(customer_id)
-        points = customer['Reward_Points__c']
-        gst = 0.18 * bill_amount
-
-        if points >= 500 and apply_rewards:
-            discount = 0.1 * bill_amount
-            final_bill = bill_amount - discount + gst
-            updated_points = points - 500
-
-            # Update the customer's reward points in Salesforce
-            sf.Customer_Login__c.update(customer_id, {'Reward_Points__c': updated_points})
-
-            message = "You saved 10% on your total bill!"
-        else:
-            # Customers with below 500 points earn 10% of their bill amount as reward points
-            discount = 0
-            earned_points = 0.1 * bill_amount
-            final_bill = bill_amount + gst
-            updated_points = points + earned_points
-
-            # Update the customer's reward points in Salesforce
-            sf.Customer_Login__c.update(customer_id, {'Reward_Points__c': updated_points})
-
-            message = f"You earned 10% of your bill amount ({earned_points} points) as reward points!"
-
-        # Render the summary page
-        return render_template(
-            'apply_rewards.html',
-            original_bill=bill_amount,
-            discount=discount,
-            gst=gst,
-            final_bill=final_bill,
-            updated_points=updated_points,
-            message=message
-        )
-    except Exception as e:
-        return f"Error applying rewards: {e}"
-
-if __name__ == '__main__':
+    email = request.form['email']
+    reward_points = request.form['reward_points']
+    
+    # Call function to update reward points in Salesforce
+    update_rewards_points(email, reward_points)
+    
+    return render_template('rewards_below.html', message="Rewards applied successfully")
+
+if __name__ == "__main__":
     app.run(debug=True)