diff --git a/games/models.py b/games/models.py index 9a23ad9..3c95a6b 100644 --- a/games/models.py +++ b/games/models.py @@ -2,12 +2,19 @@ from django.db import models # Create your models here. class GameModel(models.Model): + + finished = models.BooleanField(default=False) + started = models.BooleanField(default=False) + winner_id = models.IntegerField() def create(self, users_id: [int]): self.save() for user_id in users_id: GameMembersModel(game_id=self.pk, member_id=user_id).save() return self.pk + + def get_users(self): + return list(GameMembersModel.objects.filter(self.pk)) class GameMembersModel(models.Model): game_id = models.IntegerField()