Merge pull request #151 from chakki-works/feature-me

Feature: Implement me API

app/server/api.py

@@ -14,13 +14,21 @@ from rest_framework_csv.renderers import CSVRenderer
 from .filters import DocumentFilter
 from .models import Project, Label, Document
 from .permissions import IsAdminUserAndWriteOnly, IsProjectUser, IsOwnAnnotation
-from .serializers import ProjectSerializer, LabelSerializer, DocumentSerializer
+from .serializers import ProjectSerializer, LabelSerializer, DocumentSerializer, UserSerializer
 from .serializers import ProjectPolymorphicSerializer
 from .utils import CSVParser, JSONParser, PlainTextParser, CoNLLParser
 from .utils import JSONLRenderer
 from .utils import JSONPainter, CSVPainter
 
 
+class Me(APIView):
+    permission_classes = (IsAuthenticated,)
+
+    def get(self, request, *args, **kwargs):
+        serializer = UserSerializer(request.user, context={'request': request})
+        return Response(serializer.data)
+
+
 class ProjectList(generics.ListCreateAPIView):
     queryset = Project.objects.all()
     serializer_class = ProjectPolymorphicSerializer

app/server/api_urls.py

@@ -1,6 +1,7 @@
 from django.urls import path
 from rest_framework.urlpatterns import format_suffix_patterns
 
+from .api import Me
 from .api import ProjectList, ProjectDetail
 from .api import LabelList, LabelDetail
 from .api import DocumentList, DocumentDetail
@@ -10,6 +11,7 @@ from .api import StatisticsAPI
 
 
 urlpatterns = [
+    path('me', Me.as_view(), name='me'),
     path('projects', ProjectList.as_view(), name='project_list'),
     path('projects/<int:project_id>', ProjectDetail.as_view(), name='project_detail'),
     path('projects/<int:project_id>/statistics',

app/server/serializers.py

@@ -1,3 +1,4 @@
+from django.contrib.auth import get_user_model
 from rest_framework import serializers
 from rest_polymorphic.serializers import PolymorphicSerializer
 from rest_framework.exceptions import ValidationError
@@ -8,6 +9,13 @@ from .models import TextClassificationProject, SequenceLabelingProject, Seq2seqP
 from .models import DocumentAnnotation, SequenceAnnotation, Seq2seqAnnotation
 
 
+class UserSerializer(serializers.ModelSerializer):
+
+    class Meta:
+        model = get_user_model()
+        fields = ('id', 'username', 'first_name', 'last_name', 'email', 'is_superuser')
+
+
 class LabelSerializer(serializers.ModelSerializer):
 
     def validate(self, attrs):