moviewyrm/bookwyrm/views/updates.py
2021-12-09 18:16:34 -08:00

30 lines
861 B
Python

""" endpoints for getting updates about activity """
from django.contrib.auth.decorators import login_required
from django.http import Http404, JsonResponse
from bookwyrm import activitystreams
@login_required
def get_notification_count(request):
"""any notifications waiting?"""
return JsonResponse(
{
"count": request.user.unread_notification_count,
"has_mentions": request.user.has_unread_mentions,
}
)
@login_required
def get_unread_status_count(request, stream="home"):
"""any unread statuses for this feed?"""
stream = activitystreams.streams.get(stream)
if not stream:
raise Http404
return JsonResponse(
{
"count": stream.get_unread_count(request.user),
"count_by_type": stream.get_unread_count_by_status_type(request.user),
}
)