NDOJ/judge/views/notification.py

from django.contrib.auth.decorators import login_required
from django.views.generic import ListView
from django.utils.translation import ugettext as _
from django.utils.timezone import now
from django.http import Http404

from judge.models import Profile, Notification, NotificationProfile
from judge.models.notification import unseen_notifications_count
from judge.utils.infinite_paginator import InfinitePaginationMixin

__all__ = ["NotificationList"]


class NotificationList(InfinitePaginationMixin, ListView):
    model = Notification
    context_object_name = "notifications"
    template_name = "notification/list.html"
    paginate_by = 50

    def get_queryset(self):
        self.unseen_cnt = unseen_notifications_count(self.request.profile)

        self.queryset = Notification.objects.filter(
            owner=self.request.profile
        ).order_by("-id")

        return self.queryset

    def get_context_data(self, **kwargs):
        context = super().get_context_data(**kwargs)
        context["unseen_count"] = self.unseen_cnt
        context["title"] = _("Notifications (%d unseen)") % context["unseen_count"]
        context["first_page_href"] = "."
        return context

    def get(self, request, *args, **kwargs):
        ret = super().get(request, *args, **kwargs)
        if not request.user.is_authenticated:
            raise Http404()
        NotificationProfile.objects.filter(user=request.profile).update(unread_count=0)
        unseen_notifications_count.dirty(self.request.profile)
        return ret