dockered

django/profiles/viewsets/ProfileViewSet.py

@@ -0,0 +1,27 @@
+from django.contrib.auth.models import User
+from django.shortcuts import get_object_or_404
+
+from rest_framework import permissions
+from rest_framework import viewsets
+from rest_framework.response import Response
+
+from ..serializers import ProfileSerializer
+from ..models import ProfileModel
+
+
+class ProfileViewSet(viewsets.ModelViewSet):
+    queryset = ProfileModel.objects.all()
+    serializer_class = ProfileSerializer
+    permission_classes = (permissions.IsAuthenticatedOrReadOnly,)
+
+    def retrieve(self, request, username=None):
+        user = get_object_or_404(User, username=username)
+        return Response(self.serializer_class(user.profilemodel, context={'user': request.user}).data)
+
+    def retrieve_id(self, request, pk=None):
+        user = get_object_or_404(User, pk=pk)
+        return Response(self.serializer_class(user.profilemodel, context={'user': request.user}).data)
+
+    def list(self, request):
+        serializer = ProfileSerializer(self.get_queryset(), many=True, context={'user': request.user})
+        return Response(serializer.data)