Update app.py

app.py

@@ -42,6 +42,13 @@ def generate_referral_code(length=8):
 def home():
     return render_template("index.html")
 
+from datetime import datetime
+
+def generate_coupon_code(length=10):
+    """Generates a random alphanumeric coupon code"""
+    characters = string.ascii_uppercase + string.digits  # A-Z, 0-9
+    return ''.join(random.choice(characters) for _ in range(length))
+
 @app.route("/signup", methods=["GET", "POST"])
 def signup():
     if request.method == "POST":
@@ -53,45 +60,50 @@ def signup():
         generated_referral_code = generate_referral_code()
 
         try:
-            ref=0
+            ref=0  # Default reward points for new user
+
             # Check if a referral code is entered
             if referral_code:
                 # Query Salesforce to check if the referral code exists
-                referral_query = f"SELECT Id, Reward_Points__c FROM Customer_Login__c WHERE Referral__c = '{referral_code}'"
+                referral_query = f"SELECT Id, Email__c FROM Customer_Login__c WHERE Referral__c = '{referral_code}'"
                 referral_result = sf.query(referral_query)
-                ref=50
 
                 if not referral_result['records']:
-                    # If referral code does not exist, show an error message
                     return render_template("signup.html", error="Invalid referral code!")
 
-                # If referral exists, get the first matching record
+                # Get referrer's details
                 referrer = referral_result['records'][0]
-                referrer_id = referrer['Id']
-                # Fetch the existing reward points, defaulting to 0 if None
-                existing_reward_points = referrer.get('Reward_Points__c') or 0  # Default to 0 if the field is None
-                updated_reward_points = existing_reward_points + 100
-                # Update the referrer's Reward_Points__c field in Salesforce
-                sf.Customer_Login__c.update(referrer_id, {
-                    "Reward_Points__c": updated_reward_points
+                referrer_email = referrer.get('Email__c')
+
+                # Generate a 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,
+                    "Created_Date__c": datetime.utcnow().isoformat()  # Optional timestamp
                 })
-            
+
             # Create the new customer record in Salesforce
             sf.Customer_Login__c.create({
                 "Name": name,
                 "Phone_Number__c": phone,
                 "Email__c": email,
                 "Password__c": password,
-                "Reward_Points__c": ref,
+                "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:
             return render_template("signup.html", error=f"Error: {str(e)}")
+
     return render_template("signup.html")
 
 
+
 @app.route("/login", methods=["GET", "POST"])
 def login():
     if request.method == "POST":