Select the election

elections/mixins.py

@@ -4,7 +4,19 @@ from django.views.generic.detail import SingleObjectMixin
 from .models import Election, Option, Question
 
 
-class RestrictAccessMixin:
+class SelectElectionMixin:
+    """Sélectionne automatiquement les foreignkeys voulues"""
+
+    def get_queryset(self):
+        qs = super().get_queryset()
+        if self.model is Question:
+            return qs.select_related("election")
+        elif self.model is Option:
+            return qs.select_related("question__election")
+        return qs
+
+
+class RestrictAccessMixin(SelectElectionMixin):
     """Permet de restreindre l'accès à des élections/questions/options"""
 
     f_prefixes = {Election: "", Question: "election__", Option: "question__election__"}