77 lines
3.0 KiB
Python
77 lines
3.0 KiB
Python
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 import ProfileSerializer
|
|
from games.serializers import GameSerializer
|
|
|
|
class TournamentSerializer(serializers.ModelSerializer):
|
|
|
|
levels = serializers.SerializerMethodField(read_only=True, required=False)
|
|
state = serializers.SerializerMethodField(read_only=True, required=False)
|
|
participants = serializers.SerializerMethodField(read_only=True, required=False)
|
|
level = serializers.ReadOnlyField()
|
|
started = serializers.ReadOnlyField()
|
|
finished = serializers.ReadOnlyField()
|
|
name = serializers.CharField(default="")
|
|
|
|
class Meta:
|
|
model = TournamentModel
|
|
fields = ["name", "nb_players", "nb_players_by_game", "level", "started", "finished", "levels", "id", "state", "participants"]
|
|
|
|
def get_participants(self, instance: TournamentModel):
|
|
|
|
participants_id: list[ProfileModel]
|
|
|
|
if (instance.started):
|
|
participants_id = instance.get_participants_id()
|
|
else:
|
|
participants_id = tournament_manager.get(instance.pk).get_participants_id()
|
|
|
|
participants_profile: list[ProfileModel] = []
|
|
for participant_id in participants_id:
|
|
query: QuerySet = ProfileModel.objects.filter(user_id = participant_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": participants_id
|
|
}
|
|
participants_profile.append(profile_data)
|
|
|
|
return participants_profile
|
|
|
|
def get_state(self, instance: TournamentModel):
|
|
return ["waiting", "started", "finished"][instance.started + instance.finished]
|
|
|
|
def get_levels(self, instance: TournamentModel):
|
|
levels: list[list[int]] = []
|
|
for i in range(instance.level):
|
|
games_id: [int] = instance.get_games_id_by_level(i)
|
|
if (games_id == []):
|
|
break
|
|
levels.append(games_id)
|
|
return levels
|
|
|
|
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 |