Merge pull request #665 from olegpshenichniy/BK-2175

BK-2175 Change import in SecurityMiddleware
booktype · Mar 2, 2017 · 77314d5 · 77314d5
2 parents 2a1666b + 695def4
commit 77314d5
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/lib/booktype/apps/core/middleware.py b/lib/booktype/apps/core/middleware.py
@@ -1,4 +1,5 @@
-import rest_framework
+from rest_framework.response import Response as RestResponse
+
 from django.contrib.auth import logout
 
 from booktype.utils.security import Security, get_security
@@ -8,7 +9,7 @@ class SecurityMiddleware(object):
 
     def process_template_response(self, request, response):
 
-        if isinstance(response, rest_framework.response.Response):
+        if isinstance(response, RestResponse):
             return response
 
         sec = response.context_data.get('security', None)