handle the referral url that has ?slref=code part

app/auth/views/login_utils.py

@@ -53,4 +53,12 @@ def get_referral() -> Optional[Referral]:
         ref_code = request.cookies.get(_REFERRAL_COOKIE)
         referral = Referral.get_by(code=ref_code)
 
+    if not referral:
+        if "slref" in session:
+            ref_code = session["slref"]
+            referral = Referral.get_by(code=ref_code)
+
+    if referral:
+        LOG.d("referral found %s", referral)
+
     return referral

app/models.py

@@ -2269,6 +2269,9 @@ class Referral(Base, ModelMixin):
     def link(self):
         return f"{LANDING_PAGE_URL}?slref={self.code}"
 
+    def __repr__(self):
+        return f"<Referral {self.code}>"
+
 
 class SentAlert(Base, ModelMixin):
     """keep track of alerts sent to user.

server.py

@@ -481,6 +481,11 @@ def set_index_page(app):
         ):
             g.start_time = time.time()
 
+            # to handle the referral url that has ?slref=code part
+            ref_code = request.args.get("slref")
+            if ref_code:
+                session["slref"] = ref_code
+
     @app.after_request
     def after_request(res):
         # not logging /static call