From 5807858ef26aa90a9c212fda5c5b5272d767baf3 Mon Sep 17 00:00:00 2001
From: Hironsan <hiroki.nakayama.py@gmail.com>
Date: Sat, 18 Aug 2018 11:52:35 +0900
Subject: [PATCH] Add document serializer for sequence labeling and seq2seq

---
 app/db.sqlite3            | Bin 393216 -> 393216 bytes
 app/server/api.py         |   9 ++++++---
 app/server/models.py      |  13 +++++++++++++
 app/server/serializers.py |  39 +++++++++++++++++++++++++++++++++-----
 4 files changed, 53 insertions(+), 8 deletions(-)

diff --git a/app/db.sqlite3 b/app/db.sqlite3
index 59e5e7791be831183d05b58655ea3bea080c23a3..3ce397758fea0a053a0f0ea2dd4f70ce638a0e45 100644
GIT binary patch
delta 695
zcmaixPe>GD7{<T%`)1skZFk?X0>%EBHlV26%<P|;RRp&nHpnh@EnIcCO>|d{A}eq+
zqf1~~@Lhy03_69yQWGHtDJVi#Xc8r55uLnLP*5G3wGK&#4)5uCpWpYq&zH@XX0xSN
zw!rA{*1a%#x24|?%GDMBRFPUDumJE+U9c%XvAZKvY?d2=`jAy`7=F_XX%Xv!nd4>-
zo-h?Fx$8RYkt^{w-ohJLo!{B?+*c$r0JqAk$KkcJ6XUKsu%BWqU)=rGCkjjk)oy+S
zw!&0F;;v_oC`fr-xzD^ARVLU78IYs;K|c8wN_g)ZXxOO#4iTw0EPo(inL*9&TZGFs
zqGy_Trti7wLyzZt2}q=|ZCf;2_s2Snv_^`bXipY~Pav(RSO+pfp`d;!(Q`79Zc7}9
zw<bD|r;pi3z4Q!FO!HShdd))!8F8rIb<`7^C7dDn9lzize!yjXf%7=y+&7LcQ4xp?
z)Sll#!%dQevjl(OtnF^06X+@uWB{u0rx_Y@v;~4calv+X(-~YVM&U2}X{|$FDk$IZ
zGrn`^tN0O@9Qq`#RZA6nd5Bi6_(XY*_#S%)J^MCD@|@G)1G#FTK@^(@#i(v|q=WHf
zZ)a;zGt9oo>4+IlwTJssy{A&~R7c#_&N2bY0{Y*-YlTd0pg~vz@SY)7Uf5a`d?LmO
z#&85}?80RyNf^~#tQr3Ih{(H!7<cSRg70w(Z{kHvI`+d@xt&G9&~&R#v+Dj0Ar@p_
TdybV|8p|<f1bZwe9%=an42{I#

delta 390
zcmWNM+be@%9LJy2ysz!ue#3=@aIr$xOm90N8cL)Ue*mrAsWBHW9Lnpk+^AVokKDW~
z?$OdCtrW!#Ls3gBhvdo)E|f!gF1|OPZ+*Uf4Z~#^uJvlXnX2i<oBPoPFNXA^_q4=a
z$*7L`dkj}(b1mbOWxje}pxqy8_v_w3TSyQ3mV+h_n)s5sp@}OQ+#?1vxZp?)scFxH
zmFv6sMphB7?BRiIS^Uo-BF9)bshiDk<yx{<XA=`Mw|K4DVjo2?abP7!S~~395~Fr9
ziwQ#f-a$6hMZhZR9JBKajr8#xkp6N7@(8}+6W-tjif{*cxo`*~Uq=Gr5(Tah{0J>d
za*!G-<yVu$FJS?}uh8P;TtbBx7KlfBiW0&Xe87|RyuqvVoPj7NN#mo9*;M2>v9D--
zHZgpS1<i(?J&L;2!7Py#OE;fN(t6>WqqmGPh#_G$Xcx!>4Lp{lTw{<x@C>K00|}{}
X23AYEG55^Vs?}|i$}q8QT1KM(5gdIF

diff --git a/app/server/api.py b/app/server/api.py
index 17b8ca54..0117121f 100644
--- a/app/server/api.py
+++ b/app/server/api.py
@@ -89,13 +89,16 @@ class LabelDetail(generics.RetrieveUpdateDestroyAPIView):
 
 class DocumentList(generics.ListCreateAPIView):
     queryset = Document.objects.all()
-    from .serializers import SequenceDocumentSerializer
-    # serializer_class = DocumentSerializer
-    serializer_class = SequenceDocumentSerializer
     filter_backends = (DjangoFilterBackend, filters.SearchFilter, filters.OrderingFilter)
     search_fields = ('text', )
     permission_classes = (IsAuthenticated, IsProjectUser, IsAdminUserAndWriteOnly)
 
+    def get_serializer_class(self):
+        project = get_object_or_404(Project, pk=self.kwargs['project_id'])
+        self.serializer_class = project.get_document_serializer()
+
+        return self.serializer_class
+
     def get_queryset(self):
         queryset = self.queryset.filter(project=self.kwargs['project_id'])
         if not self.request.query_params.get('is_checked'):
diff --git a/app/server/models.py b/app/server/models.py
index 2515932b..11caa6a8 100644
--- a/app/server/models.py
+++ b/app/server/models.py
@@ -82,6 +82,19 @@ class Project(models.Model):
 
         return docs
 
+    def get_document_serializer(self):
+        from .serializers import ClassificationDocumentSerializer
+        from .serializers import SequenceDocumentSerializer
+        from .serializers import Seq2seqDocumentSerializer
+        if self.is_type_of(Project.DOCUMENT_CLASSIFICATION):
+            return ClassificationDocumentSerializer
+        elif self.is_type_of(Project.SEQUENCE_LABELING):
+            return SequenceDocumentSerializer
+        elif self.is_type_of(Project.Seq2seq):
+            return Seq2seqDocumentSerializer
+        else:
+            raise ValueError('Invalid project_type')
+
     def get_annotation_serializer(self):
         from .serializers import DocumentAnnotationSerializer
         from .serializers import SequenceAnnotationSerializer
diff --git a/app/server/serializers.py b/app/server/serializers.py
index 6e50f392..0d6b0e9b 100644
--- a/app/server/serializers.py
+++ b/app/server/serializers.py
@@ -67,17 +67,46 @@ class Seq2seqAnnotationSerializer(serializers.ModelSerializer):
         fields = ('id', 'text')
 
 
+class ClassificationDocumentSerializer(serializers.ModelSerializer):
+    annotations = serializers.SerializerMethodField()
+
+    def get_annotations(self, instance):
+        request = self.context.get('request')
+        if request:
+            annotations = instance.doc_annotations.filter(user=request.user)
+            serializer = DocumentAnnotationSerializer(annotations, many=True)
+            return serializer.data
+
+    class Meta:
+        model = Document
+        fields = ('id', 'text', 'annotations')
+
+
 class SequenceDocumentSerializer(serializers.ModelSerializer):
-    labels = SequenceAnnotationSerializer(source='seq_annotations', many=True)
     annotations = serializers.SerializerMethodField()
 
-    def get_annotations(self, obj):
+    def get_annotations(self, instance):
+        request = self.context.get('request')
+        if request:
+            annotations = instance.seq_annotations.filter(user=request.user)
+            serializer = SequenceAnnotationSerializer(annotations, many=True)
+            return serializer.data
+
+    class Meta:
+        model = Document
+        fields = ('id', 'text', 'annotations')
+
+
+class Seq2seqDocumentSerializer(serializers.ModelSerializer):
+    annotations = serializers.SerializerMethodField()
+
+    def get_annotations(self, instance):
         request = self.context.get('request')
         if request:
-            annotations = obj.seq_annotations.filter(user=request.user)
-            serializer = SequenceAnnotationSerializer(annotations.all(), many=True)
+            annotations = instance.seq2seq_annotations.filter(user=request.user)
+            serializer = Seq2seqAnnotationSerializer(annotations, many=True)
             return serializer.data
 
     class Meta:
         model = Document
-        fields = ('id', 'text', 'labels', 'annotations')
+        fields = ('id', 'text', 'annotations')