from rest_framework import serializers from django.db.models.query import QuerySet from django.contrib.auth.models import User from .models import TournamentModel, tournament_manager from profiles.models import ProfileModel from profiles.serializers.ProfileSerializer import ProfileSerializer from games.serializers import GameSerializer class TournamentSerializer(serializers.ModelSerializer): rounds = serializers.SerializerMethodField(read_only=True, required=False) state = serializers.SerializerMethodField(read_only=True, required=False) players = serializers.SerializerMethodField(read_only=True, required=False) round = serializers.ReadOnlyField() started = serializers.ReadOnlyField() finished = serializers.ReadOnlyField() name = serializers.CharField(default="") class Meta: model = TournamentModel fields = ["name", "nb_players", "nb_players_by_game", "round", "started", "finished", "rounds", "id", "state", "players"] def get_players(self, instance: TournamentModel): players_id: list[ProfileModel] if (instance.started): players_id = instance.get_players_id() else: players_id = tournament_manager.get(instance.pk).get_players_id() players_profile: list[ProfileModel] = [] for player_id in players_id: query: QuerySet = ProfileModel.objects.filter(user_id = player_id) profile_data: dict if query.exists(): profile_data = ProfileSerializer(query[0]).data else: profile_data = { "username": "deleted_user", "avatar": "/static/avatars/default.avif", "user_id": players_id } players_profile.append(profile_data) return players_profile def get_state(self, instance: TournamentModel): return ["waiting", "started", "finished"][instance.started + instance.finished] def get_rounds(self, instance: TournamentModel): rounds: list[list[int]] = [] for i in range(instance.round): games_id: list[int] = instance.get_games_id_by_round(i) if (games_id == []): break rounds.append(games_id) return rounds def validate_nb_players(self, value: int): if (value < 2): raise serializers.ValidationError("The numbers of players must be greather than 2.") return value def validate_nb_players_by_game(self, value: int): if (value < 2): raise serializers.ValidationError("The numbers of players by game must be greather than 2.") nb_players: str = self.initial_data.get("nb_players") if (nb_players is not None and nb_players.isnumeric()): nb_players: int = int(nb_players) if (value > nb_players): raise serializers.ValidationError("The numbers of players by game must be smaller than the numbers of players.") return value