8555630df7
* Upgrade edx-submissions * Upgrade edx-ora2 * Upgrade edx-val * Upgrade edx-proctoring * Update all edx-platform code that depends on DRF, including: - auth_exchange - cors_csrf - embargo - enrollment - util - commerce - course_structure - discussion_api - mobile_api - notifier_api - teams - credit - profile_images - user_api - lib/api (OAuth2 and pagination)
51 lines
1.5 KiB
Python
51 lines
1.5 KiB
Python
from django.contrib.auth.models import User
|
|
from rest_framework.viewsets import ReadOnlyModelViewSet
|
|
from rest_framework.response import Response
|
|
from rest_framework import pagination
|
|
|
|
from notification_prefs import NOTIFICATION_PREF_KEY
|
|
from notifier_api.serializers import NotifierUserSerializer
|
|
from openedx.core.lib.api.permissions import ApiKeyHeaderPermission
|
|
|
|
|
|
class NotifierPaginator(pagination.PageNumberPagination):
|
|
"""
|
|
Paginator for the notifier API.
|
|
"""
|
|
page_size = 10
|
|
page_size_query_param = "page_size"
|
|
|
|
def get_paginated_response(self, data):
|
|
"""
|
|
Construct a response with pagination information.
|
|
"""
|
|
return Response({
|
|
'next': self.get_next_link(),
|
|
'previous': self.get_previous_link(),
|
|
'count': self.page.paginator.count,
|
|
'results': data
|
|
})
|
|
|
|
|
|
class NotifierUsersViewSet(ReadOnlyModelViewSet):
|
|
"""
|
|
An endpoint that the notifier can use to retrieve users who have enabled
|
|
daily forum digests, including all information that the notifier needs about
|
|
such users.
|
|
"""
|
|
permission_classes = (ApiKeyHeaderPermission,)
|
|
serializer_class = NotifierUserSerializer
|
|
pagination_class = NotifierPaginator
|
|
|
|
# See NotifierUserSerializer for notes about related tables
|
|
queryset = User.objects.filter(
|
|
preferences__key=NOTIFICATION_PREF_KEY
|
|
).select_related(
|
|
"profile"
|
|
).prefetch_related(
|
|
"preferences",
|
|
"courseenrollment_set",
|
|
"course_groups",
|
|
"roles__permissions"
|
|
)
|