Update app.py

app.py

@@ -934,33 +934,67 @@ def get_addons():
     except Exception as e:
         print(f"Error fetching add-ons: {e}")
         return jsonify({"success": False, "error": "Unable to fetch add-ons. Please try again later."})
-@app.route('/cart/update_quantity', methods=['POST'])
+@app.route("/cart/update_quantity", methods=["POST"])
 def update_quantity():
     try:
-        data = request.get_json()  # Get JSON data from the frontend
-        item_name = data['item_name']
-        customer_email = data['email']
-        quantity = int(data['quantity'])  # Ensure quantity is cast to an integer
-
-        # Update the quantity in Salesforce or your database
-        cart_item = find_cart_item(item_name, customer_email)  # Custom function to find the cart item
+        data = request.json  # Extract JSON data from the request
+        email = data.get('email')
+        item_name = data.get('item_name')
+        
+        # Convert quantity to an integer and handle invalid data
+        try:
+            quantity = int(data.get('quantity'))
+        except (ValueError, TypeError):
+            return jsonify({"success": False, "error": "Invalid quantity provided."}), 400
         
-        if cart_item:
-            # Update the quantity in Salesforce or your database
-            cart_item['Quantity__c'] = quantity
-            cart_item['Price__c'] = cart_item['Base_Price__c'] * quantity  # Calculate new price based on updated quantity
+        # Validate inputs
+        if not email or not item_name or quantity is None:
+            return jsonify({"success": False, "error": "Email, item name, and quantity are required."}), 400
 
-            # Save changes to Salesforce
-            update_cart_item_in_salesforce(cart_item)  # Custom function to update Salesforce
+        # Query the cart item in 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 Customer_Email__c = '{email}' AND Name = '{item_name}'"
+        )['records']
 
-            # Return a success response with the updated data
-            return jsonify({"success": True, "new_item_price": cart_item['Price__c']})
-        else:
-            return jsonify({"success": False, "error": "Cart item not found."}), 400
+        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_price = cart_items[0].get('Add_Ons_Price__c', 0)
+
+        # Calculate the new item price (base price * quantity + add-ons price)
+        new_item_price = (base_price * quantity) + addons_price
+
+        # Update the record in Salesforce
+        sf.Cart_Item__c.update(cart_item_id, {
+            "Quantity__c": quantity,
+            "Price__c": new_item_price,  # Update base price
+        })
+
+        # Recalculate the subtotal for all items in the cart
+        cart_items = sf.query(f"""
+            SELECT Price__c, Add_Ons_Price__c 
+            FROM Cart_Item__c 
+            WHERE Customer_Email__c = '{email}'
+        """)['records']
+        new_subtotal = sum(item['Price__c'] for item in cart_items)
+
+        # Return updated item price and subtotal
+        return jsonify({
+            "success": True,
+            "new_item_price": round(new_item_price, 2),  # Ensure two decimal places for pricing
+            "subtotal": round(new_subtotal, 2)  # Ensure two decimal places for subtotal
+        })
 
     except Exception as e:
+        # Error handling
+        print(f"Error updating quantity: {str(e)}")
         return jsonify({"success": False, "error": str(e)}), 500
 
+
 # @app.route("/cart/update_quantity", methods=["POST"])
 # def update_quantity():
 #     data = request.json  # Extract JSON data from the request