Update app.py

app.py

@@ -586,9 +586,53 @@ def checkout():
 
         total_price = sum(item['Price__c'] for item in cart_items)
         discount = 0
+        # 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}'
+        """)
+
+        has_coupons = bool(coupon_query["records"])  # Check if user has any coupons
 
         if selected_coupon:
-            discount = total_price * 0.10  # Apply 10% discount
+            # 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")
+
+            # 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()
+
+            # 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
+
+            # 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
+
+            if customer:
+                current_reward_points = customer.get("Reward_Points__c") or 0
+                new_reward_points = current_reward_points + reward_points_to_add
+
+                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