diff --git a/reValuate/api/__pycache__/views.cpython-38.pyc b/reValuate/api/__pycache__/views.cpython-38.pyc index 5b4a555f..7dcd940e 100644 Binary files a/reValuate/api/__pycache__/views.cpython-38.pyc and b/reValuate/api/__pycache__/views.cpython-38.pyc differ diff --git a/reValuate/api/views.py b/reValuate/api/views.py index 1abd34cf..e85abf79 100644 --- a/reValuate/api/views.py +++ b/reValuate/api/views.py @@ -1,15 +1,23 @@ -from rest_framework import status, viewsets -from rest_framework.decorators import api_view +from rest_framework import status, viewsets, authentication, permissions +from rest_framework.decorators import api_view, action from rest_framework.response import Response + from users.models import Balance from upload.models import Media from .serializers import BalanceSerializer, ImageSerializer class BalanceView(viewsets.ModelViewSet): - queryset = Balance.objects.all() + permission_classes = [permissions.IsAuthenticated] serializer_class = BalanceSerializer -class ImageView(viewsets.ModelViewSet): - queryset = Media.objects.all() - serializer_class = ImageSerializer + def get_queryset(self): + return Balance.objects.filter(user=self.request.user) + +class ImageView(viewsets.ModelViewSet): + permission_classes = [permissions.IsAuthenticated] + serializer_class = ImageSerializer + + def get_queryset(self): + return Media.objects.filter(user=self.request.user) +