1024 lines
50 KiB
Python
1024 lines
50 KiB
Python
# Copyright (C) 2023 by Animath
|
|
# SPDX-License-Identifier: GPL-3.0-or-later
|
|
|
|
from collections import OrderedDict
|
|
from random import randint, shuffle
|
|
|
|
from channels.generic.websocket import AsyncJsonWebsocketConsumer
|
|
from django.conf import settings
|
|
from django.utils import translation
|
|
from django.utils.translation import gettext_lazy as _
|
|
from draw.models import Draw, Pool, Round, TeamDraw
|
|
from participation.models import Participation, Tournament
|
|
from registration.models import Registration
|
|
|
|
|
|
def ensure_orga(f):
|
|
"""
|
|
This decorator to an asynchronous receiver guarantees that the user is a volunteer.
|
|
If it is not the case, we send an alert and don't run the function.
|
|
"""
|
|
async def func(self, *args, **kwargs):
|
|
reg = self.registration
|
|
if reg.is_volunteer and not reg.is_admin and self.tournament not in reg.interesting_tournaments \
|
|
or not reg.is_volunteer:
|
|
return await self.alert(_("You are not an organizer."), 'danger')
|
|
|
|
return await f(self, *args, **kwargs)
|
|
|
|
return func
|
|
|
|
|
|
class DrawConsumer(AsyncJsonWebsocketConsumer):
|
|
"""
|
|
This consumer manages the websocket of the draw interface.
|
|
"""
|
|
def __int__(self, *args, **kwargs):
|
|
super().__init__(*args, **kwargs)
|
|
self.tournament_id = None
|
|
self.tournament = None
|
|
self.participations = None
|
|
self.registration = None
|
|
|
|
async def connect(self) -> None:
|
|
"""
|
|
This function is called when a new websocket is trying to connect to the server.
|
|
We accept only if this is a user of a team of the associated tournament, or a volunteer
|
|
of the tournament.
|
|
"""
|
|
# Get the tournament from the URL
|
|
self.tournament_id = self.scope['url_route']['kwargs']['tournament_id']
|
|
self.tournament = await Tournament.objects.filter(pk=self.tournament_id)\
|
|
.prefetch_related('draw__current_round__current_pool__current_team__participation__team').aget()
|
|
|
|
# Fetch participations from the tournament
|
|
self.participations = []
|
|
async for participation in self.tournament.participations.filter(valid=True).prefetch_related('team'):
|
|
self.participations.append(participation)
|
|
|
|
# Fetch the registration of the current user
|
|
user = self.scope['user']
|
|
reg = await Registration.objects.aget(user=user)
|
|
self.registration = reg
|
|
if reg.is_volunteer and not reg.is_admin and self.tournament not in reg.interesting_tournaments \
|
|
or not reg.is_volunteer and reg.team.participation.tournament != self.tournament:
|
|
# This user may not have access to the drawing session
|
|
await self.close()
|
|
return
|
|
|
|
# Accept the connection
|
|
await self.accept()
|
|
|
|
# Register to channel layers to get updates
|
|
await self.channel_layer.group_add(f"tournament-{self.tournament.id}", self.channel_name)
|
|
if not self.registration.is_volunteer:
|
|
await self.channel_layer.group_add(f"team-{self.registration.team.trigram}", self.channel_name)
|
|
else:
|
|
await self.channel_layer.group_add(f"volunteer-{self.tournament.id}", self.channel_name)
|
|
|
|
async def disconnect(self, close_code) -> None:
|
|
"""
|
|
Called when the websocket got disconnected, for any reason.
|
|
:param close_code: The error code.
|
|
"""
|
|
# Unregister from channel layers
|
|
await self.channel_layer.group_discard(f"tournament-{self.tournament.id}", self.channel_name)
|
|
if not self.registration.is_volunteer:
|
|
await self.channel_layer.group_discard(f"team-{self.registration.team.trigram}", self.channel_name)
|
|
else:
|
|
await self.channel_layer.group_discard(f"volunteer-{self.tournament.id}", self.channel_name)
|
|
|
|
async def alert(self, message: str, alert_type: str = 'info', **kwargs):
|
|
"""
|
|
Send an alert message to the current user.
|
|
:param message: The body of the alert.
|
|
:param alert_type: The type of the alert, which is a bootstrap color (success, warning, info, danger,…)
|
|
"""
|
|
return await self.send_json({'type': 'alert', 'alert_type': alert_type, 'message': str(message)})
|
|
|
|
async def receive_json(self, content, **kwargs):
|
|
"""
|
|
Called when the client sends us some data, parsed as JSON.
|
|
:param content: The sent data, decoded from JSON text. Must content a `type` field.
|
|
"""
|
|
# Refresh tournament
|
|
self.tournament = await Tournament.objects.filter(pk=self.tournament_id)\
|
|
.prefetch_related('draw__current_round__current_pool__current_team__participation__team').aget()
|
|
|
|
match content['type']:
|
|
case 'set_language':
|
|
# Update the translation language
|
|
translation.activate(content['language'])
|
|
case 'start_draw':
|
|
# Start a new draw
|
|
await self.start_draw(**content)
|
|
case 'abort':
|
|
# Abort the current draw
|
|
await self.abort(**content)
|
|
case 'dice':
|
|
# Launch a dice
|
|
await self.process_dice(**content)
|
|
case 'draw_problem':
|
|
# Draw a new problem
|
|
await self.select_problem(**content)
|
|
case 'accept':
|
|
# Accept the proposed problem
|
|
await self.accept_problem(**content)
|
|
case 'reject':
|
|
# Reject the proposed problem
|
|
await self.reject_problem(**content)
|
|
case 'export':
|
|
# Export the current state of the draw
|
|
await self.export(**content)
|
|
case 'continue_final':
|
|
# Continue the draw for the final tournament
|
|
await self.continue_final(**content)
|
|
|
|
@ensure_orga
|
|
async def start_draw(self, fmt: str, **kwargs) -> None:
|
|
"""
|
|
Initialize a new draw, with a given format.
|
|
:param fmt: The format of the tournament, which is the size of each pool.
|
|
Sizes must be between 3 and 5, and the sum must be the number of teams.
|
|
"""
|
|
if await Draw.objects.filter(tournament=self.tournament).aexists():
|
|
return await self.alert(_("The draw is already started."), 'danger')
|
|
|
|
try:
|
|
# Parse format from string
|
|
fmt: list[int] = sorted(map(int, fmt.split('+')), reverse=True)
|
|
except ValueError:
|
|
return await self.alert(_("Invalid format"), 'danger')
|
|
|
|
# Ensure that the number of teams is good
|
|
if sum(fmt) != len(self.participations):
|
|
return await self.alert(
|
|
_("The sum must be equal to the number of teams: expected {len}, got {sum}")
|
|
.format(len=len(self.participations), sum=sum(fmt)), 'danger')
|
|
|
|
# The drawing system works with a maximum of 1 pool of 5 teams, which is already the case in the TFJM²
|
|
if fmt.count(5) > 1:
|
|
return await self.alert(_("There can be at most one pool with 5 teams."), 'danger')
|
|
|
|
# Create the draw
|
|
draw = await Draw.objects.acreate(tournament=self.tournament)
|
|
r1 = None
|
|
for i in [1, 2]:
|
|
# Create the round
|
|
r = await Round.objects.acreate(draw=draw, number=i)
|
|
if i == 1:
|
|
r1 = r
|
|
|
|
for j, f in enumerate(fmt):
|
|
# Create the pool, and correspond the size with the wanted format
|
|
await Pool.objects.acreate(round=r, letter=j + 1, size=f)
|
|
for participation in self.participations:
|
|
# Create a team draw object per participation
|
|
await TeamDraw.objects.acreate(participation=participation, round=r)
|
|
# Send to clients the different pools
|
|
await self.channel_layer.group_send(f"tournament-{self.tournament.id}",
|
|
{'type': 'draw.send_poules', 'round': r})
|
|
|
|
draw.current_round = r1
|
|
await draw.asave()
|
|
|
|
# Make dice box visible
|
|
await self.channel_layer.group_send(f"tournament-{self.tournament.id}",
|
|
{'type': 'draw.dice_visibility', 'visible': True})
|
|
|
|
await self.alert(_("Draw started!"), 'success')
|
|
|
|
# Update user interface
|
|
await self.channel_layer.group_send(f"tournament-{self.tournament.id}",
|
|
{'type': 'draw.start', 'fmt': fmt, 'draw': draw})
|
|
await self.channel_layer.group_send(f"tournament-{self.tournament.id}",
|
|
{'type': 'draw.set_info', 'draw': draw})
|
|
await self.channel_layer.group_send(f"tournament-{self.tournament.id}",
|
|
{'type': 'draw.set_active', 'draw': self.tournament.draw})
|
|
|
|
# Send notification to everyone
|
|
await self.channel_layer.group_send(f"tournament-{self.tournament.id}",
|
|
{'type': 'draw.notify', 'title': 'Tirage au sort du TFJM²',
|
|
'body': "Le tirage au sort du tournoi de "
|
|
f"{self.tournament.name} a commencé !"})
|
|
|
|
async def draw_start(self, content) -> None:
|
|
"""
|
|
Send information to users that the draw has started.
|
|
"""
|
|
await self.alert(_("The draw for the tournament {tournament} will start.")
|
|
.format(tournament=self.tournament.name), 'warning')
|
|
await self.send_json({'type': 'draw_start', 'fmt': content['fmt'],
|
|
'trigrams': [p.team.trigram for p in self.participations]})
|
|
|
|
@ensure_orga
|
|
async def abort(self, **kwargs) -> None:
|
|
"""
|
|
Abort the current draw and delete all associated information.
|
|
"""
|
|
if not await Draw.objects.filter(tournament=self.tournament).aexists():
|
|
return await self.alert(_("The draw has not started yet."), 'danger')
|
|
|
|
# Delete draw
|
|
# All associated data will be deleted by cascade
|
|
await self.tournament.draw.adelete()
|
|
# Send information to all users
|
|
await self.channel_layer.group_send(f"tournament-{self.tournament.id}", {'type': 'draw_abort'})
|
|
|
|
async def draw_abort(self, content) -> None:
|
|
"""
|
|
Send information to users that the draw was aborted.
|
|
"""
|
|
await self.alert(_("The draw for the tournament {tournament} is aborted.")
|
|
.format(tournament=self.tournament.name), 'danger')
|
|
await self.send_json({'type': 'abort'})
|
|
|
|
async def process_dice(self, trigram: str | None = None, **kwargs):
|
|
"""
|
|
Launch the dice for a team.
|
|
If we are in the first step, that determine the passage order and the pools of each team.
|
|
For the second step, that determines the order of the teams to draw problems.
|
|
:param trigram: The team that we want to force the launch. None if we launch for our team, or for the
|
|
first free team in the case of volunteers.
|
|
"""
|
|
if not await Draw.objects.filter(tournament=self.tournament).aexists():
|
|
return await self.alert(_("The draw has not started yet."), 'danger')
|
|
|
|
state = self.tournament.draw.get_state()
|
|
|
|
if self.registration.is_volunteer:
|
|
# A volunteer can either force the launch for a specific team,
|
|
# or launch for the first team that has not launched its dice.
|
|
if trigram:
|
|
participation = await Participation.objects.filter(team__trigram=trigram)\
|
|
.prefetch_related('team').aget()
|
|
else:
|
|
# First free team
|
|
if state == 'DICE_ORDER_POULE':
|
|
participation = await Participation.objects\
|
|
.filter(teamdraw__pool=self.tournament.draw.current_round.current_pool,
|
|
teamdraw__choice_dice__isnull=True).prefetch_related('team').afirst()
|
|
else:
|
|
participation = await Participation.objects\
|
|
.filter(teamdraw__round=self.tournament.draw.current_round,
|
|
teamdraw__passage_dice__isnull=True).prefetch_related('team').afirst()
|
|
else:
|
|
# Fetch the participation of the current user
|
|
participation = await Participation.objects.filter(team__participants=self.registration)\
|
|
.prefetch_related('team').aget()
|
|
|
|
if participation is None:
|
|
# Should not happen in normal cases
|
|
return await self.alert(_("This is not the time for this."), 'danger')
|
|
|
|
trigram = participation.team.trigram
|
|
|
|
team_draw = await TeamDraw.objects.filter(participation=participation,
|
|
round_id=self.tournament.draw.current_round_id).aget()
|
|
|
|
# Ensure that this is the right state to launch a dice and that the team didn't already launch the dice
|
|
# and that it can launch a dice yet.
|
|
# Prevent some async issues
|
|
match state:
|
|
case 'DICE_SELECT_POULES':
|
|
if team_draw.passage_dice is not None:
|
|
return await self.alert(_("You've already launched the dice."), 'danger')
|
|
case 'DICE_ORDER_POULE':
|
|
if team_draw.choice_dice is not None:
|
|
return await self.alert(_("You've already launched the dice."), 'danger')
|
|
if not await self.tournament.draw.current_round.current_pool.teamdraw_set\
|
|
.filter(participation=participation).aexists():
|
|
return await self.alert(_("It is not your turn."), 'danger')
|
|
case _:
|
|
return await self.alert(_("This is not the time for this."), 'danger')
|
|
|
|
# Launch the dice and get the result
|
|
res = randint(1, 100)
|
|
if state == 'DICE_SELECT_POULES':
|
|
team_draw.passage_dice = res
|
|
else:
|
|
team_draw.choice_dice = res
|
|
await team_draw.asave()
|
|
|
|
# Send the dice result to all users
|
|
await self.channel_layer.group_send(
|
|
f"tournament-{self.tournament.id}", {'type': 'draw.dice', 'team': trigram, 'result': res})
|
|
|
|
if state == 'DICE_SELECT_POULES' and \
|
|
not await TeamDraw.objects.filter(round_id=self.tournament.draw.current_round_id,
|
|
passage_dice__isnull=True).aexists():
|
|
# Check duplicates
|
|
if await self.check_duplicate_dices():
|
|
return
|
|
# All teams launched their dice, we can process the result
|
|
await self.process_dice_select_poules()
|
|
elif state == 'DICE_ORDER_POULE' and \
|
|
not await TeamDraw.objects.filter(pool=self.tournament.draw.current_round.current_pool,
|
|
choice_dice__isnull=True).aexists():
|
|
# Check duplicates
|
|
if await self.check_duplicate_dices():
|
|
return
|
|
# All teams launched their dice for the choice order, we can process the result
|
|
await self.process_dice_order_poule()
|
|
|
|
async def check_duplicate_dices(self) -> bool:
|
|
"""
|
|
Check that all dices are distinct, and reset some dices if necessary.
|
|
:return: True if there are duplicate dices, False otherwise.
|
|
"""
|
|
state = self.tournament.draw.get_state()
|
|
|
|
# Get concerned TeamDraw objects
|
|
if state == 'DICE_SELECT_POULES':
|
|
tds = [td async for td in TeamDraw.objects.filter(round_id=self.tournament.draw.current_round_id)
|
|
.prefetch_related('participation__team')]
|
|
dices = {td: td.passage_dice for td in tds}
|
|
else:
|
|
tds = [td async for td in TeamDraw.objects
|
|
.filter(pool_id=self.tournament.draw.current_round.current_pool_id)
|
|
.prefetch_related('participation__team')]
|
|
dices = {td: td.choice_dice for td in tds}
|
|
|
|
values = list(dices.values())
|
|
error = False
|
|
for v in set(values):
|
|
if values.count(v) > 1:
|
|
# v is a duplicate value
|
|
# Get all teams that have the same result
|
|
dups = [td for td in tds if td.passage_dice == v]
|
|
|
|
for dup in dups:
|
|
# Reset the dice
|
|
dup.passage_dice = None
|
|
await dup.asave()
|
|
await self.channel_layer.group_send(
|
|
f"tournament-{self.tournament.id}",
|
|
{'type': 'draw.dice', 'team': dup.participation.team.trigram, 'result': None})
|
|
|
|
# Send notification to concerned teams
|
|
await self.channel_layer.group_send(
|
|
f"team-{dup.participation.team.trigram}",
|
|
{'type': 'draw.notify', 'title': 'Tirage au sort du TFJM²',
|
|
'body': 'Votre score de dé est identique à celui de une ou plusieurs équipes. '
|
|
'Veuillez le relancer.'}
|
|
)
|
|
# Alert the tournament
|
|
await self.channel_layer.group_send(
|
|
f"tournament-{self.tournament.id}",
|
|
{'type': 'draw.alert',
|
|
'message': _('Dices from teams {teams} are identical. Please relaunch your dices.').format(
|
|
teams=', '.join(td.participation.team.trigram for td in dups)),
|
|
'alert_type': 'warning'})
|
|
error = True
|
|
|
|
return error
|
|
|
|
async def process_dice_select_poules(self):
|
|
"""
|
|
Called when all teams launched their dice.
|
|
Place teams into pools and order their passage.
|
|
"""
|
|
r = self.tournament.draw.current_round
|
|
tds = [td async for td in TeamDraw.objects.filter(round=r).prefetch_related('participation__team')]
|
|
# Sort teams per dice results
|
|
tds.sort(key=lambda td: td.passage_dice)
|
|
tds_copy = tds.copy()
|
|
|
|
# For each pool of size N, put the N next teams into this pool
|
|
async for p in Pool.objects.filter(round_id=self.tournament.draw.current_round_id).order_by('letter').all():
|
|
# Fetch the N teams, then order them in a new order for the passages inside the pool
|
|
# We multiply the dice scores by 27 mod 100 (which order is 20 mod 100) for this new order
|
|
# This simulates a deterministic shuffle
|
|
pool_tds = sorted(tds_copy[:p.size], key=lambda td: (td.passage_dice * 27) % 100)
|
|
# Remove the head
|
|
tds_copy = tds_copy[p.size:]
|
|
for i, td in enumerate(pool_tds):
|
|
# Set the pool and the passage index for each team of the pool
|
|
td.pool = p
|
|
td.passage_index = i
|
|
await td.asave()
|
|
|
|
# The passages of the second round are determined from the scores of the dices
|
|
# The team that has the lowest dice score goes to the first pool, then the team
|
|
# that has the second-lowest score goes to the second pool, etc.
|
|
# This also determines the passage order, in the natural order this time.
|
|
# If there is a 5-teams pool, we force the last team to be in the first pool,
|
|
# which is this specific pool since they are ordered by decreasing size.
|
|
tds_copy = tds.copy()
|
|
round2 = await self.tournament.draw.round_set.filter(number=2).aget()
|
|
round2_pools = [p async for p in Pool.objects.filter(round__draw__tournament=self.tournament, round=round2)
|
|
.order_by('letter').all()]
|
|
current_pool_id, current_passage_index = 0, 0
|
|
for i, td in enumerate(tds_copy):
|
|
if i == len(tds) - 1 and round2_pools[0].size == 5:
|
|
current_pool_id = 0
|
|
current_passage_index = 4
|
|
|
|
td2 = await TeamDraw.objects.filter(participation=td.participation, round=round2).aget()
|
|
td2.pool = round2_pools[current_pool_id]
|
|
td2.passage_index = current_passage_index
|
|
current_pool_id += 1
|
|
if current_pool_id == len(round2_pools):
|
|
current_pool_id = 0
|
|
current_passage_index += 1
|
|
await td2.asave()
|
|
|
|
# The current pool is the first pool of the current (first) round
|
|
pool = await Pool.objects.filter(round=self.tournament.draw.current_round, letter=1).aget()
|
|
self.tournament.draw.current_round.current_pool = pool
|
|
await self.tournament.draw.current_round.asave()
|
|
|
|
# Display dice result in the header of the information alert
|
|
msg = "Les résultats des dés sont les suivants : "
|
|
msg += ", ".join(f"<strong>{td.participation.team.trigram}</strong> ({td.passage_dice})" for td in tds)
|
|
msg += ". L'ordre de passage et les compositions des différentes poules sont affiché⋅es sur le côté. "
|
|
msg += "Attention : les ordres de passage sont déterminés à partir des scores des dés, mais ne sont pas "
|
|
msg += "directement l'ordre croissant des dés, afin d'avoir des poules mélangées."
|
|
self.tournament.draw.last_message = msg
|
|
await self.tournament.draw.asave()
|
|
|
|
# Reset team dices
|
|
for td in tds:
|
|
await self.channel_layer.group_send(
|
|
f"tournament-{self.tournament.id}",
|
|
{'type': 'draw.dice', 'team': td.participation.team.trigram, 'result': None})
|
|
|
|
# Hide dice interface
|
|
await self.channel_layer.group_send(f"tournament-{self.tournament.id}",
|
|
{'type': 'draw.dice_visibility', 'visible': False})
|
|
|
|
# Display dice interface only for the teams in the first pool, and for volunteers
|
|
async for td in pool.teamdraw_set.prefetch_related('participation__team').all():
|
|
await self.channel_layer.group_send(f"team-{td.participation.team.trigram}",
|
|
{'type': 'draw.dice_visibility', 'visible': True})
|
|
await self.channel_layer.group_send(f"volunteer-{self.tournament.id}",
|
|
{'type': 'draw.dice_visibility', 'visible': True})
|
|
|
|
# First send the second pool to have the good team order
|
|
await self.channel_layer.group_send(f"tournament-{self.tournament.id}",
|
|
{'type': 'draw.send_poules',
|
|
'round': await self.tournament.draw.round_set.filter(number=2).aget()})
|
|
await self.channel_layer.group_send(f"tournament-{self.tournament.id}",
|
|
{'type': 'draw.send_poules',
|
|
'round': self.tournament.draw.current_round})
|
|
|
|
# Update information header and the active team on the recap menu
|
|
await self.channel_layer.group_send(f"tournament-{self.tournament.id}",
|
|
{'type': 'draw.set_info', 'draw': self.tournament.draw})
|
|
await self.channel_layer.group_send(f"tournament-{self.tournament.id}",
|
|
{'type': 'draw.set_active', 'draw': self.tournament.draw})
|
|
|
|
async def process_dice_order_poule(self):
|
|
"""
|
|
Called when all teams of the current launched their dice to determine the choice order.
|
|
Place teams into pools and order their passage.
|
|
"""
|
|
pool = self.tournament.draw.current_round.current_pool
|
|
|
|
tds = [td async for td in TeamDraw.objects.filter(pool=pool).prefetch_related('participation__team')]
|
|
# Order teams by decreasing dice score
|
|
tds.sort(key=lambda x: -x.choice_dice)
|
|
for i, td in enumerate(tds):
|
|
td.choose_index = i
|
|
await td.asave()
|
|
|
|
# The first team to draw its problem is the team that has the highest dice score
|
|
pool.current_team = tds[0]
|
|
await pool.asave()
|
|
|
|
self.tournament.draw.last_message = ""
|
|
await self.tournament.draw.asave()
|
|
|
|
# Update information header
|
|
await self.channel_layer.group_send(f"tournament-{self.tournament.id}",
|
|
{'type': 'draw.set_info', 'draw': self.tournament.draw})
|
|
await self.channel_layer.group_send(f"tournament-{self.tournament.id}",
|
|
{'type': 'draw.set_active', 'draw': self.tournament.draw})
|
|
|
|
# Hide dice button to everyone
|
|
await self.channel_layer.group_send(f"tournament-{self.tournament.id}",
|
|
{'type': 'draw.dice_visibility', 'visible': False})
|
|
|
|
# Display the box button to the first team and to volunteers
|
|
trigram = pool.current_team.participation.team.trigram
|
|
await self.channel_layer.group_send(f"team-{trigram}",
|
|
{'type': 'draw.box_visibility', 'visible': True})
|
|
await self.channel_layer.group_send(f"volunteer-{self.tournament.id}",
|
|
{'type': 'draw.box_visibility', 'visible': True})
|
|
|
|
# Notify the team that it can draw a problem
|
|
await self.channel_layer.group_send(f"team-{tds[0].participation.team.trigram}",
|
|
{'type': 'draw.notify', 'title': "À votre tour !",
|
|
'body': "C'est à vous de tirer un nouveau problème !"})
|
|
|
|
async def select_problem(self, **kwargs):
|
|
"""
|
|
Called when a team draws a problem.
|
|
We choose randomly a problem that is available and propose it to the current team.
|
|
"""
|
|
state = self.tournament.draw.get_state()
|
|
|
|
if state != 'WAITING_DRAW_PROBLEM':
|
|
return await self.alert(_("This is not the time for this."), 'danger')
|
|
|
|
pool = self.tournament.draw.current_round.current_pool
|
|
td = pool.current_team
|
|
|
|
if not self.registration.is_volunteer:
|
|
participation = await Participation.objects.filter(team__participants=self.registration)\
|
|
.prefetch_related('team').aget()
|
|
# Ensure that the user can draws a problem at this time
|
|
if participation.id != td.participation_id:
|
|
return await self.alert("This is not your turn.", 'danger')
|
|
|
|
while True:
|
|
# Choose a random problem
|
|
problem = randint(1, len(settings.PROBLEMS))
|
|
# Check that the user didn't already accept this problem for the first round
|
|
# if this is the second round
|
|
if await TeamDraw.objects.filter(participation_id=td.participation_id,
|
|
round__draw__tournament=self.tournament,
|
|
round__number=1,
|
|
accepted=problem).aexists():
|
|
continue
|
|
# Check that the problem is not already chosen once (or twice for a 5-teams pool)
|
|
if await pool.teamdraw_set.filter(accepted=problem).acount() < (2 if pool.size == 5 else 1):
|
|
break
|
|
|
|
td.purposed = problem
|
|
await td.asave()
|
|
|
|
# Update interface
|
|
trigram = td.participation.team.trigram
|
|
await self.channel_layer.group_send(f"team-{trigram}",
|
|
{'type': 'draw.box_visibility', 'visible': False})
|
|
await self.channel_layer.group_send(f"volunteer-{self.tournament.id}",
|
|
{'type': 'draw.box_visibility', 'visible': False})
|
|
await self.channel_layer.group_send(f"team-{trigram}",
|
|
{'type': 'draw.buttons_visibility', 'visible': True})
|
|
await self.channel_layer.group_send(f"volunteer-{self.tournament.id}",
|
|
{'type': 'draw.buttons_visibility', 'visible': True})
|
|
await self.channel_layer.group_send(f"team-{self.tournament.id}",
|
|
{'type': 'draw.draw_problem', 'team': trigram, 'problem': problem})
|
|
|
|
self.tournament.draw.last_message = ""
|
|
await self.tournament.draw.asave()
|
|
await self.channel_layer.group_send(f"tournament-{self.tournament.id}",
|
|
{'type': 'draw.set_info', 'draw': self.tournament.draw})
|
|
|
|
async def accept_problem(self, **kwargs):
|
|
"""
|
|
Called when a team accepts a problem.
|
|
We pass to the next team is there is one, or to the next pool, or the next round, or end the draw.
|
|
"""
|
|
|
|
if not await Draw.objects.filter(tournament=self.tournament).aexists():
|
|
return await self.alert(_("The draw has not started yet."), 'danger')
|
|
|
|
state = self.tournament.draw.get_state()
|
|
|
|
if state != 'WAITING_CHOOSE_PROBLEM':
|
|
return await self.alert(_("This is not the time for this."), 'danger')
|
|
|
|
r = self.tournament.draw.current_round
|
|
pool = r.current_pool
|
|
td = pool.current_team
|
|
if not self.registration.is_volunteer:
|
|
participation = await Participation.objects.filter(team__participants=self.registration)\
|
|
.prefetch_related('team').aget()
|
|
# Ensure that the user can accept a problem at this time
|
|
if participation.id != td.participation_id:
|
|
return await self.alert("This is not your turn.", 'danger')
|
|
|
|
td.accepted = td.purposed
|
|
td.purposed = None
|
|
await td.asave()
|
|
|
|
trigram = td.participation.team.trigram
|
|
msg = f"L'équipe <strong>{trigram}</strong> a accepté le problème <strong>{td.accepted} : " \
|
|
f"{settings.PROBLEMS[td.accepted - 1]}</strong>. "
|
|
if pool.size == 5 and await pool.teamdraw_set.filter(accepted=td.accepted).acount() < 2:
|
|
msg += "Une équipe peut encore l'accepter."
|
|
else:
|
|
msg += "Plus personne ne peut l'accepter."
|
|
self.tournament.draw.last_message = msg
|
|
await self.tournament.draw.asave()
|
|
|
|
# Send the accepted problem to the users
|
|
await self.channel_layer.group_send(f"team-{trigram}",
|
|
{'type': 'draw.buttons_visibility', 'visible': False})
|
|
await self.channel_layer.group_send(f"volunteer-{self.tournament.id}",
|
|
{'type': 'draw.buttons_visibility', 'visible': False})
|
|
await self.channel_layer.group_send(f"tournament-{self.tournament.id}",
|
|
{'type': 'draw.set_problem',
|
|
'round': r.number,
|
|
'team': trigram,
|
|
'problem': td.accepted})
|
|
|
|
if await pool.teamdraw_set.filter(accepted__isnull=True).aexists():
|
|
# Continue this pool since there is at least one team that does not have selected its problem
|
|
# Get next team
|
|
next_td = await pool.next_td()
|
|
pool.current_team = next_td
|
|
await pool.asave()
|
|
|
|
new_trigram = next_td.participation.team.trigram
|
|
await self.channel_layer.group_send(f"team-{new_trigram}",
|
|
{'type': 'draw.box_visibility', 'visible': True})
|
|
await self.channel_layer.group_send(f"volunteer-{self.tournament.id}",
|
|
{'type': 'draw.box_visibility', 'visible': True})
|
|
|
|
# Notify the team that it can draw a problem
|
|
await self.channel_layer.group_send(f"team-{new_trigram}",
|
|
{'type': 'draw.notify', 'title': "À votre tour !",
|
|
'body': "C'est à vous de tirer un nouveau problème !"})
|
|
else:
|
|
# Pool is ended
|
|
await self.end_pool(pool)
|
|
|
|
await self.channel_layer.group_send(f"tournament-{self.tournament.id}",
|
|
{'type': 'draw.set_info', 'draw': self.tournament.draw})
|
|
await self.channel_layer.group_send(f"tournament-{self.tournament.id}",
|
|
{'type': 'draw.set_active', 'draw': self.tournament.draw})
|
|
|
|
async def end_pool(self, pool: Pool) -> None:
|
|
"""
|
|
End the pool, and pass to the next one, or to the next round, or end the draw.
|
|
:param pool: The pool to end.
|
|
"""
|
|
msg = self.tournament.draw.last_message
|
|
r = pool.round
|
|
|
|
if pool.size == 5:
|
|
# Maybe reorder teams if the same problem is presented twice
|
|
problems = OrderedDict()
|
|
async for td in pool.team_draws:
|
|
problems.setdefault(td.accepted, [])
|
|
problems[td.accepted].append(td)
|
|
p_index = 0
|
|
for pb, tds in problems.items():
|
|
if len(tds) == 2:
|
|
# Le règlement demande à ce que l'ordre soit tiré au sort
|
|
shuffle(tds)
|
|
tds[0].passage_index = p_index
|
|
tds[1].passage_index = p_index + 1
|
|
p_index += 2
|
|
await tds[0].asave()
|
|
await tds[1].asave()
|
|
for pb, tds in problems.items():
|
|
if len(tds) == 1:
|
|
tds[0].passage_index = p_index
|
|
p_index += 1
|
|
await tds[0].asave()
|
|
|
|
# Send the reordered pool
|
|
await self.channel_layer.group_send(f"tournament-{self.tournament.id}", {
|
|
'type': 'draw.reorder_pool',
|
|
'round': r.number,
|
|
'pool': pool.get_letter_display(),
|
|
'teams': [td.participation.team.trigram
|
|
async for td in pool.team_draws.prefetch_related('participation__team')],
|
|
'problems': [td.accepted async for td in pool.team_draws],
|
|
})
|
|
|
|
msg += f"<br><br>Le tirage de la poule {pool.get_letter_display()}{r.number} est terminé. " \
|
|
f"Le tableau récapitulatif est en bas."
|
|
self.tournament.draw.last_message = msg
|
|
await self.tournament.draw.asave()
|
|
|
|
if await r.teamdraw_set.filter(accepted__isnull=True).aexists():
|
|
# There is a pool that does not have selected its problem, so we continue to the next pool
|
|
next_pool = await r.next_pool()
|
|
r.current_pool = next_pool
|
|
await r.asave()
|
|
|
|
async for td in next_pool.team_draws.prefetch_related('participation__team').all():
|
|
await self.channel_layer.group_send(f"team-{td.participation.team.trigram}",
|
|
{'type': 'draw.dice_visibility', 'visible': True})
|
|
# Notify the team that it can draw a dice
|
|
await self.channel_layer.group_send(f"team-{td.participation.team.trigram}",
|
|
{'type': 'draw.notify', 'title': "À votre tour !",
|
|
'body': "C'est à vous de lancer le dé !"})
|
|
|
|
await self.channel_layer.group_send(f"tournament-{self.tournament.id}",
|
|
{'type': 'draw.dice_visibility', 'visible': True})
|
|
else:
|
|
# Round is ended
|
|
await self.end_round(r)
|
|
|
|
async def end_round(self, r: Round) -> None:
|
|
"""
|
|
End the round, and pass to the next one, or end the draw.
|
|
:param r: The current round.
|
|
"""
|
|
msg = self.tournament.draw.last_message
|
|
|
|
if r.number == 1 and not self.tournament.final:
|
|
# Next round
|
|
r2 = await self.tournament.draw.round_set.filter(number=2).aget()
|
|
self.tournament.draw.current_round = r2
|
|
msg += "<br><br>Le tirage au sort du tour 1 est terminé."
|
|
self.tournament.draw.last_message = msg
|
|
await self.tournament.draw.asave()
|
|
|
|
for participation in self.participations:
|
|
await self.channel_layer.group_send(
|
|
f"tournament-{self.tournament.id}",
|
|
{'type': 'draw.dice', 'team': participation.team.trigram, 'result': None})
|
|
|
|
# Notify the team that it can draw a dice
|
|
await self.channel_layer.group_send(f"team-{participation.team.trigram}",
|
|
{'type': 'draw.notify', 'title': "À votre tour !",
|
|
'body': "C'est à vous de lancer le dé !"})
|
|
|
|
# Reorder dices
|
|
await self.channel_layer.group_send(f"tournament-{self.tournament.id}",
|
|
{'type': 'draw.send_poules',
|
|
'round': r2})
|
|
|
|
# The passage order for the second round is already determined by the first round
|
|
# Start the first pool of the second round
|
|
p1: Pool = await r2.pool_set.filter(letter=1).aget()
|
|
r2.current_pool = p1
|
|
await r2.asave()
|
|
|
|
async for td in p1.teamdraw_set.prefetch_related('participation__team').all():
|
|
await self.channel_layer.group_send(f"team-{td.participation.team.trigram}",
|
|
{'type': 'draw.dice_visibility', 'visible': True})
|
|
await self.channel_layer.group_send(f"volunteer-{self.tournament.id}",
|
|
{'type': 'draw.dice_visibility', 'visible': True})
|
|
elif r.number == 1 and self.tournament.final:
|
|
# For the final tournament, we wait for a manual update between the two rounds.
|
|
msg += "<br><br>Le tirage au sort du tour 1 est terminé."
|
|
self.tournament.draw.last_message = msg
|
|
await self.tournament.draw.asave()
|
|
|
|
await self.channel_layer.group_send(f"volunteer-{self.tournament.id}",
|
|
{'type': 'draw.export_visibility', 'visible': True})
|
|
|
|
async def reject_problem(self, **kwargs):
|
|
"""
|
|
Called when a team accepts a problem.
|
|
We pass then to the next team.
|
|
"""
|
|
if not await Draw.objects.filter(tournament=self.tournament).aexists():
|
|
return await self.alert(_("The draw has not started yet."), 'danger')
|
|
|
|
state = self.tournament.draw.get_state()
|
|
|
|
if state != 'WAITING_CHOOSE_PROBLEM':
|
|
return await self.alert(_("This is not the time for this."), 'danger')
|
|
|
|
r = self.tournament.draw.current_round
|
|
pool = r.current_pool
|
|
td = pool.current_team
|
|
if not self.registration.is_volunteer:
|
|
participation = await Participation.objects.filter(team__participants=self.registration)\
|
|
.prefetch_related('team').aget()
|
|
# Ensure that the user can reject a problem at this time
|
|
if participation.id != td.participation_id:
|
|
return await self.alert("This is not your turn.", 'danger')
|
|
|
|
# Add the problem to the rejected problems list
|
|
problem = td.purposed
|
|
already_refused = problem in td.rejected
|
|
if not already_refused:
|
|
td.rejected.append(problem)
|
|
td.purposed = None
|
|
await td.asave()
|
|
|
|
remaining = len(settings.PROBLEMS) - 5 - len(td.rejected)
|
|
|
|
# Update messages
|
|
trigram = td.participation.team.trigram
|
|
msg = f"L'équipe <strong>{trigram}</strong> a refusé le problème <strong>{problem} : " \
|
|
f"{settings.PROBLEMS[problem - 1]}</strong>. "
|
|
if remaining >= 0:
|
|
msg += f"Il lui reste {remaining} refus sans pénalité."
|
|
else:
|
|
if already_refused:
|
|
msg += "Cela n'ajoute pas de pénalité."
|
|
else:
|
|
msg += "Cela ajoute une pénalité de 0.5 sur le coefficient de l'oral de læ défenseur⋅se."
|
|
self.tournament.draw.last_message = msg
|
|
await self.tournament.draw.asave()
|
|
|
|
# Update interface
|
|
await self.channel_layer.group_send(f"team-{trigram}",
|
|
{'type': 'draw.buttons_visibility', 'visible': False})
|
|
await self.channel_layer.group_send(f"volunteer-{self.tournament.id}",
|
|
{'type': 'draw.buttons_visibility', 'visible': False})
|
|
await self.channel_layer.group_send(f"tournament-{self.tournament.id}",
|
|
{'type': 'draw.reject_problem',
|
|
'round': r.number, 'team': trigram, 'rejected': td.rejected})
|
|
|
|
if already_refused:
|
|
# The team already refused this problem, and can immediately draw a new one
|
|
next_td = td
|
|
else:
|
|
# We pass to the next team
|
|
next_td = await pool.next_td()
|
|
|
|
pool.current_team = next_td
|
|
await pool.asave()
|
|
|
|
new_trigram = next_td.participation.team.trigram
|
|
await self.channel_layer.group_send(f"team-{new_trigram}",
|
|
{'type': 'draw.box_visibility', 'visible': True})
|
|
await self.channel_layer.group_send(f"volunteer-{self.tournament.id}",
|
|
{'type': 'draw.box_visibility', 'visible': True})
|
|
|
|
await self.channel_layer.group_send(f"tournament-{self.tournament.id}",
|
|
{'type': 'draw.set_info', 'draw': self.tournament.draw})
|
|
await self.channel_layer.group_send(f"tournament-{self.tournament.id}",
|
|
{'type': 'draw.set_active', 'draw': self.tournament.draw})
|
|
|
|
# Notify the team that it can draw a problem
|
|
await self.channel_layer.group_send(f"team-{new_trigram}",
|
|
{'type': 'draw.notify', 'title': "À votre tour !",
|
|
'body': "C'est à vous de tirer un nouveau problème !"})
|
|
|
|
@ensure_orga
|
|
async def export(self, **kwargs):
|
|
"""
|
|
Exports the draw information in the participation app, for the solutions and notes management
|
|
"""
|
|
if not await Draw.objects.filter(tournament=self.tournament).aexists():
|
|
return await self.alert(_("The draw has not started yet."), 'danger')
|
|
|
|
# Export each exportable pool
|
|
async for r in self.tournament.draw.round_set.all():
|
|
async for pool in r.pool_set.all():
|
|
if await pool.is_exportable():
|
|
await pool.export()
|
|
|
|
await self.channel_layer.group_send(f"volunteer-{self.tournament.id}",
|
|
{'type': 'draw.export_visibility', 'visible': False})
|
|
|
|
@ensure_orga
|
|
async def continue_final(self, **kwargs):
|
|
"""
|
|
For the final tournament, continue the draw for the second round
|
|
"""
|
|
if not await Draw.objects.filter(tournament=self.tournament).aexists():
|
|
return await self.alert(_("The draw has not started yet."), 'danger')
|
|
|
|
if not self.tournament.final:
|
|
return await self.alert(_("This is only available for the final tournament."), 'danger')
|
|
|
|
r2 = await self.tournament.draw.round_set.filter(number=2).aget()
|
|
self.tournament.draw.current_round = r2
|
|
msg = "Le tirage au sort pour le tour 2 va commencer. " \
|
|
"L'ordre de passage est déterminé à partir du classement du premier tour."
|
|
self.tournament.draw.last_message = msg
|
|
await self.tournament.draw.asave()
|
|
|
|
# Set the first pool of the second round as the active pool
|
|
pool = await Pool.objects.filter(round=self.tournament.draw.current_round, letter=1).aget()
|
|
r2.current_pool = pool
|
|
await r2.asave()
|
|
|
|
# Fetch notes from the first round
|
|
notes = dict()
|
|
async for participation in self.tournament.participations.filter(valid=True).prefetch_related('team').all():
|
|
notes[participation] = sum([await pool.aaverage(participation)
|
|
async for pool in self.tournament.pools.filter(participations=participation)
|
|
.prefetch_related('passages').prefetch_related('tweaks')
|
|
if pool.results_available])
|
|
# Sort notes in a decreasing order
|
|
ordered_participations = sorted(notes.keys(), key=lambda x: -notes[x])
|
|
# Define pools and passage orders from the ranking of the first round
|
|
async for pool in r2.pool_set.order_by('letter').all():
|
|
for i in range(pool.size):
|
|
participation = ordered_participations.pop(0)
|
|
td = await TeamDraw.objects.aget(round=r2, participation=participation)
|
|
td.pool = pool
|
|
td.passage_index = i
|
|
await td.asave()
|
|
|
|
# Send pools to users
|
|
await self.channel_layer.group_send(f"tournament-{self.tournament.id}",
|
|
{'type': 'draw.send_poules', 'round': r2})
|
|
|
|
# Reset dices and update interface
|
|
for participation in self.participations:
|
|
await self.channel_layer.group_send(
|
|
f"tournament-{self.tournament.id}",
|
|
{'type': 'draw.dice', 'team': participation.team.trigram, 'result': None})
|
|
|
|
async for td in r2.current_pool.team_draws.prefetch_related('participation__team'):
|
|
await self.channel_layer.group_send(f"team-{td.participation.team.trigram}",
|
|
{'type': 'draw.dice_visibility', 'visible': True})
|
|
|
|
# Notify the team that it can draw a problem
|
|
await self.channel_layer.group_send(f"team-{td.participation.team.trigram}",
|
|
{'type': 'draw.notify', 'title': "À votre tour !",
|
|
'body': "C'est à vous de tirer un nouveau problème !"})
|
|
|
|
await self.channel_layer.group_send(f"volunteer-{self.tournament.id}",
|
|
{'type': 'draw.dice_visibility', 'visible': True})
|
|
await self.channel_layer.group_send(f"volunteer-{self.tournament.id}",
|
|
{'type': 'draw.continue_visibility', 'visible': False})
|
|
|
|
await self.channel_layer.group_send(f"tournament-{self.tournament.id}",
|
|
{'type': 'draw.set_info', 'draw': self.tournament.draw})
|
|
await self.channel_layer.group_send(f"tournament-{self.tournament.id}",
|
|
{'type': 'draw.set_active', 'draw': self.tournament.draw})
|
|
|
|
async def draw_alert(self, content):
|
|
"""
|
|
Send alert to the current user.
|
|
"""
|
|
return await self.alert(**content)
|
|
|
|
async def draw_notify(self, content):
|
|
"""
|
|
Send a notification (with title and body) to the current user.
|
|
"""
|
|
await self.send_json({'type': 'notification', 'title': content['title'], 'body': content['body']})
|
|
|
|
async def draw_set_info(self, content):
|
|
"""
|
|
Set the information banner to the current user.
|
|
"""
|
|
await self.send_json({'type': 'set_info', 'information': await content['draw'].ainformation()})
|
|
|
|
async def draw_dice(self, content):
|
|
"""
|
|
Update the dice of a given team for the current user interface.
|
|
"""
|
|
await self.send_json({'type': 'dice', 'team': content['team'], 'result': content['result']})
|
|
|
|
async def draw_dice_visibility(self, content):
|
|
"""
|
|
Update the visibility of the dice button for the current user.
|
|
"""
|
|
await self.send_json({'type': 'dice_visibility', 'visible': content['visible']})
|
|
|
|
async def draw_box_visibility(self, content):
|
|
"""
|
|
Update the visibility of the box button for the current user.
|
|
"""
|
|
await self.send_json({'type': 'box_visibility', 'visible': content['visible']})
|
|
|
|
async def draw_buttons_visibility(self, content):
|
|
"""
|
|
Update the visibility of the accept/reject buttons for the current user.
|
|
"""
|
|
await self.send_json({'type': 'buttons_visibility', 'visible': content['visible']})
|
|
|
|
async def draw_export_visibility(self, content):
|
|
"""
|
|
Update the visibility of the export button for the current user.
|
|
"""
|
|
await self.send_json({'type': 'export_visibility', 'visible': content['visible']})
|
|
|
|
async def draw_continue_visibility(self, content):
|
|
"""
|
|
Update the visibility of the continue button for the current user.
|
|
"""
|
|
await self.send_json({'type': 'continue_visibility', 'visible': content['visible']})
|
|
|
|
async def draw_send_poules(self, content):
|
|
"""
|
|
Send the pools and the teams to the current user to update the interface.
|
|
"""
|
|
await self.send_json({'type': 'set_poules', 'round': content['round'].number,
|
|
'poules': [{'letter': pool.get_letter_display(), 'teams': await pool.atrigrams()}
|
|
async for pool in content['round'].pool_set.order_by('letter').all()]})
|
|
|
|
async def draw_set_active(self, content):
|
|
"""
|
|
Update the user interface to highlight the current team.
|
|
"""
|
|
r = content['draw'].current_round
|
|
await self.send_json({
|
|
'type': 'set_active',
|
|
'round': r.number,
|
|
'poule': r.current_pool.get_letter_display() if r.current_pool else None,
|
|
'team': r.current_pool.current_team.participation.team.trigram
|
|
if r.current_pool and r.current_pool.current_team else None,
|
|
})
|
|
|
|
async def draw_set_problem(self, content):
|
|
"""
|
|
Send the accepted problem of a team to the current user.
|
|
"""
|
|
await self.send_json({'type': 'set_problem', 'round': content['round'],
|
|
'team': content['team'], 'problem': content['problem']})
|
|
|
|
async def draw_reject_problem(self, content):
|
|
"""
|
|
Send the rejected problems of a team to the current user.
|
|
"""
|
|
await self.send_json({'type': 'reject_problem', 'round': content['round'],
|
|
'team': content['team'], 'rejected': content['rejected']})
|
|
|
|
async def draw_reorder_pool(self, content):
|
|
"""
|
|
Send the new order of a pool to the current user.
|
|
"""
|
|
await self.send_json({'type': 'reorder_poule', 'round': content['round'],
|
|
'poule': content['pool'], 'teams': content['teams'],
|
|
'problems': content['problems']})
|