Make profile a modelviewset + add /me/ for current profile

api/views.py

@@ -1,11 +1,9 @@
-from django_filters.rest_framework import DjangoFilterBackend
-from rest_framework import viewsets, generics, filters
 from rest_framework.permissions import IsAuthenticated, IsAdminUser
-from rest_framework.response import Response
+from rest_framework import viewsets
 
 from .serializers import ContestSerializer, BandSerializer, FrequencySerializer, EntryCategorySerializer, \
     ReferenceSerializer, QSOSerializer, ShadowCallSerializer, UserSerializer
-from contest.models import Contest, Band, Frequency, EntryCategory, Reference, QSO, ShadowCall
+from contest.models import Contest, Band, Frequency, EntryCategory, Reference, QSO, ShadowCall, User
 
 
 class ContestViewSet(viewsets.ReadOnlyModelViewSet):
@@ -64,17 +62,26 @@ class QSOViewSet(viewsets.ModelViewSet):
         return serializer.save(owner=self.request.user)
 
 
-class UserProfileViewSet(generics.UpdateAPIView, viewsets.GenericViewSet):
+class UserProfileViewSet(viewsets.ModelViewSet):
+    """
+    Resource to view user-profiles, currently restricted to the current user's profile.
+    Use `me/` to get the profile of the currently logged in user.
+    """
     permission_classes = [IsAuthenticated]
     serializer_class = UserSerializer
 
-    def list(self, request, format=None):
-        user = request.user
-        serializer = UserSerializer(user)
-        return Response(serializer.data)
-
     def get_queryset(self):
-        return self.request.user
+        return User.objects.filter(id=self.request.user.id)
+
+    def get_object(self):
+        lookup_url_kwarg = self.lookup_url_kwarg or self.lookup_field
+        if self.kwargs.get(lookup_url_kwarg) == "me":
+            obj = self.request.user
+            self.check_object_permissions(self.request, obj)
+        else:
+            obj = super(ContestViewSet, self).get_object()
+
+        return obj
 
 
 class ShadowCallViewSet(viewsets.ReadOnlyModelViewSet):