Django-按日期和平均价值查询汇总

时间:2019-02-28 10:33:15

标签: django django-rest-framework django-queryset

我有一个应用,其中一个Gym与许多Surveys(多对多)相关联,而一个Survey有许多Answers

Answer具有数字value,并且与SurveyGym相关联。

models.py

class DateTimeModel(models.Model):
    creation_date = models.DateTimeField(verbose_name=_('Creation Date'), auto_now_add=True, db_index=True)
    edit_date = models.DateTimeField(verbose_name=_('Last Edit Date'), auto_now=True, db_index=True)
    ...


class Gym(DateTimeModel):
    name = models.CharField(max_length=250)
    ...


class Survey(DateTimeModel):
    text = models.CharField(max_length=500)
    valid_from = models.DateTimeField()
    valid_to = models.DateTimeField()
    gyms = models.ManyToManyField(Gym)
    ...


class Answer(DateTimeModel):
    value = models.IntegerField()
    survey = models.ForeignKey(Survey, on_delete=models.CASCADE)
    gym = models.ForeignKey(Gym, on_delete=models.PROTECT)
    ...

我需要获取特定健身房每天的选票分配。这很容易做到:

views.py

class VotesDistributionByDayViewSet(APIView):
    @staticmethod
    def get(request, gym_id, survey_id):
        votes_by_date = Answer.objects.filter(gym_id=gym_id, survey_id=survey_id)\
            .annotate(day=TruncDay('creation_date'))\
            .values("day")\
            .annotate(count=Count('gym_id'))\
            .order_by('day')

        return Response({
            'votes_by_gym': votes_by_date,
        })

这将返回以下响应(正确):

{
    "votes_by_gym": [
        {
            "day": "2018-06-11T00:00:00+02:00",
            "count": 15
        },
        {
            "day": "2018-06-12T00:00:00+02:00",
            "count": 6
        },
        {
            "day": "2018-06-13T00:00:00+02:00",
            "count": 17
        },
        {
            "day": "2018-06-14T00:00:00+02:00",
            "count": 12
        },
        ...

现在,我需要按天返回所有体育馆的平均答案数。我尝试了以下查询:

        avg_votes_by_date = Answer.objects\
            .exclude(gym_id=8) \
            .filter(survey_id=survey_id) \
            .annotate(day=TruncDay('creation_date')) \
            .values("day") \
            .annotate(count_answers=Count('gym_id')) \
            .annotate(count_gym=Count('gym_id', distinct=True)) \
            .order_by('day')

返回以下响应:

"avg_votes_by_date": [
        {
            "day": "2018-06-11T00:00:00+02:00",
            "count_answers": 15,
            "count_gym": 1
        },
        {
            "day": "2018-06-12T00:00:00+02:00",
            "count_answers": 6,
            "count_gym": 1
        },
        {
            "day": "2018-06-13T00:00:00+02:00",
            "count_answers": 17,
            "count_gym": 1
        },
        {
            "day": "2018-06-14T00:00:00+02:00",
            "count_answers": 12,
            "count_gym": 2
        },
        {
            "day": "2018-06-15T00:00:00+02:00",
            "count_answers": 29,
            "count_gym": 2
        },

如何添加带有count_answers / count_gym结果的第三个键?有没有更好的方法来获得这个平均值?

1 个答案:

答案 0 :(得分:2)

对于第二个查询,建议您在选定的值之间同时注释“ count_answers”和“ count_gym”,然后进行强制转换:

avg_votes_by_date = Answer.objects \
    .exclude(gym_id=8) \
    .filter(survey_id=survey_id) \
    .annotate(day=TruncDay('creation_date')) \
    .values('day') \
    .annotate(count_answers=Count('gym_id')) \
    .annotate(count_gym=Count('gym_id', distinct=True)) \
    .annotate(avg=Cast(F('count_answers'), FloatField()) / Cast(F('count_gym'), FloatField())) \
    .order_by('day')