Update app.py

app.py

@@ -49,20 +49,46 @@ def signup():
         phone = request.form.get("phone")
         email = request.form.get("email")
         password = request.form.get("password")
-        referral_code = generate_referral_code()
+        referral_code = request.form.get("referral")  # Fetch referral code from the form
+        generated_referral_code = generate_referral_code()
+
         try:
+            # 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_result = sf.query(referral_query)
+
+                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
+                referrer = referral_result['records'][0]
+                referrer_id = referrer['Id']
+                existing_reward_points = referrer.get('Reward_Points__c', 0)
+
+                # Add 100 points to the referrer's Reward_Points__c field
+                updated_reward_points = existing_reward_points + 100
+                sf.Customer_Login__c.update(referrer_id, {
+                    "Reward_Points__c": updated_reward_points
+                })
+
+            # Create the new customer record in Salesforce
             sf.Customer_Login__c.create({
                 "Name": name,
                 "Phone_Number__c": phone,
                 "Email__c": email,
                 "Password__c": password,
-                "Referral__c": referral_code
+                "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":