feat(Besoins): Admin : Connaitre le nombre de besoins déposés par cha…

…que entreprise (#1120)

lemarche/companies/admin.py

@@ -41,12 +41,22 @@ def queryset(self, request, queryset):
 
 @admin.register(Company, site=admin_site)
 class CompanyAdmin(admin.ModelAdmin, DynamicArrayMixin):
-    list_display = ["id", "name", "user_count_annotated_with_link", "created_at"]
+    list_display = [
+        "id",
+        "name",
+        "user_count_annotated_with_link",
+        "user_tender_count_annotated_with_link",
+        "created_at",
+    ]
     list_filter = [HasUserFilter, HasEmailDomainFilter]
     search_fields = ["id", "name"]
     search_help_text = "Cherche sur les champs : ID, Nom"
 
-    readonly_fields = ["logo_url_display", "user_count_annotated_with_link", "created_at", "updated_at"]
+    readonly_fields = [
+        "logo_url_display",
+        "user_count_annotated_with_link",
+        "user_tender_count_annotated_with_link",
+    ] + Company.READONLY_FIELDS
 
     fieldsets = (
         (
@@ -57,6 +67,7 @@ class CompanyAdmin(admin.ModelAdmin, DynamicArrayMixin):
         ),
         ("Logo", {"fields": ("logo_url", "logo_url_display")}),
         ("Utilisateurs", {"fields": ("user_count_annotated_with_link",)}),
+        ("Besoins", {"fields": ("user_tender_count_annotated_with_link",)}),
         ("Dates", {"fields": ("created_at", "updated_at")}),
     )
 
@@ -93,5 +104,12 @@ def user_count_annotated_with_link(self, company):
         url = reverse("admin:users_user_changelist") + f"?company__id__exact={company.id}"
         return format_html(f'<a href="{url}">{company.user_count_annotated}</a>')
 
-    user_count_annotated_with_link.short_description = "Nombre d'utilisateurs rattachés"
+    user_count_annotated_with_link.short_description = "Utilisateurs rattachés"
     user_count_annotated_with_link.admin_order_field = "user_count_annotated"
+
+    def user_tender_count_annotated_with_link(self, company):
+        url = reverse("admin:tenders_tender_changelist") + f"?author__company_id__exact={company.id}"
+        return format_html(f'<a href="{url}">{company.user_tender_count_annotated}</a>')
+
+    user_tender_count_annotated_with_link.short_description = "Besoins déposés par les utilisateurs"
+    user_tender_count_annotated_with_link.admin_order_field = "user_tender_count_annotated"

lemarche/companies/models.py

@@ -13,10 +13,16 @@ def has_email_domain(self):
         return self.exclude(email_domain_list=[])
 
     def with_user_stats(self):
-        return self.annotate(user_count_annotated=Count("users", distinct=True))
+        return self.annotate(user_count_annotated=Count("users", distinct=True)).annotate(
+            user_tender_count_annotated=Count("users__tenders", distinct=True)
+        )
 
 
 class Company(models.Model):
+    FIELDS_STATS_COUNT = ["user_count"]
+    FIELDS_STATS_TIMESTAMPS = ["created_at", "updated_at"]
+    READONLY_FIELDS = FIELDS_STATS_COUNT + FIELDS_STATS_TIMESTAMPS
+
     name = models.CharField(verbose_name="Nom", max_length=255)
     slug = models.SlugField(verbose_name="Slug", max_length=255, unique=True)
     description = models.TextField(verbose_name="Description", blank=True)

lemarche/companies/tests.py

@@ -2,6 +2,7 @@
 
 from lemarche.companies.factories import CompanyFactory
 from lemarche.companies.models import Company
+from lemarche.tenders.factories import TenderFactory
 from lemarche.users.factories import UserFactory
 
 
@@ -22,10 +23,16 @@ class CompanyQuerysetTest(TestCase):
     def setUpTestData(cls):
         cls.user_1 = UserFactory()
         cls.user_2 = UserFactory()
+        TenderFactory(author=cls.user_1)
+        TenderFactory(author=cls.user_1)
         cls.company_with_users = CompanyFactory(users=[cls.user_1, cls.user_2])
         cls.company = CompanyFactory()
 
     def test_with_user_stats(self):
         company_queryset = Company.objects.with_user_stats()
+        # user_count
         self.assertEqual(company_queryset.get(id=self.company.id).user_count_annotated, 0)
         self.assertEqual(company_queryset.get(id=self.company_with_users.id).user_count_annotated, 2)
+        # user_tender_count
+        self.assertEqual(company_queryset.get(id=self.company.id).user_tender_count_annotated, 0)
+        self.assertEqual(company_queryset.get(id=self.company_with_users.id).user_tender_count_annotated, 2)