Django分组查询annotate可能遇到的坑

Aria39

已于 2023-03-07 16:37:58 修改

阅读量398

点赞数

文章标签： django python 后端

于 2023-03-07 16:31:32 首次发布

本文链接：https://blog.csdn.net/qq_39117101/article/details/129384510

版权

文章讲述了在使用DjangoORM进行分组查询时，由于model的meta类中ordering属性的影响导致查询结果未按预期聚合的问题。通过示例展示了如何修正查询，以及如何在分组后根据计数结果进行排序。解决方案是显式设置order_by以匹配分组字段，并在需要时添加额外的排序条件。

摘要由CSDN通过智能技术生成

Django分组查询annotate可能遇到的坑

Django ORM的分组查询一般是values（或values_list）和annotate联用，values里面声明以什么字段分组，annotate 执行分组，这其中可能会有个坑

model大概长这样

class AdsEvent(models.Model):
    id = models.CharField(primary_key=True, max_length=32)
    dstip = models.GenericIPAddressField()
    attack_type = models.CharField(max_length=20, blank=True, null=True)
    start_time = models.DateTimeField(blank=True, null=True)
    end_time = models.DateTimeField(blank=True, null=True)
    total_in_bits = models.BigIntegerField(blank=True, null=True)
    total_in_packets = models.BigIntegerField(blank=True, null=True)
    total_drop_bits = models.BigIntegerField(blank=True, null=True)
    total_drop_packets = models.BigIntegerField(blank=True, null=True)
    attack_port = models.BigIntegerField(blank=True, null=True)
    objects = ScrubbingEnvironmentADSEventManager()

    class Meta:
        managed = False
        db_table = 't_ads_event'
        ordering = ('-end_time', '-start_time')

功能需求是要返回每种攻击类型的攻击次数，那我就根据攻击类型分组，统计个数

ret = queryset.values("attack_type").annotate(value=Count('id'))

嗯，看似没问题
但是返回的结果根本没聚合
在这里插入图片描述
why？？？
可以把ORM转成sql打印出来看看

print(queryset.values("attack_type").annotate(value=Count('id')).query)

对应的sql：

SELECT "t_ads_event"."attack_type", COUNT("t_ads_event"."id") AS "value" 
FROM "t_ads_event" 
GROUP BY "t_ads_event"."attack_type", "t_ads_event"."end_time", "t_ads_event"."start_time" 
ORDER BY "t_ads_event"."end_time" DESC, "t_ads_event"."start_time" DESC

等等，为啥分组条件里会有开始时间和结束时间？？？
看下model的meta类，原来ordering被指定成默认时间降序

class Meta:
        managed = False
        db_table = 't_ads_event'
        ordering = ('-end_time', '-start_time')

解决方法：查询时把排序参数改成用来分组的参数

ret = queryset.order_by('attack_type').values("attack_type").annotate(value=Count('id'))

对应的sql：

SELECT "t_ads_event"."attack_type", COUNT("t_ads_event"."id") AS "value" FROM "t_ads_event" 
GROUP BY "t_ads_event"."attack_type" 
ORDER BY "t_ads_event"."attack_type" ASC

ok了
在这里插入图片描述

但还存在个问题，如果我的需求是要按照攻击次数从大到小的顺序返回咋办？
那就加个排序喽

ret = queryset.order_by('attack_type').values("attack_type").annotate(value=Count('id')).order_by('-value')

或者这样

ret = queryset.values("attack_type").annotate(value=Count('id')).order_by('-value')

对应sql都是：

SELECT "t_ads_event"."attack_type", COUNT("t_ads_event"."id") AS "value" 
FROM "t_ads_event" 
GROUP BY "t_ads_event"."attack_type" 
ORDER BY "value" DESC