from flask import Flask, render_template, request, jsonify, redirect, url_for
from simple_salesforce import Salesforce
# Initialize Flask app and Salesforce connection
app = Flask(__name__)
# Function to connect to Salesforce
def get_salesforce_connection():
sf = Salesforce(
username="diggavalli98@gmail.com",
password="Sati@1020",
security_token="sSSjyhInIsUohKpG8sHzty2q"
)
return sf
sf = get_salesforce_connection()
# Home route
@app.route("/")
def home():
return render_template("index.html")
# Signup route
@app.route("/signup", methods=["GET", "POST"])
def signup():
if request.method == "POST":
name = request.form.get("name")
phone = request.form.get("phone")
email = request.form.get("email")
password = request.form.get("password")
try:
sf.Customer_Login__c.create({
"Name": name,
"Phone__c": phone,
"Email__c": email,
"Password__c": password
})
return redirect(url_for("login"))
except Exception as e:
return render_template("signup.html", error=f"Error: {str(e)}")
return render_template("signup.html")
# Login route
@app.route("/login", methods=["GET", "POST"])
def login():
if request.method == "POST":
email = request.form.get("email")
password = request.form.get("password")
try:
query = f"SELECT Name FROM Customer_Login__c WHERE Email__c='{email}' AND Password__c='{password}'"
result = sf.query(query)
if result["records"]:
return redirect(url_for("menu"))
return render_template("login.html", error="Invalid credentials!")
except Exception as e:
return render_template("login.html", error=f"Error: {str(e)}")
return render_template("login.html")
# Menu route
@app.route("/menu", methods=["GET", "POST"])
def menu():
selected_category = request.args.get("category", "All")
try:
query = "SELECT Name, Price__c, Image1__c, Cateogry__c, Description__c FROM Menu_Item__c"
result = sf.query(query)
food_items = result['records']
categories = {item['Cateogry__c'] for item in food_items if 'Cateogry__c' in item}
if selected_category != "All":
food_items = [item for item in food_items if item.get("Cateogry__c") == selected_category]
except Exception as e:
food_items = []
categories = []
print(f"Error fetching data: {e}")
return render_template("menu.html", food_items=food_items, categories=categories, selected_category=selected_category)
# Cart route
@app.route("/cart")
def cart():
try:
result = sf.query(
"""
SELECT Name, Price__c, Quantity__c, Image1__c
FROM Cart_Item__c
WHERE User__c = 'Current_User_Id' -- Replace this with logic to fetch current user ID
"""
)
cart_items = result["records"]
subtotal = sum(item['Quantity__c'] * item['Price__c'] for item in cart_items)
except Exception as e:
print(f"Error fetching cart items: {e}")
cart_items = []
subtotal = 0
return render_template("cart.html", cart_items=cart_items, subtotal=subtotal)
# Add to cart API route
@app.route("/add_to_cart", methods=["POST"])
def add_to_cart():
data = request.json
try:
sf.Cart_Item__c.create({
"Menu_Item__c": data["menu_item_id"],
"Quantity__c": 1,
"In_Cart__c": True
})
return jsonify({"success": True})
except Exception as e:
print(f"Error adding to cart: {e}")
return jsonify({"success": False, "error": str(e)})
# Addons API route
@app.route("/api/addons")
def get_addons():
menu_item_id = request.args.get("menu_item_id")
try:
query = f"""
SELECT Name, Price__c
FROM Add_On__c
WHERE Menu_Item__c = '{menu_item_id}'
AND Available_In_Menu__c = true
"""
result = sf.query(query)
addons = [{"id": addon["Id"], "name": addon["Name"], "price": addon["Price__c"]} for addon in result["records"]]
return jsonify(addons)
except Exception as e:
return jsonify({"error": str(e)})
# HTML files and JavaScript included in the templates
@app.route("/signup.html")
def signup_html():
return '''
Signup
'''
# Add more HTML for other pages similarly
if __name__ == "__main__":
app.run(debug=True, port=8080)