änderung1

2019-10-29 14:29:36 +01:00
1 parent cc1d096bc3
commit cbdda613b7
@@ -0,0 +1,4 @@
+from django.contrib.messages.api import *  # NOQA
+from django.contrib.messages.constants import *  # NOQA
+
+default_app_config = 'django.contrib.messages.apps.MessagesConfig'
@@ -0,0 +1,7 @@
+from django.apps import AppConfig
+from django.utils.translation import gettext_lazy as _
+
+
+class MessagesConfig(AppConfig):
+    name = 'django.contrib.messages'
+    verbose_name = _("Messages")
@@ -0,0 +1,26 @@
+from django.conf import settings
+from django.contrib.messages.storage import default_storage
+from django.utils.deprecation import MiddlewareMixin
+
+
+class MessageMiddleware(MiddlewareMixin):
+    """
+    Middleware that handles temporary messages.
+    """
+
+    def process_request(self, request):
+        request._messages = default_storage(request)
+
+    def process_response(self, request, response):
+        """
+        Update the storage backend (i.e., save the messages).
+
+        Raise ValueError if not all messages could be stored and DEBUG is True.
+        """
+        # A higher middleware layer may return a request which does not contain
+        # messages storage, so make no assumption that it will be there.
+        if hasattr(request, '_messages'):
+            unstored_messages = request._messages.update(response)
+            if unstored_messages and settings.DEBUG:
+                raise ValueError('Not all temporary messages could be stored.')
+        return response
@@ -0,0 +1,12 @@
+from django.conf import settings
+from django.utils.module_loading import import_string
+
+
+def default_storage(request):
+    """
+    Callable with the same interface as the storage classes.
+
+    This isn't just default_storage = import_string(settings.MESSAGE_STORAGE)
+    to avoid accessing the settings at the module level.
+    """
+    return import_string(settings.MESSAGE_STORAGE)(request)
@@ -0,0 +1,48 @@
+import json
+
+from django.conf import settings
+from django.contrib.messages.storage.base import BaseStorage
+from django.contrib.messages.storage.cookie import (
+    MessageDecoder, MessageEncoder,
+)
+
+
+class SessionStorage(BaseStorage):
+    """
+    Store messages in the session (that is, django.contrib.sessions).
+    """
+    session_key = '_messages'
+
+    def __init__(self, request, *args, **kwargs):
+        assert hasattr(request, 'session'), "The session-based temporary "\
+            "message storage requires session middleware to be installed, "\
+            "and come before the message middleware in the "\
+            "MIDDLEWARE%s list." % ("_CLASSES" if settings.MIDDLEWARE is None else "")
+        super().__init__(request, *args, **kwargs)
+
+    def _get(self, *args, **kwargs):
+        """
+        Retrieve a list of messages from the request's session. This storage
+        always stores everything it is given, so return True for the
+        all_retrieved flag.
+        """
+        return self.deserialize_messages(self.request.session.get(self.session_key)), True
+
+    def _store(self, messages, response, *args, **kwargs):
+        """
+        Store a list of messages to the request's session.
+        """
+        if messages:
+            self.request.session[self.session_key] = self.serialize_messages(messages)
+        else:
+            self.request.session.pop(self.session_key, None)
+        return []
+
+    def serialize_messages(self, messages):
+        encoder = MessageEncoder(separators=(',', ':'))
+        return encoder.encode(messages)
+
+    def deserialize_messages(self, data):
+        if data and isinstance(data, str):
+            return json.loads(data, cls=MessageDecoder)
+        return data
@@ -0,0 +1,12 @@
+from django.conf import settings
+from django.contrib.messages import constants
+
+
+def get_level_tags():
+    """
+    Return the message level tags.
+    """
+    return {
+        **constants.DEFAULT_TAGS,
+        **getattr(settings, 'MESSAGE_TAGS', {}),
+    }