Update app.py

app.py

@@ -60,7 +60,7 @@ def signup():
         generated_referral_code = generate_referral_code()
 
         try:
-            ref=0  # Default reward points for new user
+            ref = 0  # Default reward points for new user
 
             # Check if a referral code is entered
             if referral_code:
@@ -75,14 +75,31 @@ def signup():
                 referrer = referral_result['records'][0]
                 referrer_email = referrer.get('Email__c')
 
-                # Generate a unique coupon code
+                # Generate a new unique coupon code
                 new_coupon_code = generate_coupon_code()
 
-                # Create a new referral coupon entry in Salesforce
-                sf.Referral_Coupon__c.create({
-                    "Referral_Email__c": referrer_email,
-                    "Coupon_Code__c": new_coupon_code
-                })
+                # Check if referrer already has a record in Referral_Coupon__c
+                existing_coupon_query = f"SELECT Id, Coupon_Code__c FROM Referral_Coupon__c WHERE Referral_Email__c = '{referrer_email}'"
+                existing_coupon_result = sf.query(existing_coupon_query)
+
+                if existing_coupon_result['records']:
+                    # If record exists, append the new coupon on the next line
+                    referral_record = existing_coupon_result['records'][0]
+                    referral_id = referral_record['Id']
+                    existing_coupons = referral_record.get('Coupon_Code__c', '')
+
+                    updated_coupons = f"{existing_coupons}\n{new_coupon_code}".strip()
+
+                    # Update the existing record with the new coupon
+                    sf.Referral_Coupon__c.update(referral_id, {
+                        "Coupon_Code__c": updated_coupons
+                    })
+                else:
+                    # If no record exists, create a new one
+                    sf.Referral_Coupon__c.create({
+                        "Referral_Email__c": referrer_email,
+                        "Coupon_Code__c": new_coupon_code
+                    })
 
             # Create the new customer record in Salesforce
             sf.Customer_Login__c.create({
@@ -93,7 +110,7 @@ def signup():
                 "Reward_Points__c": ref,  # No points added, only coupon is created
                 "Referral__c": generated_referral_code
             })
-            
+
             return redirect(url_for("login"))
 
         except Exception as e: