Preventive file: work on inlines - 2

1 files changed, 7 insertions, 5 deletions

diff --git a/archaeological_files/admin.py b/archaeological_files/admin.py
index 6ac40a69b..629a0031d 100644
--- a/archaeological_files/admin.py
+++ b/archaeological_files/admin.py
@@ -20,10 +20,10 @@
 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
+from ishtar_common.admin import HistorizedObjectAdmin, GeneralTypeAdmin, \
+    export_as_csv_action, serialize_type_action, ImportActionAdmin
 
 from . import models
 
@@ -82,14 +82,16 @@ class EquipmentServiceTypeAdmin(GeneralTypeAdmin):
 admin_site.register(models.EquipmentServiceType, EquipmentServiceTypeAdmin)
 
 
-class EquipmentServiceCostAdmin(admin.ModelAdmin):
+class EquipmentServiceCostAdmin(ImportActionAdmin):
     search_fields = (
         "equipment_service_type__label",
         "service_provider",
     )
     list_filter = ("available",)
-    list_display = ["equipment_service_type", "specificity", "unitary_cost", "unit",
-                    "flat_rate", "available"]
+    list_display = ["equipment_service_type", "specificity", "parent",
+                    "unitary_cost", "unit", "flat_rate", "available"]
+    actions = [export_as_csv_action(), serialize_type_action]
+    model = models.EquipmentServiceCost
 
 
 admin_site.register(models.EquipmentServiceCost, EquipmentServiceCostAdmin)