Django 外键计数

问题描述

我正在尝试计算连接到 Users 的 Room 的数量。一个 User 只能连接到一个 Room，一个 Room 可以连接多个 Users。

我的模型看起来像这样

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)
    ...

我正在尝试做的是在分配了一定数量的 Room 后将 User 的任何实例锁定。如何查找连接到给定房间的 User 的数量？我正在尝试的伪代码。

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

提前致谢。

解决方法

要获取外部元素的数量，您可以这样做：

User.objects.filter(room=my_room).count()

     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

django-models foreign-keys python