使用django count或values_list counter,哪个更好?

问题描述:

我写一个视图中导出数据,我的模型是这样的:使用django count或values_list counter,哪个更好?

class Event(models.Model): 
    KIND_CHOICES = (('doing', 'doing'), 
        ('done', 'done'), 
        ('cancel', 'cancel')) 
    created_at = models.DateTimeField(auto_now_add=True) 
    created_by = models.ForeignKey('auth.User') 
    kind = models.CharField(max_length=20, choices=KIND_CHOICES) 

事件是一种三种,即每个用户都可以每月有3〜10个事件,首先,我查询这个月的活动:

events_this_month = Event.objects.filter(created_at__year=2013, 
             created_at__month=5) 

然后找到所有用户:

users = User.objects.all() 

我导出数据是这样的:

for user in users: 
    # 1000 users with 5 events each 
    user_events = events_this_month.filter(created_by=user) 
    doing_count = user_events.filter(kind='doing').count() 
    done_count = user_events.filter(kind='done').count() 
    cancel_count = user.events.filter(kind='cancel').count() 

    append_to_csv([user.username, doing_count, done_count, cancel_count]) 

然后我试图使用collections.Counter,我认为这将减少数SQL倍(实际上它从3000+降到1200):

for user in users: 
    user_events = events_this_month.filter(created_by=user) 
    counter = Counter(user_events.values_list('kind', flat=True)) 
    doing_count = counter['doing'] 
    done_count = counter['done'] 
    cancel_count = counter['cancel'] 

    ... 

哪种方法更好

更多惯用如何计算这样的数据更有效有效

+0

您应该可以使用[annotate](https://docs.djangoproject.com/en/1.5/ref/models/querysets/#annotate)功能。 – Wessie 2013-05-11 15:44:38

这不是测试,但这个想法是由user组,然后按kind

from django.db.models import Count 

events_this_month = Event.objects.values('created_by', 'kind') \ 
         .filter(created_at__year=2013, created_at__month=5) \ 
         .annotate(cc=Count('kind')) 

让我知道,如果这个工程,我没有测试过这一点。