Update app.py

app.py

@@ -45,6 +45,42 @@ def generate_coupon_code(length=10):
     """Generate a unique alphanumeric coupon code."""
     return ''.join(random.choices(string.ascii_uppercase + string.digits, k=length))
 
+def process_referral_code(referral_code):
+    """
+    Process the referral code by creating a referral coupon if the referrer is found.
+
+    Args:
+        referral_code (str): The referral code provided by the user.
+
+    Returns:
+        str: A message indicating the result of the referral code processing.
+    """
+    try:
+        # Query for the referrer using the referral code
+        query = f"SELECT Id, Name, Email__c FROM Customer_Login__c WHERE Customer_Referral_Code__c = '{referral_code}'"
+        referral_result = sf.query(query)
+
+        if len(referral_result['records']) == 0:
+            return "Referral not found. No coupon generated."
+        else:
+            # Fetch the referrer's details
+            referrer = referral_result["records"][0]
+
+            # Generate a unique coupon code
+            coupon_code = generate_coupon_code()
+
+            # Create a referral coupon in Salesforce
+            sf.Referral_Coupon__c.create({
+                "Referral_Name__c": referrer["Name"],
+                "Referral_Email__c": referrer["Email__c"],
+                "Coupon_Code__c": coupon_code,
+                "Coupon_Status__c": "Active"
+            })
+
+            return f"Coupon code '{coupon_code}' generated for referrer '{referrer['Name']}' ({referrer['Email__c']})."
+    except Exception as e:
+        return f"Error processing referral code: {str(e)}"
+
 
 
 @app.route("/signup", methods=["GET", "POST"])
@@ -76,22 +112,8 @@ def signup():
 
             # Step 3: Check if a referral code was provided
             if referral_code:
-                query = f"SELECT Id, Name, Email__c FROM Customer_Login__c WHERE Customer_Referral_Code__c = '{referral_code}'"
-                referral_result = sf.query(query)
-
-                if len(referral_result['records']) == 0:
-                        return "Referral not found. No coupon generated."
-                else:
-                    referrer = referral_result["records"][0]
-                        # Generate a unique coupon code
-                    coupon_code = generate_coupon_code()
-                    
-                    sf.Referral_Coupon__c.create({
-                        "Referral_Name__c": referrer["Name"],
-                        "Referral_Email__c": referrer["Email__c"],
-                        "Coupon_Code__c": coupon_code,
-                        "Coupon_Status__c": "Active"
-                    })
+                referral_message = process_referral_code(referral_code)
+
             return redirect(url_for("login"))
         except Exception as e:
             return render_template("signup.html", error=f"Error: {str(e)}")