ListViews et templates

apps/family/models.py

@@ -1,3 +1,6 @@
+# Copyright (C) 2018-2025 by BDE ENS Paris-Saclay
+# SPDX-License-Identifier: GPL-3.0-or-later
+
 from django.db import models, transaction
 from django.utils import timezone
 from django.contrib.auth.models import User
@@ -53,6 +56,7 @@ class FamilyMembership(models.Model):
     )
 
     class Meta:
+        unique_together = ('user', 'year',)
         verbose_name = _('family membership')
         verbose_name_plural = _('family memberships')
 
@@ -96,6 +100,10 @@ class Challenge(models.Model):
         on_delete=models.PROTECT
     )
 
+    obtained = models.PositiveIntegerField(
+        verbose_name=_('obtained')
+    )
+
     class Meta:
         verbose_name = _('challenge')
         verbose_name_plural = _('challenges')
@@ -128,12 +136,27 @@ class Achievement(models.Model):
     def __str__(self):
         return _('Challenge {challenge} carried out by Family {family}').format(challenge=self.challenge.name, family=self.family.name, )
 
+    @classmethod
+    def update_ranking(cls, *args, **kwargs):
+        """
+        Update ranking when adding or removing points
+        """
+        family_set = cls.objects.select_for_update().all().order_by("-score")
+        for i in range(family_set.count()):
+            if i == 0 or family_set[i].score != family_set[i - 1].score:
+                new_rank = i + 1
+            family = family_set[i]
+            family.rank = new_rank
+            family._force_save = True
+            family.save()
+
     @transaction.atomic
     def save(self, *args, **kwargs):
         """
         When saving, also grants points to the family
         """
         self.family = Family.objects.select_for_update().get(pk=self.family_id)
+        self.challenge = Challenge.objects.select_for_update().get(pk=self.challenge_id)
         challenge_points = self.challenge.points
         is_new = self.pk is None
 
@@ -146,6 +169,13 @@ class Achievement(models.Model):
             self.family._force_save = True
             self.family.save()
 
+            self.challenge.refresh_from_db()
+            self.challenge.obtained += 1
+            self.challenge._force_save = True
+            self.challenge.save()
+
+            self.__class__.update_ranking()
+
     @transaction.atomic
     def delete(self, *args, **kwargs):
         """
@@ -163,3 +193,10 @@ class Achievement(models.Model):
         self.family.score -= challenge_points
         self.family._force_save = True
         self.family.save()
+
+        self.challenge.refresh_from_db()
+        self.challenge.obtained -= 1
+        self.challenge._force_save = True
+        self.challenge.save()
+
+        self.__class__.update_ranking()