Update app.py

app.py

@@ -934,62 +934,88 @@ 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():
-    data = request.json  # Extract JSON data from the request
-    email = data.get('email')
-    item_name = data.get('item_name')
-    try:
-        # Convert quantity to an integer
-        quantity = int(data.get('quantity'))
-    except (ValueError, TypeError):
-        return jsonify({"success": False, "error": "Invalid quantity provided."}), 400
-
-    # 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
-
     try:
-        # 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']
-
-        if not cart_items:
-            return jsonify({"success": False, "error": "Cart item not found."}), 404
+        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
 
-        # 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
-        new_item_price = (base_price * quantity) + addons_price
+        # Update the quantity in Salesforce or your database
+        cart_item = find_cart_item(item_name, customer_email)  # Custom function to find the cart item
+        
+        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
 
-        # Update the record in Salesforce
-        sf.Cart_Item__c.update(cart_item_id, {
-            "Quantity__c": quantity,
-            "Price__c": new_item_price,  # Update base price
-        })
+            # Save changes to Salesforce
+            update_cart_item_in_salesforce(cart_item)  # Custom function to update Salesforce
 
-        # 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": new_item_price, "subtotal": new_subtotal})
-        print(f"New item price: {new_item_price}, New subtotal: {new_subtotal}")
-        return jsonify({"success": True, "new_item_price": new_item_price, "subtotal": new_subtotal})
+            # 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
 
     except Exception as e:
-        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
+#     email = data.get('email')
+#     item_name = data.get('item_name')
+#     try:
+#         # Convert quantity to an integer
+#         quantity = int(data.get('quantity'))
+#     except (ValueError, TypeError):
+#         return jsonify({"success": False, "error": "Invalid quantity provided."}), 400
+
+#     # 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
+
+#     try:
+#         # 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']
+
+#         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
+#         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": new_item_price, "subtotal": new_subtotal})
+#         print(f"New item price: {new_item_price}, New subtotal: {new_subtotal}")
+#         return jsonify({"success": True, "new_item_price": new_item_price, "subtotal": new_subtotal})
+
+#     except Exception as e:
+#         print(f"Error updating quantity: {str(e)}")
+#         return jsonify({"success": False, "error": str(e)}), 500
 @app.route("/checkout", methods=["POST"])
 def checkout():
     email = session.get('user_email')