Manage treatment files

1 files changed, 10 insertions, 1 deletions

diff --git a/archaeological_finds/admin.py b/archaeological_finds/admin.py
index 867245194..52939c03f 100644
--- a/archaeological_finds/admin.py
+++ b/archaeological_finds/admin.py
@@ -65,6 +65,14 @@ class TreatmentAdmin(HistorizedObjectAdmin):
 admin.site.register(models.Treatment, TreatmentAdmin)
 
 
+class TreatmentFileAdmin(HistorizedObjectAdmin):
+    list_display = ('year', 'index', 'name', 'internal_reference')
+    search_fields = ('cached_label',)
+    model = models.TreatmentFile
+
+admin.site.register(models.TreatmentFile, TreatmentFileAdmin)
+
+
 class TreatmentSourceAdmin(admin.ModelAdmin):
     list_display = ('treatment', 'title', 'source_type',)
     list_filter = ('source_type',)
@@ -89,6 +97,7 @@ admin.site.register(models.TreatmentType, TreatmentTypeAdmin)
 
 general_models = [
     models.ConservatoryState, models.RemarkabilityType,
-    models.PreservationType, models.IntegrityType, ]
+    models.PreservationType, models.IntegrityType,
+    models.TreatmentFileType]
 for model in general_models:
     admin.site.register(model, GeneralTypeAdmin)