Django Foregin Key Count

user_15078250 0
Viewed 31

I'm trying to get a count of the number of Users that are connected to a Room. A User can be connected to only one Room, a Room can have many Users.

My models look like this

class User(AbstractBaseUser, PermissionsMixin):
    """Database model for users"""
    screen_name = models.CharField(max_length=255, unique=True)
    room = models.ForeignKey(
        Room, related_name='room', on_delete=models.CASCADE
    )
    ...
class Room(models.Model):
    """Database model for rooms"""
    name = models.CharField(max_length=100)
    is_full = models.BooleanField(default=False)
    ...

What I'm trying to do is lock any instance of a Room out once a certain number of Users have been assigned to it. How do I go about finding the number of Users connected to a given room? Pseudo code of what I'm trying.

my_room.users.count >= max_users:
   my_room.is_full = True

Thank you in advance.

foreign-keys django-models python
2 Answers

To get number of foreign elements you can do it like that:

User.objects.filter(room=my_room).count()
user_75511840 
     from django.db.models import Count

     def get_queryset(self):  
        queryset = User.objects.filter(room=room_id) # if you want to filter for one room
        queryset = User.objects.all() # if you want to have a list of all rooms   
        return queryset.values('room').annotate(Count('id', distinct=True)).count() # will count unique rooms
user_106126220
Related