✨ treatment state refactoring -> distinct input status and treatment status (refs #6182) - treatments: search criteria

1 files changed, 8 insertions, 3 deletions

diff --git a/archaeological_finds/admin.py b/archaeological_finds/admin.py
index 5536901cf..39a8b16a4 100644
--- a/archaeological_finds/admin.py
+++ b/archaeological_finds/admin.py
@@ -101,7 +101,7 @@ admin_site.register(models.Property, PropertyAdmin)
 class TreatmentAdmin(HistorizedObjectAdmin):
     list_display = ('year', 'index', 'label', 'treatment_types_lbl', 'location',
                     'downstream_lbl', 'upstream_lbl', 'container', 'person')
-    list_filter = ('treatment_types', 'treatment_state', 'year')
+    list_filter = ('treatment_types', 'input_status', 'treatment_status', 'year')
     model = models.Treatment
     readonly_fields = HistorizedObjectAdmin.readonly_fields + [
         'cached_label', 'downstream_lbl', 'upstream_lbl',
@@ -232,11 +232,16 @@ class TreatmentFileTypeAdmin(GeneralTypeAdmin):
     extra_list_display = ["treatment_type"]
 
 
-@admin.register(models.TreatmentState, site=admin_site)
-class TreatmentStateAdmin(GeneralTypeAdmin):
+@admin.register(models.TreatmentInputStatus, site=admin_site)
+class TreatmentInputStatusAdmin(GeneralTypeAdmin):
     extra_list_display = ["order", "executed"]
 
 
+@admin.register(models.TreatmentStatus, site=admin_site)
+class TreatmentStatusAdmin(GeneralTypeAdmin):
+    extra_list_display = ["order"]
+
+
 @admin.register(models.Exhibition, site=admin_site)
 class ExhibitionAdmin(HistorizedObjectAdmin):
     list_display = ('name', 'year', 'reference', 'exhibition_type')