File size: 4,980 Bytes
ceac236 024ce39 ceac236 2e45435 ceac236 22a5f92 ceac236 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 |
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__)
@customdish_blueprint.route("/generate_custom_dish", methods=["POST"])
def generate_custom_dish():
try:
data = request.form
dish_name = data.get("name")
description = data.get("description")
item_image_url = "https://huggingface.co/spaces/nagasurendra/BiryaniHubflask30/resolve/main/static/customized.jpg"
item_image_url2 = "https://huggingface.co/spaces/nagasurendra/BiryaniHubflask30/resolve/main/static/customized1.jpg"
if not dish_name or not description:
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
# 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"
# Query to check if the dish already exists in Salesforce (Custom_Dish__c object)
existing_dish_query = f"SELECT Id, Name, Price__c, Image1__c, Image2__c, Description__c, Veg_NonVeg__c FROM Custom_Dish__c WHERE Name = '{dish_name}'"
existing_dish_result = sf.query(existing_dish_query)
if existing_dish_result['totalSize'] > 0:
# If the dish exists, use the existing details
existing_dish = existing_dish_result['records'][0]
price = existing_dish['Price__c']
item_image_url = existing_dish['Image1__c']
item_image_url2 = existing_dish['Image2__c']
category = existing_dish['Veg_NonVeg__c']
else:
# If the dish does not exist, create a new custom dish
custom_dish = {
'Name': dish_name,
'Price__c': price,
'Image1__c': item_image_url,
'Image2__c': item_image_url2,
'Description__c': description,
'Veg_NonVeg__c': category,
'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
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
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
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)
else:
# If the custom dish is not in the cart, create a 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
}
# 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'))
except Exception as e:
return jsonify({"success": False, "error": str(e)}), 500
|