django 2.0 compatibility in test project

miki725 · Jul 26, 2018 · 6bf1a7e · 6bf1a7e
1 parent 8769802
commit 6bf1a7e
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 6 deletions.
diff --git a/test_project/middleware.py b/test_project/middleware.py
@@ -9,9 +9,12 @@
 
 
 class SQLAlchemySessionMiddleware(object):
-    def process_request(self, request):
-        request.alchemy_session = Session()
+    def __init__(self, get_response):
+        self.get_response = get_response
 
-    def process_response(self, request, response):
-        request.alchemy_session.close()
-        return response
+    def __call__(self, request):
+        request.alchemy_session = Session()
+        try:
+            return self.get_response(request)
+        finally:
+            request.alchemy_session.close()
diff --git a/test_project/settings.py b/test_project/settings.py
@@ -31,7 +31,7 @@
 STATIC_URL = '/static/'
 SECRET_KEY = 'foo'
 
-MIDDLEWARE_CLASSES = [
+MIDDLEWARE = [
     'test_project.middleware.SQLAlchemySessionMiddleware',
 ]