12 lines
421 B
Python
12 lines
421 B
Python
from rest_framework import viewsets, permissions
|
|
from .models import Notification
|
|
from .serializers import NotificationSerializer
|
|
|
|
|
|
class NotificationViewSet(viewsets.ModelViewSet):
|
|
queryset = Notification.objects.all()
|
|
serializer_class = NotificationSerializer
|
|
permission_classes = [permissions.IsAuthenticated]
|
|
|
|
def get_queryset(self):
|
|
return Notification.objects.filter(user=self.request.user)
|