import random import string from django.db.models.signals import pre_save, post_save, pre_delete, post_delete from django.dispatch import receiver from django.conf import settings from tournaments.models.tournament import Tournament from tournaments.models.unregistered_player import UnregisteredPlayer from django.utils import timezone from django.db.utils import IntegrityError from .models import Club, FailedApiCall, CustomUser, Log, TeamRegistration, PlayerRegistration, UnregisteredTeam, UnregisteredPlayer, TeamSortingType import requests from tournaments.services.email_service import TournamentEmailService, TeamEmailType from tournaments.models import PlayerDataSource from shared.discord import send_discord_log_message, send_discord_failed_calls_message from datetime import datetime # @receiver(pre_save, sender=CustomUser) # def username_check(instance, sender, **kwargs): # """Ensure that username unique constraint is case insensitive""" # if sender.objects.filter(username__iexact=instance.username.lower()): # raise IntegrityError("Le nom d'utilisateur existe déjà") def generate_unique_code(): characters = string.ascii_lowercase + string.digits while True: code = ''.join(random.sample(characters, 3)) if not Club.objects.filter(broadcast_code=code).exists(): return code @receiver(post_save, sender=Club) def assign_unique_code(sender, instance, created, **kwargs): if created and not instance.broadcast_code: instance.broadcast_code = generate_unique_code() instance.save() @receiver(post_save, sender=FailedApiCall) def notify_discord_on_create(sender, instance, created, **kwargs): notify_object_creation_on_discord(created, instance) @receiver(post_save, sender=Log) def notify_log_creation_on_discord(sender, instance, created, **kwargs): notify_object_creation_on_discord(created, instance) # WARNING: using this method requires the instance to have a discord_string method def notify_object_creation_on_discord(created, instance): if created: default_db_engine = settings.DATABASES['default']['ENGINE'] if default_db_engine != 'django.db.backends.sqlite3': site_name = settings.SITE_NAME message = f'{site_name} > New {instance.__class__.__name__} created: {instance.discord_string()}' if isinstance(instance, FailedApiCall): send_discord_failed_calls_message(message) else: send_discord_log_message(message) def notify_team(team, tournament, message_type): #print(team, message_type) if tournament.enable_online_registration is False: return if team.has_registered_online() is False: return if tournament.should_be_over(): return if tournament.supposedly_in_progress(): return captain = None other_player = None for player in team.playerregistration_set.all(): if player.captain: captain = player else: other_player = player if captain: TournamentEmailService.notify(captain, other_player, tournament, message_type) if not captain.registered_online or not captain.email: TournamentEmailService.notify(other_player, captain, tournament, message_type) else: # Notify both players separately if there is no captain or the captain is unavailable players = list(team.playerregistration_set.all()) if len(players) == 2: first_player, second_player = players TournamentEmailService.notify(first_player, second_player, tournament, message_type) TournamentEmailService.notify(second_player, first_player, tournament, message_type) elif len(players) == 1: # If there's only one player, just send them the notification TournamentEmailService.notify(players[0], None, tournament, message_type) @receiver(pre_delete, sender=TeamRegistration) def unregister_team(sender, instance, **kwargs): if instance.tournament.is_deleted: return if instance.tournament.enable_online_registration is False: return notify_team(instance, instance.tournament, TeamEmailType.UNREGISTERED) teams = instance.tournament.teams(True) first_waiting_list_team = instance.tournament.first_waiting_list_team(teams) if first_waiting_list_team and first_waiting_list_team.id != instance.id: notify_team(first_waiting_list_team, instance.tournament, TeamEmailType.OUT_OF_WAITING_LIST) @receiver(post_save, sender=Tournament) def notify_players_of_tournament_cancellation(sender, instance, **kwargs): if not instance.is_deleted: return if instance.enable_online_registration is False: return for team_registration in instance.teamregistration_set.all(): notify_team(team_registration, instance, TeamEmailType.TOURNAMENT_CANCELED) @receiver(pre_save, sender=Tournament) def check_waiting_list(sender, instance, **kwargs): if instance.id is None: return if instance.enable_online_registration is False: return try: previous_state = Tournament.objects.get(id=instance.id) except Tournament.DoesNotExist: previous_state = None if previous_state is None: return teams_out_to_warn = [] teams_in_to_warn = [] if previous_state.team_count > instance.team_count: teams_to_remove_count = previous_state.team_count - instance.team_count previous_state_teams = previous_state.teams(True) sorted_teams = sorted( [team for team in previous_state_teams if team.stage != "Attente" and not (team.wildcard_bracket or team.wildcard_groupstage)], key=lambda t: ( t.registration_date is None, t.registration_date or datetime.min, t.initial_weight, t.team_registration.id ) if previous_state.team_sorting == TeamSortingType.INSCRIPTION_DATE else (t.initial_weight, t.team_registration.id) ) teams_out_to_warn = sorted_teams[-teams_to_remove_count:] elif previous_state.team_count < instance.team_count: teams_in_to_warn = [ team for team in previous_state.teams(True)[(instance.team_count - previous_state.team_count):] if team.stage == "Attente" ] for team in teams_in_to_warn: notify_team(team.team_registration, instance, TeamEmailType.IN_TOURNAMENT_STRUCTURE) for team in teams_out_to_warn: notify_team(team.team_registration, instance, TeamEmailType.OUT_OF_TOURNAMENT_STRUCTURE) @receiver(pre_save, sender=TeamRegistration) def warn_team_walkout_status_change(sender, instance, **kwargs): if instance.id is None or instance.tournament.enable_online_registration is False: return previous_instance = None try: previous_instance = TeamRegistration.objects.get(id=instance.id) except TeamRegistration.DoesNotExist: return previous_teams = instance.tournament.teams(True) current_teams = instance.tournament.teams(True, instance) previous_retrieved_teams = [team for team in previous_teams if team.team_registration.id == previous_instance.id] current_retrieved_teams = [team for team in current_teams if team.team_registration.id == instance.id] was_out = previous_instance.out_of_tournament() is_out = instance.out_of_tournament() if len(previous_retrieved_teams) > 0 and previous_retrieved_teams[0].stage == "Attente": was_out = True if len(current_retrieved_teams) > 0 and current_retrieved_teams[0].stage == "Attente": is_out = True print(was_out, previous_instance.out_of_tournament(), is_out, instance.out_of_tournament()) if not instance.out_of_tournament() and is_out and (previous_instance.out_of_tournament() or not was_out): notify_team(instance, instance.tournament, TeamEmailType.OUT_OF_WALKOUT_WAITING_LIST) elif was_out and not is_out: notify_team(instance, instance.tournament, TeamEmailType.OUT_OF_WALKOUT_IS_IN) elif not previous_instance.out_of_tournament() and instance.out_of_tournament(): notify_team(instance, instance.tournament, TeamEmailType.WALKOUT) if was_out and not is_out: first_out_of_list = instance.tournament.first_waiting_list_team(current_teams) if first_out_of_list: notify_team(first_out_of_list, instance.tournament, TeamEmailType.UNEXPECTED_OUT_OF_TOURNAMENT) elif not was_out and is_out: first_waiting_list_team = instance.tournament.first_waiting_list_team(previous_teams) if first_waiting_list_team: notify_team(first_waiting_list_team, instance.tournament, TeamEmailType.OUT_OF_WAITING_LIST)