DSatishchandra commited on
Commit
05d7139
·
verified ·
1 Parent(s): 2d1c194

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +9 -8
app.py CHANGED
@@ -1,6 +1,7 @@
1
  from flask import Flask, render_template, request, jsonify, redirect, url_for, session
2
  from flask.sessions import SecureCookieSessionInterface
3
  from salesforce import get_salesforce_connection
 
4
  import os
5
 
6
  # Initialize Flask app and Salesforce connection
@@ -15,12 +16,13 @@ print("Salesforce connection established.")
15
  # Set the secret key to handle sessions securely
16
  app.secret_key = os.getenv("SECRET_KEY", "sSSjyhInIsUohKpG8sHzty2q") # Replace with a secure key
17
 
18
- app.config["SESSION_COOKIE_PATH"] = "/" # Available across all routes
19
- app.config["SESSION_COOKIE_DOMAIN"] = None # Let the browser determine the domain
20
- app.config["SESSION_COOKIE_SECURE"] = False # Set to True if HTTPS
21
- app.config["SESSION_COOKIE_HTTPONLY"] = True # Prevent JavaScript access
22
- app.config["SESSION_PERMANENT"] = False # Do not use permanent sessions
23
 
 
24
 
25
  # Ensure secure session handling for environments like Hugging Face
26
  app.session_interface = SecureCookieSessionInterface()
@@ -54,7 +56,6 @@ def login():
54
  if request.method == "POST":
55
  email = request.form.get("email")
56
  password = request.form.get("password")
57
- print(f"Login Succesful")
58
  print(f"Login attempt with email: {email}") # Debug log
59
 
60
  try:
@@ -65,7 +66,7 @@ def login():
65
  session['user_id'] = result["records"][0]['Id']
66
  session['user_email'] = email
67
  print(f"Session variables set: user_id={session['user_id']}, user_email={session['user_email']}")
68
- print(f"Session cookie: {app.session_interface.get_signing_serializer(app).dumps(session)}") # Dump session
69
  return redirect(url_for("menu"))
70
  else:
71
  print("Invalid credentials!")
@@ -78,9 +79,9 @@ def login():
78
 
79
  @app.route("/menu", methods=["GET", "POST"])
80
  def menu():
81
- print(f"Cookies on /menu: {request.cookies}")
82
  selected_category = request.args.get("category", "All")
83
  user_id = session.get('user_id')
 
84
  print(f"Session check in /menu: user_id={user_id}")
85
 
86
  if not user_id:
 
1
  from flask import Flask, render_template, request, jsonify, redirect, url_for, session
2
  from flask.sessions import SecureCookieSessionInterface
3
  from salesforce import get_salesforce_connection
4
+ from redis import Redis
5
  import os
6
 
7
  # Initialize Flask app and Salesforce connection
 
16
  # Set the secret key to handle sessions securely
17
  app.secret_key = os.getenv("SECRET_KEY", "sSSjyhInIsUohKpG8sHzty2q") # Replace with a secure key
18
 
19
+ # Set session cookie policies
20
+ app.config["SESSION_TYPE"] = "redis"
21
+ app.config["SESSION_REDIS"] = Redis(host="localhost", port=6379) # Replace with Redis details
22
+ app.config["SESSION_COOKIE_SAMESITE"] = "None"
23
+ app.config["SESSION_COOKIE_SECURE"] = True
24
 
25
+ Session(app)
26
 
27
  # Ensure secure session handling for environments like Hugging Face
28
  app.session_interface = SecureCookieSessionInterface()
 
56
  if request.method == "POST":
57
  email = request.form.get("email")
58
  password = request.form.get("password")
 
59
  print(f"Login attempt with email: {email}") # Debug log
60
 
61
  try:
 
66
  session['user_id'] = result["records"][0]['Id']
67
  session['user_email'] = email
68
  print(f"Session variables set: user_id={session['user_id']}, user_email={session['user_email']}")
69
+ print(f"Session cookie: {request.cookies.get(app.session_cookie_name)}") # Check session cookie
70
  return redirect(url_for("menu"))
71
  else:
72
  print("Invalid credentials!")
 
79
 
80
  @app.route("/menu", methods=["GET", "POST"])
81
  def menu():
 
82
  selected_category = request.args.get("category", "All")
83
  user_id = session.get('user_id')
84
+ print(f"Cookies on /menu: {request.cookies}") # Debug: Check cookies sent
85
  print(f"Session check in /menu: user_id={user_id}")
86
 
87
  if not user_id: