tournament: add: game call func when finished

This commit is contained in:
2024-04-23 15:41:18 +02:00
parent 2a63edf739
commit 743bb7edeb
4 changed files with 29 additions and 3 deletions

View File

@ -7,6 +7,10 @@ from django.contrib.auth.models import User
import time
from typing import TYPE_CHECKING
if TYPE_CHECKING:
from tournament.models import TournamentGameModel
class GameModel(models.Model):
@ -27,13 +31,28 @@ class GameModel(models.Model):
self.start_timestamp = round(time.time() * 1000, 1)
self.started = True
self.save()
def get_tournament(self) -> None | TournamentGameModel:
from tournament.models import TournamentGameModel
query = TournamentGameModel.objects.filter(game=self)
if (not query.exists()):
return None
return query[0]
def finish(self, winner_id):
self.winner_id = winner_id
self.finished = True
self.stop_timestamp = round(time.time() * 1000, 1)
self.save()
tournament = self.get_tournament()
if tournament is not None:
from tournament.consumers import tournament_manager
room = tournament_manager.get(tournament)
room.set_game_as_finished(self)
def get_players(self) -> list[User]:
return [game_player.player for game_player in GameMembersModel.objects.filter(game = self)]