1 files changed, 16 insertions, 4 deletions
diff --git a/ctrack/caf/forms.py b/ctrack/caf/forms.py
index fb5e32a..0487682 100644
--- a/ctrack/caf/forms.py
+++ b/ctrack/caf/forms.py
@@ -12,19 +12,31 @@ from django.urls import reverse
 from django.forms import ModelChoiceField
 
 from ctrack.caf.models import CAF
-from ctrack.caf.models import DocumentFile
+from ctrack.caf.models import ApplicableSystem, DocumentFile
+from ctrack.organisations.models import Organisation
+from django.forms.models import ModelMultipleChoiceField
 
 
-class CAFForm(forms.ModelForm):
+class CAFCreateForm(forms.ModelForm):
     file = ModelChoiceField(
             queryset=DocumentFile.objects.all(),
             required=False,
             help_text="Please select an existing File. <a href='/caf/file/documentfile/create' target='_blank'>Create new File</a>" # TODO this URL does not exist
             )
+    owner = ModelChoiceField(
+        queryset=Organisation.objects.all(),
+        required=True,
+        help_text="Choose the Organisation that owns this CAF."
+    )
+    applicable_systems = ModelMultipleChoiceField(
+        queryset=ApplicableSystem.objects.all(),
+        required=True,
+        help_text="Choose the system."
+    )
 
     class Meta:
         model = CAF
-        fields = ["file"]
+        fields = ["quality_grading", "confidence_grading", "version", "file"]
 
     def __init__(self, *args, **kwargs):
         super().__init__(*args, **kwargs)
@@ -35,7 +47,7 @@ class CAFForm(forms.ModelForm):
         self.helper.form_class = "form-group"
         self.helper.form_method = "post"
         self.helper.layout = Layout(
-            Fieldset("Create/Edit CAF", "owner", "file"),
+            Fieldset("Create/Edit CAF", "owner", "applicable_systems", "quality_grading", "confidence_grading", "file", "version"),
             ButtonHolder(Submit("submit", "Submit"), Button("cancel", "Cancel")),
         )