Update customdish.py

customdish.py

@@ -2,6 +2,7 @@ from flask import Blueprint, request, jsonify, redirect, url_for, session
 import random
 from salesforce import get_salesforce_connection
 sf = get_salesforce_connection()
+
 # Create a Blueprint for custom dish-related routes
 customdish_blueprint = Blueprint('customdish', __name__)
 
@@ -18,14 +19,12 @@ def generate_custom_dish():
             return jsonify({"success": False, "error": "Both fields are required"}), 400
         dish_name = dish_name.replace("'", "+")
 
-
         # Generate a random price for the custom dish
-        price = random.randint(10, 30)  # Example logic for price setting
+        price = random.randint(10, 30)
 
         # Determine Veg/Non-Veg
         veg_keywords = ["paneer", "vegetable", "mushroom", "cheese"]
         non_veg_keywords = ["chicken", "mutton", "fish", "egg"]
-        
         category = "Veg" if any(word in description.lower() for word in veg_keywords) else \
                    "Non veg" if any(word in description.lower() for word in non_veg_keywords) else \
                    "both"
@@ -53,55 +52,46 @@ def generate_custom_dish():
                 'Section__c': 'Customized dish',
                 'Total_Ordered__c': 0
             }
-
-            # Insert the custom dish into Salesforce (Custom_Dish__c object)
             result = sf.Custom_Dish__c.create(custom_dish)
-
             if not result.get('success'):
                 return jsonify({"success": False, "error": "Failed to create custom dish in Salesforce"}), 500
 
-        # After ensuring the dish exists, check if it's already in the Cart_Item__c
-        email = session.get('user_email')  # Assuming you have the user's email in session
-        
-        # Query to check if the custom dish already exists in the cart for the logged-in user
+        # Add or update the custom dish in the cart
+        email = session.get('user_email')
+        if not email:
+            return jsonify({"success": False, "error": "User not logged in"}), 401
+
+        # Query to check if the custom dish already exists in the cart
         cart_item_query = f"SELECT Id, Quantity__c, Price__c, Base_Price__c FROM Cart_Item__c WHERE Customer_Email__c = '{email}' AND Name = '{dish_name}'"
         cart_item_result = sf.query(cart_item_query)
 
         if cart_item_result['totalSize'] > 0:
-            # If the custom dish is already in the cart, update the quantity and price
+            # Update existing cart item
             cart_item = cart_item_result['records'][0]
-            new_quantity = cart_item['Quantity__c'] + 1  # Increase quantity by 1
-            new_price = price * new_quantity  # Update price based on new quantity
-            
-            # Update the cart item in Salesforce
+            new_quantity = cart_item['Quantity__c'] + 1
+            new_price = price * new_quantity
             updated_cart_item = {
                 'Quantity__c': new_quantity,
                 'Price__c': new_price
             }
-            
-            cart_item_update = sf.Cart_Item__c.update(cart_item['Id'], updated_cart_item)
-
+            sf.Cart_Item__c.update(cart_item['Id'], updated_cart_item)
         else:
-            # If the custom dish is not in the cart, create a new cart item
+            # Create new cart item
             cart_item = {
                 'Name': dish_name,
                 'Price__c': price,
                 'Base_Price__c': price,
                 'Image1__c': item_image_url,
-                'Quantity__c': 1,  # Default quantity is 1
-                'Add_Ons__c': '',  # Set Add_ons__c to empty
-                'Add_Ons_Price__c': 0,  # Set Add_ons_Price__c to 0
-                'Customer_Email__c': email  # Associate the custom dish with the logged-in user
+                'Quantity__c': 1,
+                'Add_Ons__c': '',
+                'Add_Ons_Price__c': 0,
+                'Customer_Email__c': email,
+                'Category__c': category,
+                'Section__c': 'Customized dish'
             }
+            sf.Cart_Item__c.create(cart_item)
 
-            # Insert the custom dish as a Cart_Item__c record in Salesforce
-            cart_result = sf.Cart_Item__c.create(cart_item)
-
-        # Redirect to the cart page after successfully adding or updating the cart item
-        return redirect(url_for('cart.cart'))
-
+        return jsonify({"success": True, "message": "Custom dish added to cart successfully"})
 
     except Exception as e:
-        return jsonify({"success": False, "error": str(e)}), 500
-
-
+        return jsonify({"success": False, "error": str(e)}), 500