From 5479dc57dd71a81c1404d5c60ac71fb8db9eb709 Mon Sep 17 00:00:00 2001
From: Max Yakovenko <mxyakovenko9@gmail.com>
Date: Wed, 1 Aug 2018 08:21:49 +0300
Subject: [PATCH] add logic of finding a contact us message in context
 processor

---
 contact_us/context_processors.py | 14 +++++++++++++-
 1 file changed, 13 insertions(+), 1 deletion(-)

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
+    }