Fix two additional redirects missed in the previous commit

web2py · Sep 24, 2024 · 58eef71 · 58eef71
1 parent 3172ff1
commit 58eef71
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/py4web/utils/auth.py b/py4web/utils/auth.py
@@ -1644,7 +1644,7 @@ def login(self, model=False):
                 ):
                     self.auth.session["auth.2fa_user"] = user["id"]
                     self.auth.session["auth.2fa_next_url"] = next_url
-                    redirect(URL("auth", "two_factor"))
+                    redirect(URL(f"{self.auth.route}/two_factor"))
             self.auth.store_user_in_session(user["id"])
             self._postprocessing("login", form, user)
 
@@ -1730,7 +1730,7 @@ def two_factor(self):
                 self._set_flash(
                     self.auth.param.messages["errors"]["two_factor_max_tries"]
                 )
-                redirect(URL("auth", "login", vars=dict(next=next_url)))
+                redirect(URL(f"{self.auth.route}/login", vars=dict(next=next_url)))
         return form
 
     def request_reset_password(self, model=False):