From ec5af473b4ce2a9416eada81e345bd112273b7f9 Mon Sep 17 00:00:00 2001
From: Yirou Qiu <y2qiu@uwaterloo.ca>
Date: Thu, 23 Mar 2023 17:19:24 -0400
Subject: [PATCH] add date search
---
core/api/billPriceSum.py | 6 ++++++
1 file changed, 6 insertions(+)
diff --git a/core/api/billPriceSum.py b/core/api/billPriceSum.py
index 0bd067b..7db6754 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)
--
GitLab