Merge branch 'admin-types' into 'master'

Display only available element types in admin

See merge request !718

arkindex/documents/admin.py

@@ -58,6 +58,18 @@ class ElementAdmin(admin.ModelAdmin):
     search_fields = ('name', )
     inlines = (MetaDataInline, ClassificationInline)
 
+    def formfield_for_foreignkey(self, db_field, request, **kwargs):
+        if db_field.name == 'type':
+            # Only display types available on the element's corpus
+            # It would make no sense to pick a type from another corpus
+            # Display all elements on creation, as we do not know yet the corpus
+            element_id = request.resolver_match.kwargs.get('object_id')
+            if element_id:
+                element = self.get_object(request, element_id)
+                kwargs['queryset'] = ElementType.objects.filter(corpus=element.corpus).order_by('display_name')
+
+        return super().formfield_for_foreignkey(db_field, request, **kwargs)
+
 
 class TranscriptionAdmin(admin.ModelAdmin):
     list_display = ('id', 'text', 'score', 'element', )