Upgrading pre-Django 1.10-style middleware

nyaruka · Jun 12, 2017 · 9205f69 · 9205f69
1 parent dd64e31
commit 9205f69
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 3 deletions.
diff --git a/smartmin/middleware.py b/smartmin/middleware.py
@@ -8,8 +8,14 @@
 from django.utils import timezone
 from six.moves import StringIO
 
+try:
+    from django.utils.deprecation import MiddlewareMixin
+except ImportError:
+    class MiddlewareMixin(object):
+        pass
+
 
-class AjaxRedirect(object):
+class AjaxRedirect(MiddlewareMixin):
     def process_response(self, request, response):
         if request.is_ajax():
             if type(response) == HttpResponseRedirect:
@@ -21,7 +27,7 @@ def process_response(self, request, response):
         return response
 
 
-class ProfileMiddleware(object):
+class ProfileMiddleware(MiddlewareMixin):
     def __init__(self):
         pass
 

diff --git a/smartmin/users/middleware.py b/smartmin/users/middleware.py
@@ -7,8 +7,14 @@
 from django.http import HttpResponseRedirect
 from .models import PasswordHistory
 
+try:
+    from django.utils.deprecation import MiddlewareMixin
+except ImportError:
+    class MiddlewareMixin(object):
+        pass
 
-class ChangePasswordMiddleware:
+
+class ChangePasswordMiddleware(MiddlewareMixin):
     """
     Redirects all users to the password change form if we find that a user's
     password is expired.