Update app.py

app.py

@@ -387,51 +387,57 @@ def get_addons():
         return jsonify({"success": False, "error": "Unable to fetch add-ons. Please try again later."})
 
         
+
 @app.route("/cart/update_quantity", methods=["POST"])
 def update_quantity():
-    data = request.json  
-    item_name = data.get('item_name')
+    data = request.json
+    item_name = data.get("item_name")
+    email = data.get("email")
 
     try:
-        # Convert quantity to an integer
-        quantity = int(data.get('quantity'))
+        quantity = int(data.get("quantity"))
     except (ValueError, TypeError):
         return jsonify({"success": False, "error": "Invalid quantity provided."}), 400
 
-    if not item_name or quantity is None:
-        return jsonify({"success": False, "error": "Item name and quantity are required."}), 400
+    if not item_name or not email:
+        return jsonify({"success": False, "error": "Email and item name are required."}), 400
 
     try:
-        # Query cart item from Salesforce
-        cart_items = sf.query(
-            f"SELECT Id, Quantity__c, Price__c, Base_Price__c, Add_Ons_Price__c FROM Cart_Item__c "
-            f"WHERE Name = '{item_name}'"
-        )['records']
+        # Fetch the cart item from Salesforce
+        cart_items = sf.query(f"SELECT Id, Base_Price__c FROM Cart_Item__c WHERE Name = '{item_name}'")["records"]
 
         if not cart_items:
             return jsonify({"success": False, "error": "Cart item not found."}), 404
 
-        # Retrieve first matching record
-        cart_item_id = cart_items[0]['Id']
-        base_price = cart_items[0].get('Base_Price__c', 0)
-        addons_price = cart_items[0].get('Add_Ons_Price__c', 0)
-        print(base_price)
+        cart_item_id = cart_items[0]["Id"]
+        base_price = cart_items[0].get("Base_Price__c", 0)  # Default to 0 if None
 
         if base_price is None:
             base_price = 0
-        # Calculate new item price
-        new_item_price = (base_price * quantity) + addons_price
-        print(new_item_price)
 
-        # Update the record in Salesforce
+        new_item_price = base_price * quantity
+
+        # Update Cart Item in Salesforce
         sf.Cart_Item__c.update(cart_item_id, {
             "Quantity__c": quantity,
-            "Price__c": new_item_price,  # Update item price
+            "Price__c": new_item_price,
         })
 
-        # Recalculate subtotal for all items in cart
-        all_cart_items = sf.query("SELECT Price__c FROM Cart_Item__c")['records']
-        new_subtotal = sum(item['Price__c'] for item in all_cart_items)
+        # Fetch all cart items for this user to calculate the new subtotal
+        cart_items = sf.query(f"SELECT Price__c FROM Cart_Item__c WHERE Customer_Email__c = '{email}'")["records"]
+        new_subtotal = sum(item["Price__c"] for item in cart_items)
+
+        # Fetch the latest Order for the user
+        order_result = sf.query(f"""
+            SELECT Id FROM Order__c
+            WHERE Customer_Email__c = '{email}'
+            ORDER BY CreatedDate DESC
+            LIMIT 1
+        """)
+
+        if order_result.get("records"):
+            order_id = order_result["records"][0]["Id"]
+            sf.Order__c.update(order_id, {"Total_Amount__c": new_subtotal})
 
         return jsonify({"success": True, "new_item_price": new_item_price, "subtotal": new_subtotal})
 
@@ -439,7 +445,6 @@ def update_quantity():
         print(f"Error updating quantity: {str(e)}")
         return jsonify({"success": False, "error": str(e)}), 500
 
-
         
 @app.route("/checkout", methods=["POST"])
 def checkout():