Admin: Archaeological context records: modify display and label ordering for Unit and IdentificationType

1 files changed, 15 insertions, 1 deletions

diff --git a/archaeological_context_records/admin.py b/archaeological_context_records/admin.py
index bd2579afb..55511a4d5 100644
--- a/archaeological_context_records/admin.py
+++ b/archaeological_context_records/admin.py
@@ -65,7 +65,21 @@ class RelationTypeAdmin(admin.ModelAdmin):
 
 admin.site.register(models.RelationType, RelationTypeAdmin)
 
+class UnitAdmin(admin.ModelAdmin):
+    list_display = ['label', 'txt_idx', 'parent', 'available', 'order', 'comment']
+
+admin.site.register(models.Unit, UnitAdmin)
+
+ 
+class IdentificationTypeAdmin(admin.ModelAdmin):
+    list_display = ['label', 'txt_idx', 'available', 'order', 'comment']
+
+admin.site.register(models.IdentificationType, IdentificationTypeAdmin)
+
+
 general_models = [models.DatingType, models.DatingQuality,
-                  models.Unit, models.ActivityType, models.IdentificationType]
+                  models.ActivityType]
 for model in general_models:
     admin.site.register(model, GeneralTypeAdmin)
+
+