from rest_framework import viewsets, permissions, status
from .serializers import *

class UserViewSet(viewsets.ReadOnlyModelViewSet):
    serializer_class = UserSerializer
    permission_classes = [permissions.IsAuthenticated]

    def get_queryset(self):
        if self.request.user.is_superuser:
            obj = get_user_model().objects.all()
        else:
            # This is a bit stupid: we have a user, we create a queryset by
            # matching on the username. But I don't know a "nicer" way.
            # Nico, 2020-03-18
            obj = get_user_model().objects.filter(username=self.request.user.username)

        return obj