improve trainee creation

generic/forms.py

@@ -68,6 +68,8 @@ from django.forms.utils import from_current_timezone, to_current_timezone
 from crispy_forms.helper import FormHelper
 from crispy_forms.layout import Submit
 
+from dal import autocomplete
+
 
 class SplitDateTimeFieldDefaultTime(SplitDateTimeField):
     def compress(self, data_list):
@@ -672,13 +674,23 @@ class UserUserForm(Form):
     first_name = CharField(max_length=255, required=True)
     last_name = CharField(max_length=255, required=True)
     grade = ModelChoiceField(UserGrades.objects.all(), required=False)
+
+class TraineeForm(Form):
+    username = EmailField(
+        required=True,
+        help_text="Username / email should be the same. Ideally this should be an nhs email.",
+    )
+    first_name = CharField(max_length=255, required=True)
+    last_name = CharField(max_length=255, required=True)
+    grade = ModelChoiceField(UserGrades.objects.all(), required=False)
     supervisor = ModelChoiceField(
-        Supervisor.objects.all(), required=False
+        Supervisor.objects.all(), required=False, widget=autocomplete.ModelSelect2(url='generic:supervisor-autocomplete')
     )  # Needs to be a user/object ref
-    #
-    # class Meta:
-    #
-    #     fields = ("username", "first_name", "last_name")
+
+    #class Meta:
+    #    widgets = {
+    #        "supervisor" : autocomplete.ModelSelect2(url='generic:supervisor-autocomplete')
+    #    }
 
 
 class SupervisorForm(ModelForm):