Update app.py (#12)

- Update app.py (1ca882791ca1d5ccea004dc685010c574c6f2529)


Co-authored-by: geetha <[email protected]>

app.py

@@ -162,80 +162,25 @@ def generate_custom_dish():
     except Exception as e:
         return jsonify({"success": False, "error": str(e)}), 500
 
-import re
-@app.route("/edit_profile", methods=["GET", "POST"])
-def edit_profile():
-    email = session.get('user_email')  # Get logged-in user's email
-    if not email:
-        return redirect(url_for("login"))
-
-    try:
-        # 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 result['records']:
-            return redirect(url_for("login"))
-
-        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"})
-
-    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')
-
-        update_data = {
-            'Name': new_name,
-            'Email__c': new_email,
-            'Phone_Number__c': new_phone
-        }
-
-        if new_password:
-            update_data['Password__c'] = new_password
-
-        # Update Salesforce record
-        sf.Customer_Login__c.update(user_id, update_data)
-
-        return redirect(url_for('customer_details'))
-
-    except Exception as 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"])
 def customer_details():
     email = session.get('user_email')  # Get logged-in user's email
-
     if not email:
-        return redirect(url_for("login"))  # If no email is found, redirect to login
+        return redirect(url_for("login"))
 
     try:
         # Fetch customer details from Salesforce based on the email
         customer_record = sf.query(f"""
-            SELECT Name, Email__c, Phone_Number__c, Referral__c, Reward_Points__c
+            SELECT Id, Name, Email__c, Phone_Number__c, Referral__c, Reward_Points__c
             FROM Customer_Login__c
             WHERE Email__c = '{email}'
             LIMIT 1
         """)
 
-        # If no customer record found, handle it
         if not customer_record.get("records"):
-            return jsonify({"success": False, "message": "Customer not found in Salesforce"})
+            flash("Customer not found", "danger")
+            return redirect(url_for("login"))
 
-        # Get the customer details
         customer = customer_record["records"][0]
 
         # Prepare the data to return to the frontend
@@ -247,16 +192,59 @@ def customer_details():
             "reward_points": customer.get("Reward_Points__c", 0)
         }
 
-        # Return the customer details as JSON response
         return render_template("customer_details.html", customer=customer_data)
 
     except Exception as e:
-        print(f"Error fetching customer details: {str(e)}")
-        return jsonify({"success": False, "message": f"Error fetching customer details: {str(e)}"})
+        flash(f"Error fetching customer details: {str(e)}", "danger")
+        return redirect(url_for("login"))
 
+@app.route("/update_profile", methods=["POST"])
+def update_profile():
+    email = session.get('user_email')  # Get logged-in user's email
+    if not email:
+        return jsonify({'status': 'error', 'message': 'User not logged in'})
 
-from datetime import datetime
-import pytz  # Library to handle timezone conversions
+    try:
+        # Fetch user details from Salesforce
+        result = sf.query(f"""
+            SELECT Id, Name, Email__c, Phone_Number__c, Referral__c, Reward_Points__c
+            FROM Customer_Login__c
+            WHERE Email__c = '{email}'
+        """)
+
+        if not result['records']:
+            return jsonify({'status': 'error', 'message': 'User not found'})
+
+        user = result['records'][0]
+        user_id = user.get("Id")
+
+        # Get updated profile data from the form
+        new_name = request.form.get('customerName')
+        new_email = request.form.get('email')
+        new_phone = request.form.get('phone')
+        new_referral_code = request.form.get('referralCode')
+        new_reward_points = request.form.get('rewardPoints')
+
+        # Prepare data for Salesforce update
+        update_data = {
+            'Name': new_name,
+            'Email__c': new_email,
+            'Phone_Number__c': new_phone,
+            'Referral__c': new_referral_code,
+            'Reward_Points__c': new_reward_points
+        }
+
+        # Update Salesforce record
+        sf.Customer_Login__c.update(user_id, update_data)
+
+        return jsonify({
+            'status': 'success',
+            'message': 'Profile updated successfully!',
+            'data': update_data
+        })
+
+    except Exception as e:
+        return jsonify({'status': 'error', 'message': str(e)})
 
 @app.route("/order-history", methods=["GET"])
 def order_history():