Source code for ploneintranet.notifications.browser.notifications

# -*- coding: utf-8 -*-
from plone import api
from ploneintranet.layout.memoize.view import memoize
from ploneintranet.notifications.channel import AllChannel
from Products.Five.browser import BrowserView


class NotificationsView(BrowserView):
    @memoize
    def your_notifications(self):
        show_all = bool(self.request.form.get("showall"))
        display_message = []
        user = api.user.get_current()
        channel = AllChannel(user.getUserId())
        if show_all:
            display_message = channel.get_all_messages()
        else:
            display_message = channel.get_unread_messages()
        return display_message