Update app.py

app.py

@@ -66,60 +66,56 @@ def generate_coupon_code(length=10):
 
 
 import re
-@app.route("/edit_profile", methods=["GET", "POST"])
 def edit_profile():
-    user_email = session.get('user_email')
-
-    if not user_email:
-        return redirect(url_for('login'))  # If user is not logged in, redirect to login page
+    email = session.get('user_email')  # Get logged-in user's email
+    if not email:
+        return redirect(url_for("login"))
 
     try:
-        # Fetch current user details from Salesforce
-        user_query = f"SELECT Name, Email__c, Phone_Number__c FROM Customer_Login__c WHERE Email__c = '{user_email}'"
-        user_result = sf.query(user_query)
+        # Fetch user details from Salesforce
+        result = sf.query(f"""
+            SELECT Id, Name, Email__c, Phone_Number__c, Password__c
+            FROM Customer_Login__c
+            WHERE Email__c = '{email}'
+        """)
 
-        if not user_result['records']:
-            return redirect(url_for('login'))  # If no user found, redirect to login
+        if not result['records']:
+            return redirect(url_for("login"))
 
-        current_user = user_result['records'][0]
-        user_name = current_user.get('Name')
-        user_phone = current_user.get('Phone_Number__c')
-        user_email = current_user.get('Email__c')
+        user = result['records'][0]
+        user_id = user.get("Id")
+        user_name = user.get("Name")
+        user_phone = user.get("Phone_Number__c")
+        user_email = user.get("Email__c")
 
     except Exception as e:
         print(f"Error fetching user data: {str(e)}")
         return jsonify({"success": False, "message": "Error fetching user data"})
 
-    # Handle form submission if POST request
-    if request.method == 'POST':
+    try:
+        # Process user profile update
         new_name = request.form.get('name')
         new_email = request.form.get('email')
         new_phone = request.form.get('phone')
-        new_password = request.form.get('password')  # Assuming password change is optional
+        new_password = request.form.get('password')
 
-        # Update the customer details in Salesforce
-        try:
-            # Create the update dictionary
-            update_data = {
-                'Name': new_name,
-                'Email__c': new_email,
-                'Phone_Number__c': new_phone
-            }
+        update_data = {
+            'Name': new_name,
+            'Email__c': new_email,
+            'Phone_Number__c': new_phone
+        }
 
-            # Only update password if provided
-            if new_password:
-                update_data['Password__c'] = new_password
+        if new_password:
+            update_data['Password__c'] = new_password
 
-            # Update user details in Salesforce
-            sf.Customer_Login__c.update(current_user['Id'], update_data)
-            flash("Profile updated successfully!", "success")
-            return redirect(url_for('customer_details'))  # Redirect back to profile page
+        # Update Salesforce record
+        sf.Customer_Login__c.update(user_id, update_data)
 
-        except Exception as e:
-            flash(f"Error updating profile: {str(e)}", "danger")
-            return render_template('edit_profile.html', user_name=user_name, user_phone=user_phone, user_email=user_email)
+        return render_template("edit_profile.html", user_name=new_name, user_phone=new_phone, user_email=new_email, success=True)
 
-    return render_template('edit_profile.html', user_name=user_name, user_phone=user_phone, user_email=user_email)
+    except Exception as e:
+        print(f"Error updating profile: {str(e)}")
+        return render_template("edit_profile.html", user_name=user_name, user_phone=user_phone, user_email=user_email, error=str(e))
 
 import re
 @app.route("/customer_details", methods=["GET"])