Show only teachers in Substituion forms

aleksis/apps/chronos/forms.py

@@ -4,6 +4,7 @@ from django_select2.forms import ModelSelect2MultipleWidget, ModelSelect2Widget
 from material import Layout
 
 from .models import AutomaticPlan, LessonSubstitution, SupervisionSubstitution
+from .util.chronos_helpers import get_teachers
 
 
 class LessonSubstitutionForm(forms.ModelForm):
@@ -23,6 +24,11 @@ class LessonSubstitutionForm(forms.ModelForm):
             ),
         }
 
+    def __init__(self, request, *args, **kwargs):
+        super().__init__(*args, **kwargs)
+        self.request = request
+        self.fields["teachers"].queryset = get_teachers(request.user)
+
 
 class SupervisionSubstitutionForm(forms.ModelForm):
     """Form to manage supervisions substitutions."""
@@ -41,6 +47,11 @@ class SupervisionSubstitutionForm(forms.ModelForm):
             ),
         }
 
+    def __init__(self, request, *args, **kwargs):
+        super().__init__(*args, **kwargs)
+        self.request = request
+        self.fields["teacher"].queryset = get_teachers(request.user)
+
 
 class AutomaticPlanForm(forms.ModelForm):
     layout = Layout("slug", "name", "number_of_days", "show_header_box")

aleksis/apps/chronos/views.py

@@ -276,10 +276,11 @@ def edit_substitution(request: HttpRequest, id_: int, week: int) -> HttpResponse
 
     if lesson_substitution:
         edit_substitution_form = LessonSubstitutionForm(
-            request.POST or None, instance=lesson_substitution
+            request, request.POST or None, instance=lesson_substitution
         )
     else:
         edit_substitution_form = LessonSubstitutionForm(
+            request,
             request.POST or None,
         )
 
@@ -410,10 +411,11 @@ def edit_supervision_substitution(request: HttpRequest, id_: int, week: int) ->
 
     if supervision_substitution:
         edit_supervision_substitution_form = SupervisionSubstitutionForm(
-            request.POST or None, instance=supervision_substitution
+            request, request.POST or None, instance=supervision_substitution
         )
     else:
         edit_supervision_substitution_form = SupervisionSubstitutionForm(
+            request,
             request.POST or None,
         )