Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -275,13 +275,13 @@ def menu():
|
|
275 |
print(f"Session check in /menu: user_email={user_email}")
|
276 |
|
277 |
|
278 |
-
|
279 |
# Fetch the user's Referral__c and Reward_Points__c
|
280 |
user_query = f"SELECT Referral__c, Reward_Points__c FROM Customer_Login__c WHERE Email__c = '{user_email}'"
|
281 |
user_result = sf.query(user_query)
|
282 |
|
283 |
if not user_result['records']:
|
284 |
-
print("User not found!")
|
285 |
return redirect(url_for('login'))
|
286 |
|
287 |
referral_code = user_result['records'][0].get('Referral__c', 'N/A') # Default to 'N/A' if empty
|
|
|
275 |
print(f"Session check in /menu: user_email={user_email}")
|
276 |
|
277 |
|
278 |
+
try:
|
279 |
# Fetch the user's Referral__c and Reward_Points__c
|
280 |
user_query = f"SELECT Referral__c, Reward_Points__c FROM Customer_Login__c WHERE Email__c = '{user_email}'"
|
281 |
user_result = sf.query(user_query)
|
282 |
|
283 |
if not user_result['records']:
|
284 |
+
print("❌ User not found!")
|
285 |
return redirect(url_for('login'))
|
286 |
|
287 |
referral_code = user_result['records'][0].get('Referral__c', 'N/A') # Default to 'N/A' if empty
|