Update app.py

app.py

@@ -120,68 +120,78 @@ def signup():
 
 
 
-@app.route("/redeem_coupon", methods=["POST"])
-def redeem_coupon():
-    email = session.get('user_email')
-
-    if not email:
-        return jsonify({"success": False, "message": "User not logged in"})
-
-    try:
-        # Fetch reward points
-        customer_record = sf.query(f"""
-            SELECT Id, Reward_Points__c FROM Customer_Login__c WHERE Email__c = '{email}'
-        """)
-        customer = customer_record.get("records", [])[0] if customer_record else None
-
-        if not customer:
-            return jsonify({"success": False, "message": "User not found"})
-
-        reward_points = customer.get("Reward_Points__c", 0)
-
-        if reward_points < 500:
-            return jsonify({"success": False, "message": "Not enough reward points to redeem a coupon"})
-
-        # Generate a new coupon code
-        new_coupon_code = generate_coupon_code()
-
-        # Check if user already has a record in Referral_Coupon__c
-        coupon_query = sf.query(f"""
-            SELECT Id, Coupon_Code__c FROM Referral_Coupon__c WHERE Referral_Email__c = '{email}'
-        """)
+@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
 
-        if coupon_query["records"]:
-            # ✅ Case 1: User already has a referral coupon record → Append new coupon
-            coupon_record = coupon_query["records"][0]
-            referral_coupon_id = coupon_record["Id"]
-            existing_coupons = coupon_record.get("Coupon_Code__c", "")
+        try:
+            # Fetch user details, including Reward_Points__c
+            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']
+                session['user_email'] = email
+                print(f"Session variables set: user_id={session['user_id']}, user_email={session['user_email']}")
+
+                reward_points = user.get("Reward_Points__c", 0)
+
+                if reward_points >= 500:
+                    print(f"User {email} has {reward_points} reward points. Generating coupon...")
+
+                    # Generate a new coupon code
+                    new_coupon_code = generate_coupon_code()
+
+                    # Check if user already has a record in Referral_Coupon__c
+                    coupon_query = sf.query(f"""
+                        SELECT Id, Coupon_Code__c FROM Referral_Coupon__c WHERE Referral_Email__c = '{email}'
+                    """)
+
+                    if coupon_query["records"]:
+                        # If record exists, append the new coupon
+                        coupon_record = coupon_query["records"][0]
+                        referral_coupon_id = coupon_record["Id"]
+                        existing_coupons = coupon_record.get("Coupon_Code__c", "")
+
+                        # Append the new coupon on the next line
+                        updated_coupons = f"{existing_coupons}\n{new_coupon_code}".strip()
+
+                        # Update the Referral_Coupon__c record
+                        sf.Referral_Coupon__c.update(referral_coupon_id, {
+                            "Coupon_Code__c": updated_coupons
+                        })
+                        print(f"Updated existing coupon record for {email}. New Coupon: {new_coupon_code}")
+                    else:
+                        # If no record exists, create a new one
+                        sf.Referral_Coupon__c.create({
+                            "Referral_Email__c": email,
+                            "Coupon_Code__c": new_coupon_code
+                        })
+                        print(f"Created new coupon record for {email}. Coupon: {new_coupon_code}")
+
+                    # Subtract 500 reward points from user's account
+                    new_reward_points = reward_points - 500
+                    sf.Customer_Login__c.update(user['Id'], {
+                        "Reward_Points__c": new_reward_points
+                    })
+                    print(f"Coupon {new_coupon_code} generated and 500 points deducted. New balance: {new_reward_points}")
 
-            updated_coupons = f"{existing_coupons}\n{new_coupon_code}".strip()
+                return redirect(url_for("menu"))
 
-            # ✅ Update the Referral_Coupon__c record
-            sf.Referral_Coupon__c.update(referral_coupon_id, {
-                "Coupon_Code__c": updated_coupons
-            })
-            print(f"✅ Coupon appended for existing user: {email}")
-        else:
-            # ❌ Case 2: No record found → Create a new record in Referral_Coupon__c
-            sf.Referral_Coupon__c.create({
-                "Referral_Email__c": email,
-                "Coupon_Code__c": new_coupon_code
-            })
-            print(f"✅ New Referral_Coupon__c record created for {email}")
+            else:
+                print("Invalid credentials!")
+                return render_template("login.html", error="Invalid credentials!")
 
-        # ✅ Deduct 500 reward points
-        new_reward_points = reward_points - 500
-        sf.Customer_Login__c.update(customer['Id'], {
-            "Reward_Points__c": new_reward_points
-        })
-        print(f"✅ Reward points updated: New balance = {new_reward_points}")
+        except Exception as e:
+            print(f"Error during login: {str(e)}")
+            return render_template("login.html", error=f"Error: {str(e)}")
 
-        return jsonify({"success": True, "message": f"Coupon {new_coupon_code} generated!", "new_reward_points": new_reward_points})
+    return render_template("login.html")
 
-    except Exception as e:
-        return jsonify({"success": False, "error": str(e)})
 
 @app.route("/menu", methods=["GET", "POST"])
 def menu():