Skip to content
Snippets Groups Projects
Commit ec5af473 authored by Yirou Qiu's avatar Yirou Qiu
Browse files

add date search

parent c8a84d71
No related branches found
No related tags found
1 merge request!8Yrq/feature/bill sum price
...@@ -14,6 +14,8 @@ class BillSumPriceListView(generics.ListAPIView): ...@@ -14,6 +14,8 @@ class BillSumPriceListView(generics.ListAPIView):
def get(self, request, *args, **kwargs): def get(self, request, *args, **kwargs):
item = request.GET.get('item')
categorieModel = { categorieModel = {
1: "Food", 1: "Food",
2: "Groceries", 2: "Groceries",
...@@ -35,6 +37,10 @@ class BillSumPriceListView(generics.ListAPIView): ...@@ -35,6 +37,10 @@ class BillSumPriceListView(generics.ListAPIView):
for i in range(1,12): for i in range(1,12):
qs = BillSearchQuerySet(Bill).searchCategories(query=i, user=user) 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')) total_price = qs.aggregate(Sum('price'))
data[categorieModel[i]] = total_price['price__sum'] data[categorieModel[i]] = total_price['price__sum']
return response.Response(data) return response.Response(data)
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment