Show only correctors in Script corrector field in admin

diff --git a/teleforma/exam/admin.py b/teleforma/exam/admin.py
index 66ebd56d843a190658c09d8ec13be1d8ea5d540e..36f007a6b2e3853474d8facc245b6cbbf9295165 100644 (file)
--- a/teleforma/exam/admin.py
+++ b/teleforma/exam/admin.py
@@ -3,6 +3,8 @@ import os
 
 from django.contrib import admin
 from django.template.defaultfilters import filesizeformat
+from django.contrib.auth.models import User
+from django.db.models import Q
 
 from ..exam.models import Quota, Script, ScriptPage, ScriptType
 
@@ -44,6 +46,11 @@ class ScriptAdmin(admin.ModelAdmin):
     class Media:
         js = ("exam/js/admin.js",)
 
+
+    def render_change_form(self, request, context, *args, **kwargs):
+         context['adminform'].form.fields['corrector'].queryset = User.objects.filter(is_active=True).filter(Q(corrector__isnull=False) | Q(is_superuser=True))
+         return super(ScriptAdmin, self).render_change_form(request, context, *args, **kwargs)
+
     def author_name(self, instance):
         return instance.author.username