Update app.py

app.py

@@ -472,7 +472,6 @@ def checkout():
 
     try:
         data = request.json
-        use_reward_points = data.get("useRewardPoints", False)
         selected_coupon = data.get("selectedCoupon", "").strip()  # Get selected coupon
 
         # Fetch cart items
@@ -486,48 +485,53 @@ def checkout():
             return jsonify({"success": False, "message": "Cart is empty"})
 
         total_price = sum(item['Price__c'] for item in cart_items)
-        discount = 0
+        discount = 0  # Default discount is 0
 
-        # Apply reward points if selected
-        customer_record = sf.query(f"""
-            SELECT Id, Reward_Points__c FROM Customer_Login__c
-            WHERE Email__c = '{email}'
+        # Fetch the user's existing coupons
+        coupon_query = sf.query(f"""
+            SELECT Id, Coupon_Code__c FROM Referral_Coupon__c WHERE Referral_Email__c = '{email}'
         """)
-        customer = customer_record.get("records", [])[0] if customer_record else None
 
-        if customer and use_reward_points:
-            if customer['Reward_Points__c'] >= 500:
-                discount += total_price * 0.10  # 10% discount
-                sf.Customer_Login__c.update(customer['Id'], {
-                    "Reward_Points__c": customer['Reward_Points__c'] - 500
-                })
-            else:
-                return jsonify({"success": False, "message": "Insufficient reward points to apply."})
+        has_coupons = bool(coupon_query["records"])  # Check if user has any coupons
 
-        # Apply coupon discount and remove used coupon
         if selected_coupon:
-            discount += total_price * 0.15  # Example: 15% discount for coupon
+            # Case 3: User selected a valid coupon → Apply discount & remove coupon
+            discount = total_price * 0.10  # 10% discount
+            referral_coupon_id = coupon_query["records"][0]["Id"]
+            existing_coupons = coupon_query["records"][0]["Coupon_Code__c"].split("\n")
 
-            # Fetch the user's existing coupons
-            coupon_query = sf.query(f"""
-                SELECT Id, Coupon_Code__c FROM Referral_Coupon__c WHERE Referral_Email__c = '{email}'
-            """)
+            # Remove only the selected coupon
+            updated_coupons = [coupon for coupon in existing_coupons if coupon.strip() != selected_coupon]
+
+            # Convert list back to a string with newlines
+            updated_coupons_str = "\n".join(updated_coupons).strip()
 
-            if coupon_query["records"]:
-                coupon_record = coupon_query["records"][0]
-                referral_coupon_id = coupon_record["Id"]
-                existing_coupons = coupon_record["Coupon_Code__c"].split("\n")  # Convert string to list
+            # Update the Referral_Coupon__c record with the remaining coupons
+            sf.Referral_Coupon__c.update(referral_coupon_id, {
+                "Coupon_Code__c": updated_coupons_str
+            })
+        else:
+            # Case 1 & Case 2: User has no coupons or has coupons but didn’t select one → Add 10% to reward points
+            reward_points_to_add = total_price * 0.10
 
-                # Remove only the selected coupon
-                updated_coupons = [coupon for coupon in existing_coupons if coupon.strip() != selected_coupon]
+            # Fetch current 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
 
-                # Convert list back to a string with newlines
-                updated_coupons_str = "\n".join(updated_coupons).strip()
+            if customer:
+                current_reward_points = customer.get("Reward_Points__c") or 0
+                new_reward_points = current_reward_points + reward_points_to_add
 
-                # Update the Referral_Coupon__c record with the remaining coupons
-                sf.Referral_Coupon__c.update(referral_coupon_id, {
-                    "Coupon_Code__c": updated_coupons_str
+                print(f"Updating reward points: Current = {current_reward_points}, Adding = {reward_points_to_add}, New = {new_reward_points}")
+
+                # Update reward points in Salesforce
+                sf.Customer_Login__c.update(customer["Id"], {
+                    "Reward_Points__c": new_reward_points
                 })
+                print(f"Successfully updated reward points for {email}")
 
         total_bill = total_price - discount
 
@@ -551,7 +555,9 @@ def checkout():
         return jsonify({"success": True, "message": "Order placed successfully!"})
 
     except Exception as e:
+        print(f"Error updating reward points: {str(e)}")
         return jsonify({"success": False, "error": str(e)})
+
 @app.route("/order", methods=["GET"])
 def order_summary():
     email = session.get('user_email')  # Fetch logged-in user's email