Display rounds for current machine if session in progress

gym/templates/gym/equipment.html

@@ -56,4 +56,21 @@
             {% endif %}
         </div>
     </div>
+    {% if session %}
+        <div class="row">
+            <div class="col-12">
+                <h2>Séries de la séance en cours</h2>
+                {% if rounds %}
+                    <ul class="list-group">
+                        {% for round in rounds %}
+                            <a href="#" class="list-group-item">{{ round.repetition_number }} répétitions
+                                &times; {{ round.chosen_weight }}</a>
+                        {% endfor %}
+                    </ul>
+                {% else %}
+                    <p>Pas de série pour cette machine aujourd'hui.</p>
+                {% endif %}
+            </div>
+        </div>
+    {% endif %}
 {% endblock %}

gym/views.py

@@ -40,10 +40,19 @@ class EquipmentDetailView(generic.DetailView):
     model = Equipment
     context_object_name = 'equipment'
     template_name = 'gym/equipment.html'
+    session = None
+
+    def dispatch(self, request, *args, **kwargs):
+        session_pk = self.request.session.get('session_pk')
+        if session_pk:
+            self.session = Session.objects.get(pk=session_pk)
+        return super().dispatch(request, *args, **kwargs)
 
     def get_context_data(self, **kwargs):
         context = super().get_context_data(**kwargs)
-        context['session'] = self.request.session.get('session_pk')
+        if self.session:
+            context['session'] = self.session.pk
+            context['rounds'] = Round.objects.filter(session=self.session, equipment=self.get_object())
         return context