import random import string from django.db.models.signals import 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 .models import Club, FailedApiCall, CustomUser, Log, TeamRegistration, PlayerRegistration, UnregisteredTeam, UnregisteredPlayer import requests from tournaments.services.email_service import TournamentEmailService from tournaments.models import PlayerDataSource 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() DISCORD_FAILED_CALLS_WEBHOOK_URL = 'https://discord.com/api/webhooks/1248191778134163486/sSoTL6cULCElWr2YFwyllsg7IXxHcCx_YMDJA_cUHtVUU4WOfN-5M7drCJuwNBBfAk9a' DISCORD_LOGS_WEBHOOK_URL = 'https://discord.com/api/webhooks/1257987637449588736/TtOUwzYgSlQH2d3Ps7SfIKRcFALQVa3hfkC-j9K4_UAcWtsfiw4v8NUPbnX2_ZPOYzuv' @receiver(post_save, sender=FailedApiCall) def notify_discord_on_create(sender, instance, created, **kwargs): notify_object_creation_on_discord(created, instance, DISCORD_FAILED_CALLS_WEBHOOK_URL) # @receiver(post_save, sender=CustomUser) # def notify_user_creation_on_discord(sender, instance, created, **kwargs): # notify_object_creation_on_discord(created, instance, DISCORD_LOGS_WEBHOOK_URL) @receiver(post_save, sender=Log) def notify_log_creation_on_discord(sender, instance, created, **kwargs): notify_object_creation_on_discord(created, instance, DISCORD_LOGS_WEBHOOK_URL) # WARNING: using this method requires the instance to have a discord_string method def notify_object_creation_on_discord(created, instance, webhook_url): 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()}' send_discord_message(webhook_url, message) def send_discord_message(webhook_url, content): data = { "content": content } response = requests.post(webhook_url, json=data) if response.status_code != 204: raise ValueError( f'Error sending message to Discord webhook: {response.status_code}, {response.text}' ) @receiver(pre_delete, sender=TeamRegistration) def unregister_team(sender, instance, **kwargs): team_registration = instance tournament = instance.tournament if tournament.is_deleted is True: return # Create unregistered player records and track captain/other player captain = None other_player = None for player in team_registration.playerregistration_set.all(): if player.captain is True: captain = player else: other_player = player # Send unregistration confirmation if captain and captain.registered_online and captain.email: TournamentEmailService.send_unregistration_confirmation( captain, tournament, other_player ) first_waiting_list_team = tournament.first_waiting_list_team() print("first_waiting_list_team", first_waiting_list_team) # Handle waiting list notifications if first_waiting_list_team: waiting_captain = None waiting_other_player = None for player in first_waiting_list_team.playerregistration_set.all(): if player.captain is True: waiting_captain = player else: waiting_other_player = player if waiting_captain and waiting_captain.registered_online and waiting_captain.email: TournamentEmailService.send_out_of_waiting_list_confirmation( waiting_captain, tournament, waiting_other_player ) @receiver(post_save, sender=Tournament) def notify_players_of_tournament_cancellation(sender, instance, **kwargs): tournament = instance if tournament.is_deleted is False: return # Get all team registrations team_registrations = tournament.teamregistration_set.all() for team_registration in team_registrations: captain = None other_player = None # Get players who registered online and have email for player in team_registration.playerregistration_set.all(): print(player, player.registered_online) if player.captain: captain = player else: other_player = player # Send email to captain if captain and captain.registered_online and captain.email: TournamentEmailService.send_tournament_cancellation_notification( captain, tournament, other_player ) # Send email to other player if they exist and registered online if other_player and other_player.registered_online and other_player.email: TournamentEmailService.send_tournament_cancellation_notification( other_player, tournament, captain )