add referrer

MagicalAuth.py

@@ -233,7 +233,7 @@ def count_failed_logins(self):
         session.close()
         return failed_logins
 
-    def send_magic_link(self, ip_address, login: Login):
+    def send_magic_link(self, ip_address, login: Login, referrer=None):
         self.email = login.email.lower()
         session = get_session()
         user = session.query(User).filter(User.email == self.email).first()
@@ -284,6 +284,8 @@ def send_magic_link(self, ip_address, login: Login):
             .replace("`", "%60")
             .replace("~", "%7E")
         )
+        if referrer is not None:
+            self.link = referrer
         magic_link = f"{self.link}?token={token}"
         if (
             getenv("SENDGRID_API_KEY") != ""

endpoints/Auth.py

@@ -48,8 +48,13 @@ def log_in(
     summary="Login with email and OTP token",
 )
 def send_magic_link(request: Request, login: Login):
-    magic_link = MagicalAuth().send_magic_link(
-        ip_address=request.client.host, login=login
+    auth = MagicalAuth()
+    data = request.json()
+    referrer = None
+    if "referrer" in data:
+        referrer = data["referrer"]
+    magic_link = auth.send_magic_link(
+        ip_address=request.client.host, login=login, referrer=referrer
     )
     return Detail(detail=magic_link)