django 消息框架 message使用详解

前言

在网页应用中，我们经常需要在处理完表单或其它类型的用户输入后，显示一个通知信息给用户。

对于这个需求，Django提供了基于Cookie或者会话的消息框架messages，无论是匿名用户还是认证的用户。这个消息框架允许你临时将消息存储在请求中，并在接下来的请求（通常就是下一个请求）中提取它们并显示。每个消息都带有一个特定的level标签，表示其优先级（例如info、 warning或error）。

一、启用消息框架

Django的messages消息框架的实现，依赖messages中间件和对应的context processor。

通过django-admin startproject xxx命令创建工程时，已经默认在settings.py中开启了消息框架功能需要的所有的设置：

• INSTALLED_APPS中注册的'django.contrib.messages'。
• MIDDLEWARE中添加'django.contrib.sessions.middleware.SessionMiddleware'和'django.contrib.messages.middleware.MessageMiddleware'。Django的messages框架默认使用的存储后端为sessions。所以Session中间件必须被启用，并出现在Message中间件之前。
• TEMPLATES设置中的DjangoTemplates选项包含的'context_processors'配置项要包含'django.contrib.messages.context_processors.messages'。

二、配置消息引擎

通常我们使用默认的就好，可以跳过这节，但如果真有需要，也可以配置：

1. 存储后端

Django提供了三种内置的消息存储后端：

class storage.session.SessionStorage class storage.cookie.CookieStorage class storage.fallback.FallbackStorage

FallbackStorage是默认的存储后端。如果它不适合你的需要，你可以通过设置MESSAGE_STORAGE选择另外一个存储后端，例如：

MESSAGE_STORAGE = 'django.contrib.messages.storage.cookie.CookieStorage'

2. 消息级别

消息框架的级别是可配置的，与Python的logging模块类似

Django内置的message级别有下面几种：

3. 消息样式

通常，我们在前端HTML页面中，希望给不同级别的消息，增加不同的CSS样式，比如警告为黄色，error为红色等等。

Django为我们提供了一个默认的样式对应关系：

也就是说SUCCESS级别的消息，在前端会被赋予一个success样式class。

若要修改消息级别的默认样式，设置MESSAGE_TAGS，按如下例子所示：。

from django.contrib.messages import constants as messages
MESSAGE_TAGS = {
  messages.INFO: '',
  50: 'critical',
}

三、使用消息框架

1. 添加消息

方法原型：add_message(request, level, message, extra_tags='', fail_silently=False)[source]

新增一条消息：

from django.contrib import messages
messages.add_message(request, messages.INFO, 'Hello world.')

提供请求对象request（直接用就行），消息级别、消息内容字符串三个参数即可。

或者使用下面的快捷方式

messages.debug(request, '%s SQL statements were executed.' % count)
messages.info(request, 'Three credits remain in your account.')
messages.success(request, 'Profile details updated.')
messages.warning(request, 'Your account expires in three days.')
messages.error(request, 'Document deleted.')

2. 显示消息

方法原型：get_messages(request)[source]

在你的模板文件中，像下面这样使用：

{% if messages %}
<ul class="messages">
  {% for message in messages %}
  <li{% if message.tags %} class="{{ message.tags }}"{% endif %}>{{ message }}</li>
  {% endfor %}
</ul>
{% endif %}

相关说明：

• 通过if判断是否有消息；
• messages是一个列表，必须用for标签循环它；
• 即使你知道只有一条消息，也要迭代messages列表，否则下个请求中，上个请求的消息不会被清除。
• 可以通过message.tags拿到每个消息的CSS样式

有一个DEFAULT_MESSAGE_LEVELS变量，它映射消息级别的名称到它们的数值：

{% if messages %}
<ul class="messages">
  {% for message in messages %}
  <li{% if message.tags %} class="{{ message.tags }}"{% endif %}>
    {% if message.level == DEFAULT_MESSAGE_LEVELS.ERROR %}Important: {% endif %}
    {{ message }}
  </li>
  {% endfor %}
</ul>
{% endif %}

说明：

• 可以通过message.level拿到当前消息的级别数值；
• 将它与DEFAULT_MESSAGE_LEVELS.ERROR进行对比；
• 如果一样，就说明当前消息级别为ERROR，需要显示到页面上。

在模板的外面，比如视图中，可以使用get_messages()方法获取消息：

from django.contrib.messages import get_messages

storage = get_messages(request)
for message in storage:
  do_something_with_the_message(message)

说明：

• get_messages()返回的是存储后端的一个实例。
• 循环这个实例，可以获得每条消息

对于每一个消息实例，都包含下面的属性，可以在模版或视图中调用：

• message: 消息的实际内容文本。不要使用message.message，直接message。
• level: 消息级别，一个整数。
• tags: 一个字符串，由该消息的所有标签(extra_tags和tags)组合而成，组合时用空格分割开这些标签。
• extra_tags: 一个字符串，由该消息的定制标签组合而成，并用空格分割。默认为空。
• level_tag: 当前消息级别对应的CSS字符串，前面介绍过。

3. 自定义消息级别

消息级别只是一个整数常量，所以，可以定义自己的级别常量，例如：

CRITICAL = 50

def my_view(request):
  messages.add_message(request, CRITICAL, 'A serious error occurred.')

在自定义消息级别时，应小心避免覆盖现有级别。内置级别的值为：

如果你需要在HTML或CSS中使用自定义级别，则需要通过MESSAGE_TAGS设置提供相应的映射关系。

4. 自定义每个请求的最小记录级别

每个请求都可以通过set_level()方法设置最小记录级别，如下所示:

from django.contrib import messages

# 修改最小级别为DEBUG
messages.set_level(request, messages.DEBUG)
messages.debug(request, 'Test message...')

# 在另外一个视图中修改最小级别为WARNING
messages.set_level(request, messages.WARNING)
messages.success(request, 'Your profile was updated.') # 被忽略，不记录
messages.warning(request, 'Your account is about to expire.') # 记录

# 将最小级别恢复到默认值
messages.set_level(request, None)
set_level()方法接收request为第一参数，消息级别为第二参数。

类似的，当前有效的记录级别可以用get_level()方法获取:

from django.contrib import messages
current_level = messages.get_level(request)

5. 添加额外的消息CSS样式

要添加自定义的消息CSS样式，可以通过extra_tags参数：

messages.add_message(request, messages.INFO, 'Over 9000!', extra_tags='dragonball')
messages.error(request, 'Email box full', extra_tags='email')

四、消息过期机制

默认情况下，如果包含消息的迭代器完成迭代后，当前请求中的消息都将被删除。

如果你不想这么做，想保留这些消息，那么需要显式的指定used参数为False，如下所示：

storage = messages.get_messages(request)
for message in storage:
  do_something_with(message)
storage.used = False