File size: 5,745 Bytes
a7abf85 d3245ed 36d2eb6 b12f5e4 9bf1d7d 5324aa9 8369d3e 386c140 d8f342f 6218638 174e074 6218638 a7abf85 6218638 01b8424 a7abf85 6218638 a7abf85 4c2d5e8 d831144 a69087c d3245ed 13d210d 6218638 145b38f 6218638 d8f342f 174e074 d8f342f 6218638 3ad292c a7abf85 c096c2c 4ad81b7 c096c2c 4ad81b7 c096c2c 1fd9c90 7136825 040f053 2731d4a 040f053 f62a0a9 80f989c 0270ecb 80f989c 9de76b8 a7abf85 fcdec6b |
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 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 |
from flask import Flask, render_template, request, jsonify, redirect, url_for, session
from flask_session import Session # Import the Session class
from flask.sessions import SecureCookieSessionInterface # Import the class
from salesforce import get_salesforce_connection
from datetime import timedelta
from email.mime.multipart import MIMEMultipart
from email.mime.text import MIMEText
from menu import menu_blueprint # Make sure this import is correct
from cart import cart_blueprint # Same for other blueprints
from order import order_blueprint # Same for user blueprint
from orderhistory import orderhistory_blueprint
from user_details import user_details_blueprint
from customdish import customdish_blueprint
from datetime import datetime
from datetime import datetime
import pytz # Library to handle timezone conversions
import os
import smtplib
import random
import string
app = Flask(__name__)
# Add debug logs in Salesforce connection setup
sf = get_salesforce_connection()
# Set the secret key to handle sessions securely
app.secret_key = os.getenv("SECRET_KEY", "sSSjyhInIsUohKpG8sHzty2q") # Replace with a secure key
app.config["SESSION_TYPE"] = "filesystem" # Storing sessions in filesystem
app.config["SESSION_COOKIE_SECURE"] = True # Enabling secure cookies (ensure your app is served over HTTPS)
app.config["SESSION_COOKIE_SAMESITE"] = "None" # Cross-site cookies allowed
# Initialize the session
Session(app) # Correctly initialize the Session object
app.session_interface = SecureCookieSessionInterface()
app.register_blueprint(cart_blueprint, url_prefix='/cart')
app.register_blueprint(user_details_blueprint, url_prefix='/user')
app.register_blueprint(menu_blueprint)
app.register_blueprint(order_blueprint)
app.register_blueprint(orderhistory_blueprint, url_prefix='/orderhistory')
app.register_blueprint(customdish_blueprint, url_prefix='/customdish')
@app.route("/")
def home():
# Fetch user details from URL parameters
user_email = request.args.get("email")
user_name = request.args.get("name")
table_number = request.args.get("table") # Capture table number
if user_email and user_name:
session["user_email"] = user_email
session["user_name"] = user_name
session["table_number"] = table_number # Store table number in session
print(f"User logged in: {user_email} - {user_name} - Table: {table_number}")
# Ensure session is saved before redirecting
session.modified = True
return redirect(url_for("menu.menu")) # Redirect to menu directly
return render_template("index.html")
@app.route("/login", methods=["GET", "POST"])
def login():
if request.method == "POST":
email = request.form.get("email")
password = request.form.get("password")
print(f"Login attempt with email: {email}") # Debug log
try:
# Fetch user details from Salesforce
query = f"SELECT Id, Name, Email__c, Reward_Points__c FROM Customer_Login__c WHERE Email__c='{email}' AND Password__c='{password}'"
result = sf.query(query)
if result["records"]:
user = result["records"][0]
session['user_id'] = user['Id']
# ✅ Always store or update session email
if 'user_email' not in session or session['user_email'] != email:
session['user_email'] = email
session['user_name'] = user.get("Name", "")
print(f"✅ Session email updated: {session['user_email']}")
reward_points = user.get("Reward_Points__c") or 0
# Coupon generation logic (if reward points >= 500)
if reward_points >= 500:
new_coupon_code = generate_coupon_code()
coupon_query = sf.query(f"SELECT Id, Coupon_Code__c FROM Referral_Coupon__c WHERE Referral_Email__c = '{email}'")
if coupon_query["records"]:
coupon_record = coupon_query["records"][0]
referral_coupon_id = coupon_record["Id"]
existing_coupons = coupon_record.get("Coupon_Code__c", "")
updated_coupons = f"{existing_coupons}\n{new_coupon_code}".strip()
sf.Referral_Coupon__c.update(referral_coupon_id, {"Coupon_Code__c": updated_coupons})
else:
sf.Referral_Coupon__c.create({
"Referral_Email__c": email,
"Name": user.get("Name", ""),
"Coupon_Code__c": new_coupon_code
})
new_reward_points = reward_points - 500
sf.Customer_Login__c.update(user['Id'], {"Reward_Points__c": new_reward_points})
return redirect(url_for("menu.menu"))
else:
print("Invalid credentials!")
return render_template("login.html", error="Invalid credentials!")
except Exception as e:
print(f"Error during login: {str(e)}")
return render_template("login.html", error=f"Error: {str(e)}")
return render_template("login.html")
@app.route("/logout")
def logout():
# Retrieve table number before clearing session
table_number = session.get('table_number', '')
# Clear session variables
session.pop('name', None)
session.pop('email', None)
session.pop('rewardPoints', None)
session.pop('coupon', None)
# Pass table number to redirect page
return render_template("redirect_page.html", table_number=table_number)
if __name__ == "__main__":
app.run(debug=True, host="0.0.0.0", port=7860) |