From ef0f09ef1b2a44940d4a69029b5ce256e538b733 Mon Sep 17 00:00:00 2001
From: Étienne Loks <etienne.loks@iggdrasil.net>
Date: Fri, 6 Mar 2020 12:15:26 +0100
Subject: Quick actions - tables - edit finds: add appraisal date and treatment
 emergency fields

---
 archaeological_finds/forms.py | 13 ++++++++++---
 1 file changed, 10 insertions(+), 3 deletions(-)

(limited to 'archaeological_finds/forms.py')

diff --git a/archaeological_finds/forms.py b/archaeological_finds/forms.py
index bf5854cc6..316253a7a 100644
--- a/archaeological_finds/forms.py
+++ b/archaeological_finds/forms.py
@@ -569,12 +569,14 @@ class QAFindFormMulti(QAForm):
         'qa_checked_type': models.CheckedType,
         'qa_period': Period,
         "qa_conservatory_state": models.ConservatoryState,
+        "qa_treatment_emergency": models.TreatmentEmergencyType,
     }
 
     MULTI = True
     REPLACE_FIELDS = [
         'qa_denomination', 'qa_ue', 'qa_manufacturing_place', 'qa_checked_type',
-        'qa_check_date', 'qa_conservatory_state'
+        'qa_check_date', 'qa_conservatory_state', 'qa_treatment_emergency',
+        'qa_appraisal_date',
     ]
 
     HEADERS = {
@@ -629,8 +631,10 @@ class QAFindFormMulti(QAForm):
     qa_alteration_causes = widgets.Select2MultipleField(
         label=_(u"Alteration cause"), required=False
     )
-    qa_conservatory_state= forms.ChoiceField(label=_("Conservatory state"),
-                                             required=False, choices=[])
+    qa_conservatory_state = forms.ChoiceField(label=_("Conservatory state"),
+                                              required=False, choices=[])
+    qa_treatment_emergency = forms.ChoiceField(label=_("Treatment emergency"),
+                                               choices=[], required=False)
     qa_comment = forms.CharField(
         label=_(u"Comment"), required=False,
         widget=forms.Textarea)
@@ -638,6 +642,8 @@ class QAFindFormMulti(QAForm):
     qa_checked_type = forms.ChoiceField(label=_(u"Check"), required=False)
     qa_check_date = forms.DateField(
         label=_(u"Check date"), widget=DatePicker, required=False)
+    qa_appraisal_date = forms.DateField(
+        label=_("Appraisal date"), widget=DatePicker, required=False)
 
     qa_period = widgets.Select2MultipleField(
         label=_("Period"), choices=[], required=False)
@@ -655,6 +661,7 @@ class QAFindFormMulti(QAForm):
                   is_multiple=True),
         FieldType('qa_checked_type', models.CheckedType),
         FieldType('qa_conservatory_state', models.ConservatoryState),
+        FieldType('qa_treatment_emergency', models.TreatmentEmergencyType),
         FieldType('qa_period', Period, is_multiple=True),
     ]
 
-- 
cgit v1.2.3