+2

upvcarshare/config/router.py

reviewed

··· 6 6 7 7 from journeys.api.v1.resources import TransportResource, ResidenceResource, CampusResource, JourneyResource, \ 8 8 join_journey, leave_journey, recommended_journeys, cancel_journey 9 9 + from notifications.api.v1.resources import NotificationResource 9 10 from users.api.v1.resources import me 10 11 11 12 router = SimpleRouter() ··· 14 15 router.register(r'residences', viewset=ResidenceResource) 15 16 router.register(r'campus', viewset=CampusResource) 16 17 router.register(r'journeys', viewset=JourneyResource) 18 18 + router.register(r'notifications', viewset=NotificationResource) 17 19 18 20 urlpatterns = [ 19 21 url(r'^journeys/recommended/$', recommended_journeys, kwargs={'pk': None}, name='all-recommended-journeys'),

+2

upvcarshare/notifications/api/__init__.py

reviewed

··· 1 1 + # -*- coding: utf-8 -*- 2 2 + from __future__ import unicode_literals, print_function, absolute_import

+2

upvcarshare/notifications/api/v1/__init__.py

reviewed

··· 1 1 + # -*- coding: utf-8 -*- 2 2 + from __future__ import unicode_literals, print_function, absolute_import

+22

upvcarshare/notifications/api/v1/resources.py

reviewed

··· 1 1 + # -*- coding: utf-8 -*- 2 2 + from __future__ import unicode_literals, print_function, absolute_import 3 3 + 4 4 + from rest_framework import viewsets 5 5 + from rest_framework.permissions import IsAuthenticated 6 6 + 7 7 + from notifications.api.v1.serializers import NotificationSerializer 8 8 + from notifications.models import Notification 9 9 + 10 10 + 11 11 + class NotificationResource(viewsets.ReadOnlyModelViewSet): 12 12 + 13 13 + queryset = Notification.objects.filter(read=False) 14 14 + serializer_class = NotificationSerializer 15 15 + permission_classes = [ 16 16 + IsAuthenticated, 17 17 + ] 18 18 + 19 19 + def get_queryset(self): 20 20 + queryset = super(NotificationResource, self).get_queryset() 21 21 + queryset = queryset.filter(user=self.request.user) 22 22 + return queryset

+52

upvcarshare/notifications/api/v1/serializers.py

reviewed

··· 1 1 + # -*- coding: utf-8 -*- 2 2 + from __future__ import unicode_literals, print_function, absolute_import 3 3 + 4 4 + from rest_framework import serializers 5 5 + 6 6 + from journeys.api.v1.serializers import JourneySerializer 7 7 + from journeys.models import Journey 8 8 + from notifications.models import Notification 9 9 + from users.api.v1.serializers import UserSerializer 10 10 + from users.models import User 11 11 + 12 12 + 13 13 + class BaseObjectRelatedField(serializers.RelatedField): 14 14 + """A custom field to use for the `actor`, `target` or `action` generic relationship.""" 15 15 + 16 16 + def to_internal_value(self, data): 17 17 + pass 18 18 + 19 19 + def to_representation(self, value): 20 20 + """Serialize tagged objects to a simple textual representation.""" 21 21 + if isinstance(value, User): 22 22 + serializer = UserSerializer(value) 23 23 + elif isinstance(value, Journey): 24 24 + serializer = JourneySerializer(value) 25 25 + else: 26 26 + raise Exception('Unexpected type of tagged object') 27 27 + return serializer.data 28 28 + 29 29 + 30 30 + class ActorObjectRelatedField(BaseObjectRelatedField): 31 31 + pass 32 32 + 33 33 + 34 34 + class TargetObjectRelatedField(BaseObjectRelatedField): 35 35 + pass 36 36 + 37 37 + 38 38 + class ActionObjectRelatedField(BaseObjectRelatedField): 39 39 + pass 40 40 + 41 41 + 42 42 + class NotificationSerializer(serializers.ModelSerializer): 43 43 + """Serializes a notification.""" 44 44 + 45 45 + text = serializers.CharField() 46 46 + actor = ActionObjectRelatedField(read_only=True) 47 47 + target = TargetObjectRelatedField(read_only=True) 48 48 + action = ActionObjectRelatedField(read_only=True) 49 49 + 50 50 + class Meta: 51 51 + model = Notification 52 52 + fields = ["actor", "verb", "action", "target", "text", "read", "created"]

+2

upvcarshare/notifications/tests/test_api.py

reviewed

··· 1 1 + # -*- coding: utf-8 -*- 2 2 + from __future__ import unicode_literals, print_function, absolute_import

+2

upvcarshare/notifications/tests/test_models.py

reviewed

··· 1 1 + # -*- coding: utf-8 -*- 2 2 + from __future__ import unicode_literals, print_function, absolute_import