klmp200/Sith
1
0
Fork 0
mirror of https://github.com/ae-utbm/sith.git synced 2024-11-22 14:13:21 +00:00
Code Issues Packages Projects Releases Wiki Activity

pedagogy: add filters to search api

Browse Source
This commit is contained in:
Antoine Bartuccio 2019-06-19 01:26:11 +02:00
parent 2cbef2babc
commit 502ae09523
Signed by: klmp200
GPG Key ID: E7245548C53F904B
1 changed files with 18 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
pedagogy/views.py
View File

@ -150,9 +150,10 @@ class UVListView(CanViewMixin, CanCreateUVFunctionMixin, ListView):
template_name = "pedagogy/guide.jinja" template_name = "pedagogy/guide.jinja"
def get(self, *args, **kwargs): def get(self, *args, **kwargs):
resp = super(UVListView, self).get(*args, **kwargs)
if not self.request.GET.get("json", None): if not self.request.GET.get("json", None):
# Return normal full template response # Return normal full template response
return super(UVListView, self).get(*args, **kwargs) return resp
# Return serialized response # Return serialized response
return HttpResponse( return HttpResponse(
@ -163,8 +164,22 @@ class UVListView(CanViewMixin, CanCreateUVFunctionMixin, ListView):
def get_queryset(self): def get_queryset(self):
query = self.request.GET.get("query", None) query = self.request.GET.get("query", None)
additional_filters = {}
for filter_type in ["credit_type", "language", "department"]:
arg = self.request.GET.get(filter_type, None)
if arg:
additional_filters[filter_type] = arg
semester = self.request.GET.get("semester", None)
if semester:
if semester in ["AUTUMN", "SPRING"]:
additional_filters["semester__in"] = [semester, "AUTOMN_AND_SPRING"]
else:
additional_filters["semester"] = semester
if not query: if not query:
return super(UVListView, self).get_queryset() return super(UVListView, self).get_queryset().filter(**additional_filters)
try: try:
queryset = ( queryset = (
@ -179,6 +194,7 @@ class UVListView(CanViewMixin, CanCreateUVFunctionMixin, ListView):
super(UVListView, self) super(UVListView, self)
.get_queryset() .get_queryset()
.filter(id__in=([o.object.id for o in queryset])) .filter(id__in=([o.object.id for o in queryset]))
.filter(**additional_filters)
) )