diff --git a/core/api/billPriceSum.py b/core/api/billPriceSum.py index 0bd067b3a80665c4d8a490b80a309240224ec365..7db6754a6f41dc6c74a3cf031032db8ae706714b 100644 --- a/core/api/billPriceSum.py +++ b/core/api/billPriceSum.py @@ -14,6 +14,8 @@ class BillSumPriceListView(generics.ListAPIView): def get(self, request, *args, **kwargs): + item = request.GET.get('item') + categorieModel = { 1: "Food", 2: "Groceries", @@ -35,6 +37,10 @@ class BillSumPriceListView(generics.ListAPIView): for i in range(1,12): qs = BillSearchQuerySet(Bill).searchCategories(query=i, user=user) + if item == 'today': + qs = qs.searchToday(user=user) + if item == 'month': + qs = qs.searchThisMonth(user=user) total_price = qs.aggregate(Sum('price')) data[categorieModel[i]] = total_price['price__sum'] return response.Response(data)