from django.contrib import messages

from async_messages import get_message


class AsyncMiddleware(object):

    def process_request(self, request):
        # Check for message for this user and, if it exists,
        # call the messages API with it
        if not request.user.is_authenticated():
            return
        msg, level = get_message(request.user)
        if msg:
            messages.add_message(request, level, msg)

from django.contrib import messages

from async_messages import get_message


class AsyncMiddleware(object):

    def process_response(self, request, response):
        # Check for message for this user and, if it exists,
        # call the messages API with it
        if not request.user.is_authenticated():
            return
        msg, level = get_message(request.user)
        if msg:
            messages.add_message(request, level, msg)

Add the message during the processing of the response.
Add the message during the processing of the response.
  from django.contrib import messages
  
  from async_messages import get_message
  
  
  class AsyncMiddleware(object):
  
-     def process_request(self, request):
+     def process_response(self, request, response):
          # Check for message for this user and, if it exists,
          # call the messages API with it
          if not request.user.is_authenticated():
              return
          msg, level = get_message(request.user)
          if msg:
              messages.add_message(request, level, msg)
  
Add the message during the processing of the response.
## Code Before:
from django.contrib import messages

from async_messages import get_message


class AsyncMiddleware(object):

    def process_request(self, request):
        # Check for message for this user and, if it exists,
        # call the messages API with it
        if not request.user.is_authenticated():
            return
        msg, level = get_message(request.user)
        if msg:
            messages.add_message(request, level, msg)

## Instruction:
Add the message during the processing of the response.
## Code After:
from django.contrib import messages

from async_messages import get_message


class AsyncMiddleware(object):

    def process_response(self, request, response):
        # Check for message for this user and, if it exists,
        # call the messages API with it
        if not request.user.is_authenticated():
            return
        msg, level = get_message(request.user)
        if msg:
            messages.add_message(request, level, msg)
