Django: Transform dict of queryset



我使用以下代码将我的数据从字典中取出。

test = self.events_max_total_gross()
events = organizer.events.all()
    for event in events:
        test.get(event.pk, {}).values()
        [...]

我使用此查询集来获取数据。我的问题是:最后的转换是否有意义,或者是否有更好的方法来访问字典(无需先转换它(。由于我有几个这样的方法,我的方法似乎不遵循 DRY 原则。

def events_max_total_gross(self):
    events_max_total_gross = (
        Event.objects.filter(
            organizer__in=self.organizers,
            status=EventStatus.LIVE
        )
        .annotate(total_gross=Sum(F('tickets__quantity') * F('tickets__price_gross')))
        .values('pk', 'total_gross')
    )
    """
    Convert this
    [
        {
            'pk': 2,
            'total_gross': 12345
        },
        {
            'pk': 3,
            'total_gross': 54321
        },
        ...
    ]
    to this:
    {
        2: {
            'total_gross': 12345,
        },
        3: {
            'total_gross': 12345,
        }
        ...
    }
    """
    events_max_total_gross_transformed = {}
    for item in events_max_total_gross:
        events_max_total_gross_transformed.setdefault(
            item['pk'], {}
        ).update({'total_gross': item['total_gross']})
    return events_max_total_gross_transformed

使用:

transformed = { 
    v['pk']: { 'total_gross': v['total_gross'] } for v in events_max_total_gross 
}

这称为python字典理解。如果你想要教程或示例,谷歌这个术语。

如果我理解正确,那么对于每个组织者的活动,您都需要total_gross,您可以像这样查询,而不是循环events_max_total_gross事件和。

  1. 首先获取该特定组织者的所有事件。

    event_ids = Event.objects.filter(organizer=organizer).values_list('id',flat=True)

  2. 然后运行以下命令:

    Event.objects.filter( id__in=event_ids, organizer__in=self.organizers, status=EventStatus.LIVE ) .annotate(total_gross=Sum(F('tickets__quantity') * F('tickets__price_gross'))) .values('pk', 'total_gross') )

这样,您将保存字典的转换并再次循环。

如果你因为其他一些要求而需要这样做,那么你可以使用python字典理解:

events_max_total_gross_transformed = { event['pk']: { 'total_gross': event['total_gross'] } for event in events_max_total_gross }

但是,由于您有几个这样的模型,您可能还想看看代理模型,这可能会有所帮助。在这里,您可以编写管理器函数来帮助您进行查询。

最新更新