Update app.py

app.py

@@ -268,35 +268,47 @@ def get_addons():
 def update_quantity():
     data = request.json  # Extract JSON data from the request
     email = data.get('email')  # Customer email
-    item_name = data.get('item_name')  # Item name (Cart Item Name in Salesforce)
+    item_name = data.get('item_name')  # Item name
     quantity = data.get('quantity')  # New quantity
 
     # Validate inputs
-    if not email or not item_name:
-        return jsonify({"success": False, "error": "Email and item name are required."}), 400
+    if not email or not item_name or quantity is None:
+        return jsonify({"success": False, "error": "Email, item name, and quantity are required."}), 400
 
     try:
-        # Query the cart item using the correct fields
+        # Query the cart item in Salesforce
         cart_items = sf.query(
-            f"SELECT Id, Quantity__c FROM Cart_Item__c WHERE Customer_Email__c = '{email}' AND Name = '{item_name}'"
+            f"SELECT Id, Quantity__c, Price__c, Base_Price__c, Add_Ons__c FROM Cart_Item__c "
+            f"WHERE Customer_Email__c = '{email}' AND Name = '{item_name}'"
         )['records']
 
-        if cart_items:
-            cart_item_id = cart_items[0]['Id']
-            base_price = cart_items[0]['Base_Price__c']
-            addons_price = sum([float(price) for price in cart_items[0].get('Add_Ons__c', "0").split("$")[1::2]])  # Example parsing logic
-
-            new_price = (base_price + addons_price) * quantity
-
-            # Update Salesforce record
-            sf.Cart_Item__c.update(cart_item_id, {
-                "Quantity__c": quantity,
-                "Price__c": new_price
-            })
-
-            return jsonify({"success": True, "new_quantity": quantity, "new_price": new_price})
+        if not cart_items:
+            return jsonify({"success": False, "error": "Cart item not found."}), 404
+
+        # Retrieve the first matching record
+        cart_item_id = cart_items[0]['Id']
+        base_price = cart_items[0]['Base_Price__c']
+        addons_string = cart_items[0].get('Add_Ons__c', "None")
+
+        # Calculate the add-ons price
+        addons_price = 0
+        if addons_string and addons_string != "None":
+            addons_price = sum(
+                [float(price) for price in addons_string.split("$")[1::2]]  # Example parsing logic
+            )
+
+        # Calculate the new price
+        new_price = (base_price + addons_price) * quantity
+
+        # Update the record in Salesforce
+        sf.Cart_Item__c.update(cart_item_id, {
+            "Quantity__c": quantity,
+            "Price__c": new_price
+        })
 
+        return jsonify({"success": True, "new_quantity": quantity, "new_price": new_price})
     except Exception as e:
+        print(f"Error updating quantity: {str(e)}")
         return jsonify({"success": False, "error": str(e)}), 500