From b564db3bb2dbbb71c0d1c1eb7f40ef44fa50e811 Mon Sep 17 00:00:00 2001
From: Étienne Loks <etienne.loks@iggdrasil.net>
Date: Thu, 30 Jun 2022 15:46:37 +0200
Subject: Preventive file: add price agreement

---
 archaeological_files/admin.py | 17 +++++++++++++++--
 1 file changed, 15 insertions(+), 2 deletions(-)

(limited to 'archaeological_files/admin.py')

diff --git a/archaeological_files/admin.py b/archaeological_files/admin.py
index 433b521bc..edc6be716 100644
--- a/archaeological_files/admin.py
+++ b/archaeological_files/admin.py
@@ -84,7 +84,7 @@ for model in general_models:
 
 
 class JobAdmin(GeneralTypeAdmin):
-    list_filter = ("available", "permanent_contract")
+    list_filter = ("available", "price_agreement")
     LIST_DISPLAY = [
         "label",
         "permanent_contract",
@@ -113,7 +113,7 @@ class EquipmentServiceCostAdmin(ImportActionAdmin):
         "equipment_service_type__label",
         "service_provider",
     )
-    list_filter = ("available",)
+    list_filter = ("available", "price_agreement")
     list_display = [
         "equipment_service_type",
         "specificity",
@@ -128,3 +128,16 @@ class EquipmentServiceCostAdmin(ImportActionAdmin):
 
 
 admin_site.register(models.EquipmentServiceCost, EquipmentServiceCostAdmin)
+
+
+class PriceAgreementAdmin(GeneralTypeAdmin):
+    list_filter = ("available",)
+    extra_list_display = [
+        "start_date",
+        "end_date",
+    ]
+    actions = []
+    model = models.PriceAgreement
+
+
+admin_site.register(models.PriceAgreement, PriceAgreementAdmin)
-- 
cgit v1.2.3