Update app.py

app.py

@@ -489,35 +489,52 @@ def checkout():
         return jsonify({"success": False, "error": str(e)})
 
 
+@app.route('/get_coupon_code', methods=['GET'])
+def get_coupon_code():
+    email = request.args.get('email')  # Get the email from the request
+
+    if not email:
+        return jsonify({'success': False, 'message': 'Email is required.'})
+
+    try:
+        # Query Salesforce to find the coupon code for the given email
+        query = f"SELECT Coupon_Code__c FROM Referral_Coupon__c WHERE Referral_Email__c = '{email}' AND Coupon_Status__c = 'Active'"
+        result = sf.query(query)
+
+        if result['records']:
+            # If a valid coupon code is found, return it
+            coupon_code = result['records'][0]['Coupon_Code__c']
+            return jsonify({'success': True, 'coupon_code': coupon_code})
+
+        return jsonify({'success': False, 'message': 'No active coupon found for this email.'})
+
+    except Exception as e:
+        return jsonify({'success': False, 'message': f"Error fetching coupon code: {str(e)}"})
+
+
 @app.route('/apply_coupon', methods=['POST'])
 def apply_coupon():
-    # Get the coupon code and subtotal from the POST request
     data = request.json
     coupon_code = data.get('coupon_code')
     subtotal = data.get('subtotal')
-    user_email = session.get('user_email')  # Assuming user is logged in and email is in session
 
     if not coupon_code:
         return jsonify({'success': False, 'message': 'Coupon code is required.'})
 
     try:
-        # Query the Salesforce `Referral_Coupon__c` object to check if the coupon code is valid
-        query = f"SELECT Coupon_Code__c, Discount__c FROM Referral_Coupon__c WHERE Coupon_Code__c = '{coupon_code}' AND Referral_Email__c = '{user_email}' AND Coupon_Status__c = 'Active'"
+        # Query Salesforce to find the discount for the provided coupon code
+        query = f"SELECT Coupon_Code__c, Discount__c FROM Referral_Coupon__c WHERE Coupon_Code__c = '{coupon_code}' AND Coupon_Status__c = 'Active'"
         result = sf.query(query)
 
         if not result['records']:
             return jsonify({'success': False, 'message': 'Invalid or expired coupon code.'})
 
-        # Assuming the discount is a percentage, we will calculate the discount
         coupon = result['records'][0]
-        discount = coupon['Discount__c']  # Discount value from Salesforce (should be in percentage, e.g., 0.10 for 10%)
+        discount = coupon['Discount__c']  # Assuming discount is a percentage (e.g., 0.10 for 10%)
 
         # Apply discount (if it's a percentage)
-        if discount > 1:  # Fixed discount
-            return jsonify({'success': True, 'discount': discount})
-        else:  # Percentage discount
-            discount_amount = subtotal * discount  # Calculate discount on the subtotal
-            return jsonify({'success': True, 'discount': discount_amount})
+        discount_amount = subtotal * discount
+        return jsonify({'success': True, 'discount': discount_amount})
 
     except Exception as e:
         return jsonify({'success': False, 'message': f'Error applying coupon: {str(e)}'})