first draft for managing brackets

Author: istar410

insalan/tournament/admin.py

@@ -528,4 +528,65 @@ def launch_group_matchs_action(self,request,queryset):
         self.message_user(request,_("Les matchs ont bien été lancés"))
 
 
-admin.site.register(GroupMatch, GroupMatchAdmin)
+admin.site.register(GroupMatch, GroupMatchAdmin)
+
+
+class BracketAdmin(admin.ModelAdmin):
+    """Admin handle for Brackets"""
+
+    list_display = ("id", "name", "tournament")
+    search_fields = ["name","tournament","tournament__event","tournament__game"]
+    actions = ["create_empty_knockout_matchs_action","fill_knockout_matchs_action"]
+
+    list_filter = ["tournament","tournament__event"]
+
+    @admin.action(description=_("Créer les matchs"))
+    def create_empty_knockout_matchs_action(self,request,queryset):
+        for bracket in queryset:
+            if any([MatchStatus.SCHEDULED != m.status for m in KnockoutMatch.objects.filter(bracket=bracket)]):
+                self.message_user(request,_("Des matchs existent déjà et sont en cours ou terminés"))
+
+            create_empty_knockout_matchs(bracket)
+            self.message_user(request,_("Matchs créer avec succès"))
+
+    @admin.action(description=_("Remplir les matchs"))
+    def fill_knockout_matchs_action(self,request,queryset):
+        for bracket in queryset:
+            # fill_knockout_matchs(bracket)
+            pass
+
+admin.site.register(Bracket, BracketAdmin)
+
+class KnockoutMatchAdmin(admin.ModelAdmin):
+    """Admin handle for Knockout matchs"""
+
+    list_display = ("id", "bracket", "status")
+    filter_horizontal = ("teams",)
+    actions = [""]
+
+    list_filter = ["bracket", "bracket__tournament", "round_number", "index_in_round"]
+
+    @admin.action(description=_("Lancer les matchs"))
+    def launch_knockout_matchs_action(self,request,queryset):
+        for match in queryset:
+            for team in match.get_teams():
+                team_matchs = KnockoutMatch.objects.filter(teams=team,round_index__gt=match.round_number)
+                for team_match in team_matchs:
+                    if team_match.status == MatchStatus.ONGOING:
+                        self.message_user(request,_(f"L'équipe {team.name} est encore dans un match en cours"), messages.ERROR)
+                        return
+                    if team_match.status == MatchStatus.SCHEDULED:
+                        self.message_user(request,_(f"L'équipe {team.name} n'a pas encore joué un ou des matchs des rounds précédent"), messages.ERROR)
+                        return
+
+            if len(match.get_teams()) == 1:
+                match.status = MatchStatus.COMPLETED
+                # score
+            else:
+                match.status = MatchStatus.ONGOING
+
+            match.save()
+        self.message_user(request,_("Les matchs ont bien été lancés"))
+
+
+admin.site.register(KnockoutMatch, KnockoutMatchAdmin)

insalan/tournament/manage/__init__.py

@@ -1 +1,2 @@
-from .group import *
+from .group import *
+from .bracket import *

insalan/tournament/manage/bracket.py

@@ -0,0 +1,16 @@
+from ..models import Bracket, KnockoutMatch, BracketType, BracketSet
+
+def create_empty_knockout_matchs(bracket: Bracket):
+    depth = bracket.depth
+
+    for match in KnockoutMatch.objects.filter(bracket=bracket):
+        match.delete()
+
+    for round_n in range(1,depth+1):
+        for match_id in range(1,2**(round_n-1)+1):
+            KnockoutMatch.objects.create(round_number=round_n,index_in_round=match_id,bracket=bracket)
+    if bracket.bracket_type == BracketType.DOUBLE:
+        for round_n in range(1,2*depth-1):
+            for match_id in range(1,2**((round_n-1)//2)+1):
+                KnockoutMatch.objects.create(round_number=round_n,index_in_round=match_id,bracket=bracket,bracket_set=BracketSet.LOOSER)
+        KnockoutMatch.objects.create(round_number=0,index_in_round=1,bracket=bracket)

insalan/tournament/models/__init__.py

@@ -1,4 +1,4 @@
-from .bracket import Bracket, KnockoutMatch
+from .bracket import Bracket, KnockoutMatch, BracketSet, BracketType
 from .event import Event
 from .caster import Caster
 from .game import Game

insalan/tournament/models/bracket.py

@@ -1,12 +1,20 @@
 from django.db import models
 from django.utils.translation import gettext_lazy as _
 
+from . import match
+
 class BracketType(models.TextChoices):
     """Information about the type of a bracket, single or double elimination"""
 
     SINGLE = "SINGLE", _("Simple élimination")
     DOUBLE = "DOUBLE", _("Double élimination")
 
+class BracketSet(models.TextChoices):
+    """Information on the bracket set, winner or looser"""
+
+    WINNER = "WINNER"
+    LOOSER = "LOOSER"
+
 class Bracket(models.Model):
     name = models.CharField(
         max_length=40,
@@ -22,7 +30,59 @@ class Bracket(models.Model):
         choices=BracketType.choices,
         max_length=10
     )
+    depth = models.IntegerField(
+        default=1
+    )
 
     class Meta:
         verbose_name = _("Arbre de tournoi")
-        verbose_name_plural = _("Arbres de tournoi")
+        verbose_name_plural = _("Arbres de tournoi")
+
+    # def __init__(self, *args, **kwargs) -> None:
+    #     super().__init__(*args,**kwargs)
+    #     self.cache_depth = self.depth
+
+    # def save(self, *args, **kwargs):
+    #     """
+    #     Override default save method to automatically create matchs when the Bracket is created or modified.
+    #     """
+    #     need_save = False
+
+    #     # if no match exist, then create them
+    #     if len(KnockoutMatch.objects.filter(bracket=self)) == 0:
+    #         for round_n in range(1,depth+1):
+    #             for match_id in range(1,2**(round_n-1)+1):
+    #                 KnockoutMatch.objects.create(round_number=round_n,index_in_round=match_id,bracket=self)
+    #         if self.bracket_type == BracketType.DOUBLE:
+    #             for round_n in range(1,2*depth-1):
+    #                 for match_id in range(1,2**((round_n-1)//2)+1):
+    #                     KnockoutMatch.objects.create(round_number=round_n,index_in_round=match_id,bracket=self,bracket_set=BracketSet.LOOSER)
+    #             KnockoutMatch.objects.create(round_number=0,index_in_round=1,bracket=self)
+    #         need_save = True
+    #     # if the depth has changed and no game is started, create or delete matchs
+    #     elif self.cache_depth < self.depth and all([match.MatchStatus.SCHEDULED == m.status for m in KnockoutMatch.objects.filter(bracket=self)]):
+
+    #         pass
+    #     elif self.cache_depth > self.depth and all([match.MatchStatus.SCHEDULED == m.status for m in KnockoutMatch.objects.filter(bracket=self)]):
+    #         # 
+    #         need_save = True
+
+    #     if need_save:
+    #         super().save()
+
+class KnockoutMatch(match.Match):
+    bracket = models.ForeignKey(
+        "Bracket",
+        verbose_name=_("Arbre de tournoi"),
+        on_delete=models.CASCADE
+    )
+    bracket_set = models.CharField(
+        max_length=10,
+        default=BracketSet.WINNER,
+        choices=BracketSet.choices,
+        verbose_name=_("Type de tableau, gagnant ou perdant")
+    )
+
+    class Meta:
+        verbose_name = _("Match dans un arbre")
+        verbose_name_plural = _("Matchs dans un arbre")

insalan/tournament/models/match.py

@@ -1,6 +1,7 @@
 from django.db import models
 from django.utils.translation import gettext_lazy as _
 from django.contrib.postgres.fields import ArrayField
+from typing import List
 
 class MatchStatus(models.TextChoices):
     """Information about the status of a match"""
@@ -52,6 +53,9 @@ class Match(models.Model):
     class Meta:
         abstract = True
 
+    def get_teams(self) -> List["Team"]:
+        return self.teams.all()
+
 class Score(models.Model):
     score = models.IntegerField(
         verbose_name=_("Score")