diff --git a/profiles/models.py b/profiles/models.py index 77fa73d..8239e51 100644 --- a/profiles/models.py +++ b/profiles/models.py @@ -16,12 +16,12 @@ class ProfileModel(models.Model): user = models.ForeignKey(User, on_delete=models.CASCADE) avatar_url = models.ImageField(upload_to=upload_to, default="./profiles/static/avatars/default.avif") #blank=True, null=True) - def is_in_game(self): + def get_game(self) -> int: for game in game_manager._game_list: for player in game.get_players_connected(): if (player.user_id == self.user.pk): - return True - return False + return game.game_id + return None @receiver(post_save, sender=User) def on_user_created(sender, instance, created, **kwargs):