1 files changed, 52 insertions, 0 deletions
diff --git a/user_messages/context_processors.py b/user_messages/context_processors.py
new file mode 100644
index 00000000..894f5801
--- /dev/null
+++ b/user_messages/context_processors.py
@@ -0,0 +1,52 @@
+"""
+Context processor for lightweight session messages.
+
+Time-stamp: <2008-07-19 23:16:19 carljm context_processors.py>
+
+"""
+from django.utils.encoding import StrAndUnicode
+
+from user_messages import get_and_delete_messages
+
+def user_messages (request):
+    """
+    Returns session messages for the current session.
+
+    """
+    messages = request.user.get_and_delete_messages()
+    #if request.user.is_authenticated():
+    #else:
+    #    messages = LazyMessages(request)
+    return { 'user_messages': messages }
+
+class LazyMessages (StrAndUnicode):
+    """
+    Lazy message container, so messages aren't actually retrieved from
+    session and deleted until the template asks for them.
+
+    """
+    def __init__(self, request):
+        self.request = request
+
+    def __iter__(self):
+        return iter(self.messages)
+
+    def __len__(self):
+        return len(self.messages)
+
+    def __nonzero__(self):
+        return bool(self.messages)
+
+    def __unicode__(self):
+        return unicode(self.messages)
+
+    def __getitem__(self, *args, **kwargs):
+        return self.messages.__getitem__(*args, **kwargs)
+
+    def _get_messages(self):
+        if hasattr(self, '_messages'):
+            return self._messages
+        self._messages = get_and_delete_messages(self.request)
+        return self._messages
+    messages = property(_get_messages)
+