ft_transcendence/tournament/models.py

from django.db import models

from games.models import GameModel

# Create your models here.tu
class TournamentModel(models.Model):

	name = models.CharField(max_length=100)
	nb_players = models.IntegerField()
	nb_players_by_game = models.IntegerField()
	level = models.IntegerField()
	started = models.BooleanField(default=False)
	finished = models.BooleanField(default=False)

	def create_game(self, users_id):
		game_id = GameModel.create(users_id=users_id)
		TournamentGamesModel(game_id=game_id, tournament_id=self.pk).save()
		return game_id
	
	def get_games_id_by_level(self, level):
		return list(TournamentGamesModel.objects.filter(tournament_id=self.pk, tournament_level=level))

	def get_games_id(self):
		return list(TournamentGamesModel.objects.filter(tournament_id=self.pk))
	
	def get_players_id(self):
		lst: [int] = []
		for game_id in self.get_games_id():
			lst.append(GameMembersModel.objects.filter(game_id=game_id))
		return lst

class TournamentGamesModel(models.Model):

	tournament_id = models.IntegerField()
	tournament_level = models.IntegerField()
	game_id = models.IntegerField()