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
|