DSatishchandra commited on
Commit
10d040f
·
verified ·
1 Parent(s): 8818c88

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -1
app.py CHANGED
@@ -2,13 +2,16 @@ from flask import Flask, render_template, request, jsonify, redirect, url_for, s
2
  import random
3
  from salesforce import get_salesforce_connection
4
  import os
 
 
 
5
 
6
  # Initialize Flask app and Salesforce connection
7
  app = Flask(__name__)
8
  sf = get_salesforce_connection()
9
 
10
  # Set the secret key to handle sessions securely
11
- app.secret_key = 'sSSjyhInIsUohKpG8sHzty2q' # Replace with a secure key
12
 
13
  @app.route("/")
14
  def home():
@@ -38,6 +41,7 @@ def login():
38
  if request.method == "POST":
39
  email = request.form.get("email")
40
  password = request.form.get("password")
 
41
  try:
42
  query = f"SELECT Id, Name, Email__c FROM Customer_Login__c WHERE Email__c='{email}' AND Password__c='{password}'"
43
  result = sf.query(query)
@@ -45,6 +49,7 @@ def login():
45
  session['user_id'] = result["records"][0]['Id']
46
  session['user_email'] = email
47
  return redirect(url_for("menu"))
 
48
  else:
49
  return render_template("login.html", error="Invalid credentials!")
50
  except Exception as e:
 
2
  import random
3
  from salesforce import get_salesforce_connection
4
  import os
5
+ from flask_session import Session
6
+ app.config["SESSION_TYPE"] = "filesystem"
7
+ Session(app)
8
 
9
  # Initialize Flask app and Salesforce connection
10
  app = Flask(__name__)
11
  sf = get_salesforce_connection()
12
 
13
  # Set the secret key to handle sessions securely
14
+ app.secret_key = os.getenv("sSSjyhInIsUohKpG8sHzty2q", "sSSjyhInIsUohKpG8sHzty2q") # Replace with a secure key
15
 
16
  @app.route("/")
17
  def home():
 
41
  if request.method == "POST":
42
  email = request.form.get("email")
43
  password = request.form.get("password")
44
+ print(f"Login attempt with email: {email}")
45
  try:
46
  query = f"SELECT Id, Name, Email__c FROM Customer_Login__c WHERE Email__c='{email}' AND Password__c='{password}'"
47
  result = sf.query(query)
 
49
  session['user_id'] = result["records"][0]['Id']
50
  session['user_email'] = email
51
  return redirect(url_for("menu"))
52
+ print(f"Session variables: {session}")
53
  else:
54
  return render_template("login.html", error="Invalid credentials!")
55
  except Exception as e: