Preventive file: choose price agreement form

1 files changed, 11 insertions, 0 deletions

diff --git a/archaeological_files/forms.py b/archaeological_files/forms.py
index 46e0aae9c..1958cb2ae 100644
--- a/archaeological_files/forms.py
+++ b/archaeological_files/forms.py
@@ -669,6 +669,17 @@ class PreventiveFileForm(forms.ModelForm):
         return super().save(commit=commit)
 
 
+class PreventiveFileSimpleForm(forms.ModelForm):
+    price_agreement = forms.ModelChoiceField(
+        label=_("Price agreement"),
+        queryset=models.PriceAgreement.objects.filter(available=True)
+    )
+
+    class Meta:
+        model = models.File
+        fields = ["price_agreement"]
+
+
 class PreventiveFileGenJobForm(PreventiveFileForm):
     def __init__(self, *args, **kwargs):
         super(PreventiveFileGenJobForm, self).__init__(*args, **kwargs)