15 lines
520 B
Python
15 lines
520 B
Python
from rest_framework import viewsets, permissions
|
|
from .models import Subscription
|
|
from .serializers import SubscriptionSerializer
|
|
|
|
|
|
class SubscriptionViewSet(viewsets.ModelViewSet):
|
|
queryset = Subscription.objects.all()
|
|
serializer_class = SubscriptionSerializer
|
|
permission_classes = [permissions.IsAuthenticated]
|
|
|
|
def get_queryset(self):
|
|
user = self.request.user
|
|
if user.role == "admin":
|
|
return Subscription.objects.all()
|
|
return Subscription.objects.filter(user=user)
|