diff --git a/contact_us/context_processors.py b/contact_us/context_processors.py index 31027db..b55b91c 100644 --- a/contact_us/context_processors.py +++ b/contact_us/context_processors.py @@ -1,5 +1,17 @@ +from django.contrib.messages import get_messages + from contact_us.forms import ContactUsForm def contact_us_form(request): - return {'contact_us_form': ContactUsForm(next=request.resolver_match.view_name)} + storage = get_messages(request) + message = None + # if len(storage._loaded_messages) > 0: + for item in storage: + if "contact_us_form_message" in item.extra_tags: + message = item + + return { + 'contact_us_form': ContactUsForm(next=request.resolver_match.view_name), + 'contact_us_form_message': message + }