Django 中间件 process_template_response 设置

Question

我想创建一个函数来检查用户是否有任何通知。如果是，则该数字应显示在导航栏中。

有人可以帮我重构一下吗？

谢谢！

middleware.py：

def process_template_response(self, request, response):
    if request.user.is_authenticated():
        try:
            notifications = Notification.objects.all_for_user(request.user).unread()
            count = notifications.count()
            context = {
                "count": count
            }
            response = TemplateResponse(request, 'navbar.html', context)
            return response
        except:
            pass
    else:
        pass

navbar.html：

<li >
    <a href="{% url 'notifications_all' %}">
        {% if count > 0 %}Notifications ({{ count }}){% else %}Notifications{% endif %}
    </a>
</li>

Answer 1

我以前做过类似的工作，我认为你应该使用context_data响应的属性：

class NotificationMiddleware(object):
    def process_template_response(self, request, response):
        if request.user.is_authenticated():
            try:
                notifications = Notification.objects.all_for_user(request.user).unread()
                count = notifications.count()
                response.context_data['count'] = count  # I recomend you to use other name instead of 'count'
                return response
            except Exception, e:
                print e  # Fix possible errors
                return response
        else:
            return response

然后你需要在settings 文件中的MIDDLEWARE_CLASSES 元组中注册这个类：

# settings.py
MIDDLEWARE_CLASSES = (
    # Django middlewares
    ....
    'yourapp.middleware.NotificationMiddleware',
)

上面的示例假设您的应用程序中有一个名为“yourapp”的middleware 文件夹。

最后你应该可以在模板中使用{{ count }}了。