django 5.1: support LoginRequiredMiddleware

xi · Jun 18, 2024 · 6dcc378 · 6dcc378
1 parent 072e8db
commit 6dcc378
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 0 deletions.
diff --git a/mfa/views.py b/mfa/views.py
@@ -27,6 +27,12 @@
     def stronghold_public(view_func):
         return view_func
 
+try:
+    from django.contrib.auth.decorators import login_not_required
+except ImportError:
+    def login_not_required(view_func):
+        return view_func
+
 
 class LoginView(DjangoLoginView):
     def no_key_exists(self, form):
@@ -90,6 +96,7 @@ def form_valid(self, form):
         return super().form_valid(form)
 
 
+@method_decorator(login_not_required, name='dispatch')
 @method_decorator(stronghold_public, name='dispatch')
 class MFAAuthView(MFAFormView):
     form_class = MFAAuthForm

diff --git a/tests/settings.py b/tests/settings.py
@@ -21,6 +21,9 @@
     'mfa.middleware.MFAEnforceMiddleware',
 ]
 
+if django.VERSION >= (5, 1):
+    MIDDLEWARE.append('django.contrib.auth.middleware.LoginRequiredMiddleware')
+
 AUTHENTICATION_BACKENDS = [
     'django.contrib.auth.backends.ModelBackend',
 ]