Django中annotate与distinct(fields)联用报错的解决方案咨询

问题原因 #

Django 4.2.7的ORM不支持**annotate()和带指定字段的distinct(fields)**组合使用，这就是触发NotImplementedError的直接原因。你的查询需求是：统计指定时间段内，当前酒店中处于「未入住/已入住」状态的预订所占用的、未被删除的房间，按房间类型分组，且每个房间仅被统计一次。

修改方案 #

方案一：拆分查询，先获取唯一房间再统计类型

先筛选出所有符合条件的唯一房间ID，再基于Room模型分组统计各类型数量，完全避开annotate与distinct(fields)的冲突：

# 1. 获取所有被占用的唯一房间ID
occupied_room_ids = RoomBooking.objects.filter(
    (Q(booking__start_date__range=(lookup_start_date, lookup_end_date))
     | Q(booking__end_date__range=(lookup_start_date, lookup_end_date))
     | Q(booking__start_date__lt=lookup_start_date, booking__end_date__gt=lookup_end_date)),
    booking__hotel=self.hotel,
    booking__guest_status__in=[settings.NOT_CHECKED, settings.CHECKED_IN],
    room__is_deleted=False
).values_list('room_id', flat=True).distinct()

# 2. 基于房间ID统计各类型数量
booked_rooms = Room.objects.filter(
    id__in=occupied_room_ids,
    hotel=self.hotel
).values('room_type').annotate(count=Count('id')).order_by('room_type')

方案二：使用Count的distinct参数替代外层distinct(fields)

直接调整原查询，去掉distinct('room__room_number')，改用Count的distinct=True参数实现去重统计，逻辑更紧凑：

booked_rooms = RoomBooking.objects.filter(
    (Q(booking__start_date__range=(lookup_start_date, lookup_end_date))
     | Q(booking__end_date__range=(lookup_start_date, lookup_end_date))
     | Q(booking__start_date__lt=lookup_start_date, booking__end_date__gt=lookup_end_date)),
    booking__hotel=self.hotel,
    booking__guest_status__in=[settings.NOT_CHECKED, settings.CHECKED_IN],
    room__is_deleted=False
).select_related('room').values(room_type=F('room__room_type')).annotate(
    count=Count('room__room_number', distinct=True)
).order_by('room_type')

方案说明 #

• 方案一逻辑清晰，拆分后的查询更易维护，适合复杂场景；
• 方案二更简洁，利用ORM内置的统计去重能力，性能与方案一相近，取决于数据库的查询优化。

内容的提问来源于stack exchange，提问作者Abk