admin for documents

Author: alesanchezr

api/admin.py

@@ -67,6 +67,10 @@ class EmployeeDocumentAdmin(admin.ModelAdmin):
 
     def get_name(self, obj):
         return obj.document_type.title if obj.document_type is not None else 'Missing document type'
+admin.site.register(EmployeeDocument, EmployeeDocumentAdmin)
+class DocumentAdmin(admin.ModelAdmin):
+    list_display = ('id', 'title', 'validates_identity', 'validates_employment', 'is_form')
+admin.site.register(Document, DocumentAdmin)
 
 
 
@@ -77,7 +81,6 @@ def get_name(self, obj):
 admin.site.register(Notification)
 admin.site.register(JobCoreInvite)
 admin.site.register(BankAccount)
-admin.site.register(EmployeeDocument, EmployeeDocumentAdmin)
 
 class AppVersionAdmin(admin.ModelAdmin):
     list_display = ('id', 'version', 'force_update', 'created_at', 'updated_at')

api/models.py

@@ -682,6 +682,9 @@ class Document(models.Model):
     created_at = models.DateTimeField(auto_now_add=True, editable=False)
     updated_at = models.DateTimeField(auto_now=True, editable=False)
 
+    def __str__(self):
+        return self.title
+
 
 class EmployeeDocument(models.Model):
     PENDING = 'PENDING'