Preventive file: work on inlines

1 files changed, 22 insertions, 4 deletions

diff --git a/archaeological_files/admin.py b/archaeological_files/admin.py
index 9ba3d17d5..6ac40a69b 100644
--- a/archaeological_files/admin.py
+++ b/archaeological_files/admin.py
@@ -20,6 +20,7 @@
 from ajax_select import make_ajax_form
 
 from django.conf import settings
+from django.contrib import admin
 
 from ishtar_common.apps import admin_site
 from ishtar_common.admin import HistorizedObjectAdmin, GeneralTypeAdmin
@@ -65,13 +66,30 @@ admin_site.register(models.File, FileAdmin)
 
 
 general_models = [
-    models.FileType, models.PermitType, models.Job, models.GenericEquipmentType,
-    models.EquipmentType, models.TechnicalService,
+    models.FileType, models.PermitType, models.Job,
+    models.GenericEquipmentServiceType,
 ]
 if settings.COUNTRY == "fr":
     general_models.append(models.SaisineType)
 for model in general_models:
     admin_site.register(model, GeneralTypeAdmin)
 
-admin_site.register(models.EquipmentCost)
-admin_site.register(models.TechnicalServiceCost)
+
+class EquipmentServiceTypeAdmin(GeneralTypeAdmin):
+    list_filter = ("available", 'generic_equipment_type')
+
+
+admin_site.register(models.EquipmentServiceType, EquipmentServiceTypeAdmin)
+
+
+class EquipmentServiceCostAdmin(admin.ModelAdmin):
+    search_fields = (
+        "equipment_service_type__label",
+        "service_provider",
+    )
+    list_filter = ("available",)
+    list_display = ["equipment_service_type", "specificity", "unitary_cost", "unit",
+                    "flat_rate", "available"]
+
+
+admin_site.register(models.EquipmentServiceCost, EquipmentServiceCostAdmin)