From c0efa07144598504c90b7fa75857456c09063033 Mon Sep 17 00:00:00 2001 From: starnakin Date: Fri, 19 Jan 2024 13:29:57 +0100 Subject: [PATCH] games: add: is_in_game method --- profiles/models.py | 9 +++++++++ 1 file changed, 9 insertions(+) diff --git a/profiles/models.py b/profiles/models.py index 1362463..77fa73d 100644 --- a/profiles/models.py +++ b/profiles/models.py @@ -6,6 +6,8 @@ from django.dispatch import receiver from django.conf import settings from django.db.models import IntegerField +from games.consumers import game_manager + def upload_to(instance, filename: str): return f"./profiles/static/avatars/{instance.pk}.{filename.split('.')[1]}" @@ -13,6 +15,13 @@ def upload_to(instance, filename: str): 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): + 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 @receiver(post_save, sender=User) def on_user_created(sender, instance, created, **kwargs):