Merge pull request doccano#239 from guillim/guillim-admin-columns

we add some columns to the admin page

Author: Hironsan

app/server/admin.py

@@ -4,12 +4,49 @@
 from .models import DocumentAnnotation, SequenceAnnotation, Seq2seqAnnotation
 from .models import TextClassificationProject, SequenceLabelingProject, Seq2seqProject
 
-admin.site.register(DocumentAnnotation)
-admin.site.register(SequenceAnnotation)
-admin.site.register(Seq2seqAnnotation)
-admin.site.register(Label)
-admin.site.register(Document)
-admin.site.register(Project)
-admin.site.register(TextClassificationProject)
-admin.site.register(SequenceLabelingProject)
-admin.site.register(Seq2seqProject)
+
+class LabelAdmin(admin.ModelAdmin):
+    list_display = ('text', 'project', 'text_color', 'background_color')
+    ordering = ('project',)
+    search_fields = ('project',)
+
+
+class DocumentAdmin(admin.ModelAdmin):
+    list_display = ('text', 'project', 'meta')
+    ordering = ('project',)
+    search_fields = ('project',)
+
+
+class ProjectAdmin(admin.ModelAdmin):
+    list_display = ('name', 'description', 'project_type', 'randomize_document_order')
+    ordering = ('project_type',)
+    search_fields = ('name',)
+
+
+class SequenceAnnotationAdmin(admin.ModelAdmin):
+    list_display = ('document', 'label', 'start_offset', 'user')
+    ordering = ('document',)
+    search_fields = ('document',)
+
+
+class DocumentAnnotationAdmin(admin.ModelAdmin):
+    list_display = ('document', 'label', 'user')
+    ordering = ('document',)
+    search_fields = ('document',)
+
+
+class Seq2seqAnnotationAdmin(admin.ModelAdmin):
+    list_display = ('document', 'text', 'user')
+    ordering = ('document',)
+    search_fields = ('document',)
+
+
+admin.site.register(DocumentAnnotation, DocumentAnnotationAdmin)
+admin.site.register(SequenceAnnotation, SequenceAnnotationAdmin)
+admin.site.register(Seq2seqAnnotation, Seq2seqAnnotationAdmin)
+admin.site.register(Label, LabelAdmin)
+admin.site.register(Document, DocumentAdmin)
+admin.site.register(Project, ProjectAdmin)
+admin.site.register(TextClassificationProject, ProjectAdmin)
+admin.site.register(SequenceLabelingProject, ProjectAdmin)
+admin.site.register(Seq2seqProject, ProjectAdmin)