Middleware: add check if request has user attribute (e.g. WSGIRequest…

… doesn’t)
codeinthehole · Feb 26, 2014 · 95c449f · 95c449f
1 parent 440f6a7
commit 95c449f
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/async_messages/middleware.py b/async_messages/middleware.py
@@ -6,9 +6,11 @@
 class AsyncMiddleware(object):
 
     def process_response(self, request, response):
-        # Check for messages for this user and, if it exists,
-        # call the messages API with it
-        if hasattr(request, "session") and request.user.is_authenticated():
+        """
+        Check for messages for this user and, if it exists,
+        call the messages API with it
+        """
+        if hasattr(request, "session") and hasattr(request, "user") and request.user.is_authenticated():
             msgs = get_messages(request.user)
             if msgs:
                 for msg, level in msgs: