/api/geometries: fix broken prefetch_related

This commit is contained in:
Laura Klünder 2016-12-06 22:55:15 +01:00
parent 32d6033301
commit 2a13d7ec93

View file

@ -63,7 +63,12 @@ class GeometryViewSet(ViewSet):
if names:
queryset = queryset.filter(name__in=names)
queryset = filter_queryset_by_package_access(request, queryset)
queryset.prefetch_related('package', 'level').order_by('name')
queryset = queryset.order_by('name')
print(mapitemtype)
for field_name in ('package', 'level', 'levels', 'crop_to_level', 'elevator'):
if hasattr(mapitemtype, field_name):
print(field_name)
queryset = queryset.prefetch_related(field_name)
results.extend(sum((obj.to_geojson() for obj in queryset), []))
return Response(results)