From 538917340d9339880c4b532afea7a518b512f3bb Mon Sep 17 00:00:00 2001
From: Hironsan <hiroki.nakayama.py@gmail.com>
Date: Wed, 27 Jun 2018 06:21:48 +0900
Subject: [PATCH] Add API

---
 doccano/app/app/settings.py                   |   2 +-
 doccano/app/db.sqlite3                        | Bin 196608 -> 196608 bytes
 doccano/app/server/templates/base.html        |   2 +-
 .../{project_list.html => projects.html}      |   0
 doccano/app/server/urls.py                    |  13 +-
 doccano/app/server/views.py                   | 114 ++++++++----------
 6 files changed, 60 insertions(+), 71 deletions(-)
 rename doccano/app/server/templates/{project_list.html => projects.html} (100%)

diff --git a/doccano/app/app/settings.py b/doccano/app/app/settings.py
index a9fbcfd7..f5532545 100644
--- a/doccano/app/app/settings.py
+++ b/doccano/app/app/settings.py
@@ -114,7 +114,7 @@ REST_FRAMEWORK = {
         'rest_framework.permissions.DjangoModelPermissionsOrAnonReadOnly'
     ],
     'DEFAULT_PAGINATION_CLASS': 'rest_framework.pagination.LimitOffsetPagination',
-    'PAGE_SIZE': 2,
+    'PAGE_SIZE': 10,
     'DEFAULT_FILTER_BACKENDS': ('django_filters.rest_framework.DjangoFilterBackend',),
     'SEARCH_PARAM': 'q',
 }
diff --git a/doccano/app/db.sqlite3 b/doccano/app/db.sqlite3
index fb73d843beb6aecad6f026d624b9e1584c71d5df..848dfb1738d562206d1319d9e569106cb682a096 100644
GIT binary patch
delta 254
zcmZo@;Av>!nIO&RJyFJ))tf<2;>5<3ocX+#RwkxahNgPv=7we_&3EUw-<{96{qB4w
zbp;M4-WCSFL%c1U6%{IYHJYrMSbbYq9St)xGD^z~b4)UeD$=t{DvOHEN|OysOpA=N
zO_C}t)3?7m$Hc)XhA`L6(%7)cnh~sd`h)vS3Lq6yj2uk-+Zp(`^WWJln6R2(RfXA?
zk&w|$)9>78(g12a%)tMM|1ePFc7Ay|W=BRN149d419M$N0|f(Ai<!1Rde0;$000v#
BPIv$S

delta 117
zcmV-*0E+*BfDC|u43HZENRb>v1xNrb{c^EnY@ZD>Ix#UiG%zkXH#IppgWaEp-Jb!s
z-Jb#{AO-;r!T=7!vmr3J4hVuf0R>Hj1wohK-vS`F>dpch0R{mN+yD>Uvk_p@50~8E
X0w@Ln58?n1;<FK8*blej?*bkW134(8

diff --git a/doccano/app/server/templates/base.html b/doccano/app/server/templates/base.html
index 7b26985b..3ac066d4 100644
--- a/doccano/app/server/templates/base.html
+++ b/doccano/app/server/templates/base.html
@@ -25,7 +25,7 @@
         <div class="container">
             <div class="navbar-brand">
                 <a class="navbar-item" href="{% url 'index' %}">
-                    <img src="{% static 'images/logo.png' %}" width="40" height="48">doccano
+                    <img src="{% static 'images/logo.png' %}" width="40" height="48">
                 </a>
             </div>
             <div id="topNav" class="navbar-menu">
diff --git a/doccano/app/server/templates/project_list.html b/doccano/app/server/templates/projects.html
similarity index 100%
rename from doccano/app/server/templates/project_list.html
rename to doccano/app/server/templates/projects.html
diff --git a/doccano/app/server/urls.py b/doccano/app/server/urls.py
index 6ddf6fa4..116b7cea 100644
--- a/doccano/app/server/urls.py
+++ b/doccano/app/server/urls.py
@@ -2,21 +2,22 @@ from django.urls import path
 
 from .views import IndexView
 from .views import AnnotationAPIView, ProgressAPI, SearchAPI, InboxView
-from .views import ProjectListView, ProjectAdminView, RawDataAPI, LabelAPI, DataDownloadAPI
+from .views import ProjectsView, ProjectAdminView, RawDataAPI, LabelAPI, DataDownloadAPI
 from rest_framework import routers
-from .views import LabelViewSet, ProjectViewSet, DocumentViewSet
+from .views import ProjectViewSet
+from .views import ProjectLabelsAPI, ProjectLabelAPI, ProjectDocsAPI
 
 
 router = routers.DefaultRouter()
-router.register(r'labels', LabelViewSet)
 router.register(r'projects', ProjectViewSet)
-router.register(r'documents', DocumentViewSet)
-#router.register(r'users', UserViewSet)
 
 
 urlpatterns = [
     path('', IndexView.as_view(), name='index'),
-    path('projects/', ProjectListView.as_view(), name='project-list'),
+    path('api/projects/<int:project_id>/labels/', ProjectLabelsAPI.as_view(), name='labels'),
+    path('api/projects/<int:project_id>/labels/<int:label_id>', ProjectLabelAPI.as_view(), name='label'),
+    path('api/projects/<int:project_id>/docs/', ProjectDocsAPI.as_view(), name='docs'),
+    path('projects/', ProjectsView.as_view(), name='project-list'),
     path('projects/<int:pk>/admin', ProjectAdminView.as_view(), name='project-admin'),
     path('projects/<int:project_id>/download', DataDownloadAPI.as_view(), name='download'),
     path('projects/<int:project_id>/', InboxView.as_view(), name='annotation'),
diff --git a/doccano/app/server/views.py b/doccano/app/server/views.py
index 5c3860f6..a6c58237 100644
--- a/doccano/app/server/views.py
+++ b/doccano/app/server/views.py
@@ -2,32 +2,39 @@ import json
 
 from django_filters.rest_framework import DjangoFilterBackend
 from django.http import JsonResponse, HttpResponse
-from django.shortcuts import render
+from django.shortcuts import render, get_object_or_404
 from django.views import View
+from django.views.generic import TemplateView
 from django.views.generic.list import ListView
 from django.views.generic.detail import DetailView
+from django.contrib.auth.mixins import LoginRequiredMixin
 from django.core.paginator import Paginator
-from rest_framework import viewsets, filters
+from rest_framework import viewsets, filters, generics
 from rest_framework.decorators import action
 from rest_framework.response import Response
-
+from rest_framework.permissions import IsAdminUser
 
 from .models import Annotation, Label, Document, Project
 from .serializers import LabelSerializer, ProjectSerializer, DocumentSerializer
 
 
-class IndexView(View):
+class IndexView(TemplateView):
     template_name = 'index.html'
 
-    def get(self, request, *args, **kwargs):
-        return render(request, self.template_name)
-
 
-class InboxView(View):
+class InboxView(LoginRequiredMixin, TemplateView):
     template_name = 'annotation.html'
 
-    def get(self, request, *args, **kwargs):
-        return render(request, self.template_name)
+
+class ProjectsView(LoginRequiredMixin, ListView):
+    model = Project
+    paginate_by = 100
+    template_name = 'projects.html'
+
+
+class ProjectAdminView(LoginRequiredMixin, DetailView):
+    model = Project
+    template_name = 'project_admin.html'
 
 
 class AnnotationAPIView(View):
@@ -171,27 +178,6 @@ class RawDataAPI(View):
         return JsonResponse({'status': 'ok'})
 
 
-class ProjectListView(ListView):
-
-    model = Project
-    paginate_by = 100  # if pagination is desired
-    template_name = 'project_list.html'
-
-    def get_context_data(self, **kwargs):
-        context = super().get_context_data(**kwargs)
-        return context
-
-
-class ProjectAdminView(DetailView):
-
-    model = Project
-    template_name = 'project_admin.html'
-
-    def get_context_data(self, **kwargs):
-        context = super().get_context_data(**kwargs)
-        return context
-
-
 class DataDownloadAPI(View):
 
     def get(self, request, *args, **kwargs):
@@ -207,18 +193,6 @@ class ProjectViewSet(viewsets.ModelViewSet):
     queryset = Project.objects.all()
     serializer_class = ProjectSerializer
 
-    @action(methods=['get'], detail=True)
-    def labels(self, request, pk=None):
-        project = self.get_object()
-        res = {label.id: label.text for label in project.labels.all()}
-        return Response(res)
-
-    @action(methods=['get'], detail=True)
-    def docs(self, request, pk=None):
-        project = self.get_object()
-        res = [doc.as_dict() for doc in project.documents.all()]
-        return Response(res)
-
     @action(methods=['get'], detail=True)
     def progress(self, request, pk=None):
         project = self.get_object()
@@ -226,30 +200,44 @@ class ProjectViewSet(viewsets.ModelViewSet):
         remaining = docs.filter(labels__isnull=True).count()
         return Response({'total': docs.count(), 'remaining': remaining})
 
-    @action(methods=['post'], detail=True)
-    def upload_doc(self, request, pk=None):
-        project = self.get_object()
-        f = request.FILES['file']
-        content = ''.join(chunk.decode('utf-8') for chunk in f.chunks())
-        for line in content.split('\n'):
-            j = json.loads(line)
-            Document(project=project, text=j['text']).save()
 
-    @action(methods=['get'], detail=True)
-    def download_doc(self, request, pk=None):
-        project = self.get_object()
-        res = []
-        return res
+class ProjectLabelsAPI(generics.ListCreateAPIView):
+    queryset = Label.objects.all()
+    serializer_class = LabelSerializer
+    pagination_class = None
 
+    def get_queryset(self):
+        project_id = self.kwargs['project_id']
+        queryset = self.queryset.filter(project=project_id)
 
-class DocumentViewSet(viewsets.ModelViewSet):
-    queryset = Document.objects.all()
-    serializer_class = DocumentSerializer
-    filter_backends = (DjangoFilterBackend, filters.SearchFilter)
-    search_fields = ('text', )
+        return queryset
 
 
-class LabelViewSet(viewsets.ModelViewSet):
+class ProjectLabelAPI(generics.RetrieveUpdateDestroyAPIView):
     queryset = Label.objects.all()
     serializer_class = LabelSerializer
-    filter_fields = ('project',)
+
+    def get_queryset(self):
+        project_id = self.kwargs['project_id']
+        queryset = self.queryset.filter(project=project_id)
+
+        return queryset
+
+    def get_object(self):
+        label_id = self.kwargs['label_id']
+        queryset = self.filter_queryset(self.get_queryset())
+        obj = get_object_or_404(queryset, pk=label_id)
+        self.check_object_permissions(self.request, obj)
+
+        return obj
+
+
+class ProjectDocsAPI(generics.ListCreateAPIView):
+    queryset = Document.objects.all()
+    serializer_class = DocumentSerializer
+
+    def get_queryset(self):
+        project_id = self.kwargs['project_id']
+        queryset = self.queryset.filter(project=project_id)
+
+        return queryset