From 31e8ad8a3e8788a5eceb67db04ca7d4a2ab320ed Mon Sep 17 00:00:00 2001
From: Thomas Girod <thgirod@hotmail.com>
Date: Sat, 17 Dec 2022 16:31:12 +0100
Subject: [PATCH] redirect directly on counter if user is barman

---
 core/templates/core/base.jinja | 34 ++++++++++++++++++++--------------
 counter/models.py              | 30 ++++++++++++++++++++++++++++++
 counter/tests.py               | 14 ++++++++++++++
 3 files changed, 64 insertions(+), 14 deletions(-)

diff --git a/core/templates/core/base.jinja b/core/templates/core/base.jinja
index 4ef8d911..e343f2a1 100644
--- a/core/templates/core/base.jinja
+++ b/core/templates/core/base.jinja
@@ -65,20 +65,26 @@
             <div id="header_bar">
                 <ul id="header_bars_infos">
                 {% cache 100 "counters_activity" %}
-                  {% for bar in Counter.objects.filter(type="BAR").all() %}
-                      <li>
-                      <a href="{{ url('counter:activity', counter_id=bar.id) }}" style="padding: 0px">
-                      {% if bar.is_inactive(): %}
-                          <i class="fa fa-question" style="color: #f39c12"></i>
-                      {% elif bar.is_open(): %}
-                          <i class="fa fa-check" style="color: #2ecc71"></i>
-                      {% else %}
-                          <i class="fa fa-times" style="color: #eb2f06"></i>
-                      {% endif %}
-                          {{ bar }}
-                      </a>
-                      </li>
-                  {% endfor %}
+                    {% for bar in Counter.objects.annotate_has_barman(user).filter(type="BAR") %}
+                        <li>
+                            {# If the user is a barman, we redirect him directly to the barman page
+                            else we redirect him to the activity page #}
+                            {% if bar.has_annotated_barman %}
+                                <a href="{{ url('counter:details', counter_id=bar.id) }}" style="padding: 0">
+                            {% else %}
+                                <a href="{{ url('counter:activity', counter_id=bar.id) }}" style="padding: 0">
+                            {% endif %}
+                                {% if bar.is_inactive(): %}
+                                  <i class="fa fa-question" style="color: #f39c12"></i>
+                                {% elif bar.is_open(): %}
+                                  <i class="fa fa-check" style="color: #2ecc71"></i>
+                                {% else %}
+                                  <i class="fa fa-times" style="color: #eb2f06"></i>
+                                {% endif %}
+                                {{ bar }}
+                            </a>
+                        </li>
+                    {% endfor %}
                   </ul>
                 {% endcache %}
                 <form action="{{ url('core:search') }}" method="GET" id="header_search">
diff --git a/counter/models.py b/counter/models.py
index 4f6f25ea..64da8a68 100644
--- a/counter/models.py
+++ b/counter/models.py
@@ -22,7 +22,9 @@
 #
 #
 
+from __future__ import annotations
 from django.db import models
+from django.db.models import OuterRef, Exists, QuerySet
 from django.db.models.functions import Length
 from django.utils.translation import gettext_lazy as _
 from django.utils import timezone
@@ -309,6 +311,32 @@ class Product(models.Model):
         return "%s (%s)" % (self.name, self.code)
 
 
+class CounterQuerySet(models.QuerySet):
+    def annotate_has_barman(self, user: User) -> CounterQuerySet:
+        """
+        Annotate the queryset with the `user_is_barman` field.
+        For each counter, this field has value True if the user
+        is a barman of this counter, else False.
+
+        :param user: the user we want to check if he is a barman
+
+        Example::
+
+            sli = User.objects.get(username="sli")
+            counters = (
+                Counter.objects
+                .annotate_has_barman(sli)  # add the user_has_barman boolean field
+                .filter(has_annotated_barman=True)  # keep only counters where this user is barman
+            )
+            print("Sli est barman dans les comptoirs suivants :")
+            for counter in counters:
+                print(f"- {counter.name}")
+        """
+        subquery = user.counters.filter(pk=OuterRef("pk"))
+        # noinspection PyTypeChecker
+        return self.annotate(has_annotated_barman=Exists(subquery))
+
+
 class Counter(models.Model):
     name = models.CharField(_("name"), max_length=30)
     club = models.ForeignKey(
@@ -333,6 +361,8 @@ class Counter(models.Model):
     )
     token = models.CharField(_("token"), max_length=30, null=True, blank=True)
 
+    objects = CounterQuerySet.as_manager()
+
     class Meta:
         verbose_name = _("counter")
 
diff --git a/counter/tests.py b/counter/tests.py
index 0a25a9ff..9049e55e 100644
--- a/counter/tests.py
+++ b/counter/tests.py
@@ -138,6 +138,20 @@ class CounterTest(TestCase):
         )
         self.assertTrue(response.status_code == 200)
 
+    def test_annotate_has_barman_queryset(self):
+        """
+        Test if the custom queryset method ``annotate_has_barman``
+        works as intended
+        """
+        self.sli.counters.clear()
+        self.sli.counters.add(self.foyer, self.mde)
+        counters = Counter.objects.annotate_has_barman(self.sli)
+        for counter in counters:
+            if counter.name in ("Foyer", "MDE"):
+                self.assertTrue(counter.has_annotated_barman)
+            else:
+                self.assertFalse(counter.has_annotated_barman)
+
 
 class CounterStatsTest(TestCase):
     def setUp(self):