core: tournament

games/models.py

@@ -1,8 +1,9 @@
 from __future__ import annotations
 
 from django.db import models
+from django.db.models import QuerySet, CASCADE
 
-from django.db.models import QuerySet
+from profiles.models import ProfileModel
 
 import time
 
@@ -12,16 +13,15 @@ class GameModel(models.Model):
 
     finished = models.BooleanField(default = False)
     started = models.BooleanField(default = False)
-    winner_id = models.IntegerField(default = -1)
+    winner = models.ForeignKey(ProfileModel, on_delete=CASCADE, null=True, blank=True)
     start_timestamp = models.BigIntegerField(null = True, blank = True)
     stop_timestamp = models.BigIntegerField(null = True, blank = True)
     game_type = models.CharField(max_length = 60, default = "pong")
     
-    def create(self, game_type: str, players_id: list[int]):
-        self.game_type = game_type
+    def create(self, players: list[ProfileModel]):
         self.save()
-        for player_id in players_id:
-            GameMembersModel(game_id = self.pk, player_id = player_id).save()
+        for player in players:
+            GameMembersModel(game = self.pk, player=player).save()
         return self.pk
 
     def start(self):
@@ -35,8 +35,8 @@ class GameModel(models.Model):
         self.stop_timestamp = round(time.time() * 1000, 1)
         self.save()
     
-    def get_players_id(self):
-        return [game_player.player_id for game_player in GameMembersModel.objects.filter(game_id = self.pk)]
+    def get_players(self) -> list[ProfileModel]:
+        return [game_player.player for game_player in GameMembersModel.objects.filter(game = self)]
     
     def get_score_by_player_id(self, player_id: int) -> list[int]:
         query: QuerySet = GameGoalModel.objects.filter(game_id = self.pk, player_id = player_id)
@@ -55,11 +55,11 @@ class GameModel(models.Model):
         return timestamp
 
 class GameMembersModel(models.Model):
-    game_id = models.IntegerField()
-    player_id = models.IntegerField()
+    game = models.ForeignKey(GameModel, on_delete=CASCADE)
+    player = models.ForeignKey(ProfileModel, on_delete=CASCADE)
     
 class GameGoalModel(models.Model):
     
-    game_id = models.IntegerField()
-    player_id = models.IntegerField()
+    game = models.ForeignKey(GameModel, on_delete=CASCADE)
+    player = models.ForeignKey(ProfileModel, on_delete=CASCADE)
     timestamp = models.IntegerField()