2025-summer/team-3
12
Fork 0
Code Issues Pull requests Projects Releases Activity

improve querys on mapdata API /locations/

Browse source
This commit is contained in:
Laura Klünder 2017-06-22 19:46:06 +02:00
parent f2e300415e
commit 37d95700d3
1 changed files with 13 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

16
src/c3nav/mapdata/api.py
View file

@ -1,4 +1,6 @@
import mimetypes import mimetypes
import operator
from functools import reduce
from itertools import chain from itertools import chain
from django.db.models import Prefetch, Q from django.db.models import Prefetch, Q
@ -152,9 +154,17 @@ class LocationViewSet(RetrieveModelMixin, GenericViewSet):
lookup_field = 'slug' lookup_field = 'slug'
def list(self, request, *args, **kwargs): def list(self, request, *args, **kwargs):
queryset = sorted(chain(*(optimize_query(model.objects.filter(Q(can_search=True) | Q(can_describe=True))) queryset = self.get_queryset()
for model in get_submodels(Location))), key=lambda obj: obj.id) queryset = queryset.filter(reduce(operator.or_, (Q(**{model._meta.default_related_name+'__isnull': False})
return Response([obj.serialize(include_type=True, detailed='detailed' in request.GET) for obj in queryset]) for model in get_submodels(Location))))
for model in get_submodels(Location):
if model == LocationGroup:
continue
queryset = queryset.prefetch_related(Prefetch(model._meta.default_related_name+'__groups',
queryset=LocationGroup.objects.only('id', 'titles')))
return Response([obj.get_child().serialize(include_type=True, detailed='detailed' in request.GET)
for obj in queryset])
def retrieve(self, request, slug=None, *args, **kwargs): def retrieve(self, request, slug=None, *args, **kwargs):
result = Location.get_by_slug(slug, self.get_queryset()) result = Location.get_by_slug(slug, self.get_queryset())