295 lines
12 KiB
Python
295 lines
12 KiB
Python
import mimetypes
|
|
import operator
|
|
from functools import reduce
|
|
|
|
from django.db.models import Prefetch, Q
|
|
from django.http import HttpResponse
|
|
from django.shortcuts import redirect
|
|
from django.utils.translation import ugettext_lazy as _
|
|
from rest_framework.decorators import detail_route, list_route
|
|
from rest_framework.exceptions import NotFound, ValidationError
|
|
from rest_framework.mixins import RetrieveModelMixin
|
|
from rest_framework.response import Response
|
|
from rest_framework.viewsets import GenericViewSet, ReadOnlyModelViewSet, ViewSet
|
|
|
|
from c3nav.mapdata.models import AccessRestriction, Building, Door, Hole, LocationGroup, Source, Space
|
|
from c3nav.mapdata.models.geometry.level import LevelGeometryMixin
|
|
from c3nav.mapdata.models.geometry.space import POI, Area, Column, LineObstacle, Obstacle, SpaceGeometryMixin, Stair
|
|
from c3nav.mapdata.models.level import Level
|
|
from c3nav.mapdata.models.locations import (Location, LocationGroupCategory, LocationRedirect, LocationSlug,
|
|
SpecificLocation)
|
|
from c3nav.mapdata.utils.models import get_submodels
|
|
|
|
|
|
def optimize_query(qs):
|
|
if issubclass(qs.model, SpecificLocation):
|
|
base_qs = LocationGroup.objects.select_related('category').only('id', 'titles', 'category')
|
|
qs = qs.prefetch_related(Prefetch('groups', queryset=base_qs))
|
|
return qs
|
|
|
|
|
|
class MapViewSet(ViewSet):
|
|
"""
|
|
Map API
|
|
/bounds/ returns the maximum bounds of the map
|
|
"""
|
|
|
|
@list_route(methods=['get'])
|
|
def bounds(self, request, *args, **kwargs):
|
|
return Response({
|
|
'bounds': Source.max_bounds(),
|
|
})
|
|
|
|
|
|
class MapdataViewSet(ReadOnlyModelViewSet):
|
|
def get_queryset(self):
|
|
qs = super().get_queryset()
|
|
if hasattr(qs.model, 'qs_for_request'):
|
|
return qs.model.qs_for_request(self.request)
|
|
return qs
|
|
|
|
def list(self, request, *args, **kwargs):
|
|
qs = optimize_query(self.get_queryset())
|
|
geometry = ('geometry' in request.GET)
|
|
if issubclass(qs.model, LevelGeometryMixin) and 'level' in request.GET:
|
|
if not request.GET['level'].isdigit():
|
|
raise ValidationError(detail={'detail': _('%s is not an integer.') % 'level'})
|
|
try:
|
|
level = Level.qs_for_request(request).get(pk=request.GET['level'])
|
|
except Level.DoesNotExist:
|
|
raise NotFound(detail=_('level not found.'))
|
|
qs = qs.filter(level=level)
|
|
if issubclass(qs.model, SpaceGeometryMixin) and 'space' in request.GET:
|
|
if not request.GET['space'].isdigit():
|
|
raise ValidationError(detail={'detail': _('%s is not an integer.') % 'space'})
|
|
try:
|
|
space = Space.qs_for_request(request).get(pk=request.GET['space'])
|
|
except Space.DoesNotExist:
|
|
raise NotFound(detail=_('space not found.'))
|
|
qs = qs.filter(space=space)
|
|
if issubclass(qs.model, LocationGroup) and 'category' in request.GET:
|
|
kwargs = {('pk' if request.GET['category'].isdigit() else 'name'): request.GET['category']}
|
|
try:
|
|
category = LocationGroupCategory.objects.get(**kwargs)
|
|
except LocationGroupCategory.DoesNotExist:
|
|
raise NotFound(detail=_('category not found.'))
|
|
qs = qs.filter(category=category)
|
|
if issubclass(qs.model, SpecificLocation) and 'group' in request.GET:
|
|
if not request.GET['group'].isdigit():
|
|
raise ValidationError(detail={'detail': _('%s is not an integer.') % 'group'})
|
|
try:
|
|
group = LocationGroup.objects.get(pk=request.GET['group'])
|
|
except LocationGroupCategory.DoesNotExist:
|
|
raise NotFound(detail=_('group not found.'))
|
|
qs = qs.filter(groups=group)
|
|
if qs.model == Level and 'on_top_of' in request.GET:
|
|
if request.GET['on_top_of'] == 'null':
|
|
qs = qs.filter(on_top_of__isnull=False)
|
|
else:
|
|
if not request.GET['on_top_of'].isdigit():
|
|
raise ValidationError(detail={'detail': _('%s is not null or an integer.') % 'on_top_of'})
|
|
try:
|
|
level = Level.objects.get(pk=request.GET['on_top_of'])
|
|
except Level.DoesNotExist:
|
|
raise NotFound(detail=_('level not found.'))
|
|
qs = qs.filter(on_top_of=level)
|
|
return Response([obj.serialize(geometry=geometry) for obj in qs.order_by('id')])
|
|
|
|
def retrieve(self, request, *args, **kwargs):
|
|
return Response(self.get_object().serialize())
|
|
|
|
@staticmethod
|
|
def list_types(models_list, **kwargs):
|
|
return Response([
|
|
model.serialize_type(**kwargs) for model in models_list
|
|
])
|
|
|
|
|
|
class LevelViewSet(MapdataViewSet):
|
|
""" Add ?on_top_of=<null or id> to filter by on_top_of, add ?group=<id> to filter by group. """
|
|
queryset = Level.objects.all()
|
|
|
|
@list_route(methods=['get'])
|
|
def geometrytypes(self, request):
|
|
return self.list_types(get_submodels(LevelGeometryMixin))
|
|
|
|
@detail_route(methods=['get'])
|
|
def svg(self, request, pk=None):
|
|
level = self.get_object()
|
|
response = HttpResponse(level.render_svg(request), 'image/svg+xml')
|
|
return response
|
|
|
|
|
|
class BuildingViewSet(MapdataViewSet):
|
|
""" Add ?geometry=1 to get geometries, add ?level=<id> to filter by level. """
|
|
queryset = Building.objects.all()
|
|
|
|
|
|
class SpaceViewSet(MapdataViewSet):
|
|
""" Add ?geometry=1 to get geometries, add ?level=<id> to filter by level, add ?group=<id> to filter by group. """
|
|
queryset = Space.objects.all()
|
|
|
|
@list_route(methods=['get'])
|
|
def geometrytypes(self, request):
|
|
return self.list_types(get_submodels(SpaceGeometryMixin))
|
|
|
|
|
|
class DoorViewSet(MapdataViewSet):
|
|
""" Add ?geometry=1 to get geometries, add ?level=<id> to filter by level. """
|
|
queryset = Door.objects.all()
|
|
|
|
|
|
class HoleViewSet(MapdataViewSet):
|
|
""" Add ?geometry=1 to get geometries, add ?space=<id> to filter by space. """
|
|
queryset = Hole.objects.all()
|
|
|
|
|
|
class AreaViewSet(MapdataViewSet):
|
|
""" Add ?geometry=1 to get geometries, add ?space=<id> to filter by space, add ?group=<id> to filter by group. """
|
|
queryset = Area.objects.all()
|
|
|
|
|
|
class StairViewSet(MapdataViewSet):
|
|
""" Add ?geometry=1 to get geometries, add ?space=<id> to filter by space. """
|
|
queryset = Stair.objects.all()
|
|
|
|
|
|
class ObstacleViewSet(MapdataViewSet):
|
|
""" Add ?geometry=1 to get geometries, add ?space=<id> to filter by space. """
|
|
queryset = Obstacle.objects.all()
|
|
|
|
|
|
class LineObstacleViewSet(MapdataViewSet):
|
|
""" Add ?geometry=1 to get geometries, add ?space=<id> to filter by space. """
|
|
queryset = LineObstacle.objects.all()
|
|
|
|
|
|
class ColumnViewSet(MapdataViewSet):
|
|
""" Add ?geometry=1 to get geometries, add ?space=<id> to filter by space. """
|
|
queryset = Column.objects.all()
|
|
|
|
|
|
class POIViewSet(MapdataViewSet):
|
|
""" Add ?geometry=1 to get geometries, add ?space=<id> to filter by space, add ?group=<id> to filter by group. """
|
|
queryset = POI.objects.all()
|
|
|
|
|
|
class LocationGroupCategoryViewSet(MapdataViewSet):
|
|
queryset = LocationGroupCategory.objects.all()
|
|
|
|
|
|
class LocationGroupViewSet(MapdataViewSet):
|
|
""" Add ?category=<id or name> to filter by category. """
|
|
queryset = LocationGroup.objects.all()
|
|
|
|
|
|
class LocationViewSet(RetrieveModelMixin, GenericViewSet):
|
|
"""
|
|
only accesses locations that have can_search or can_describe set to true.
|
|
add ?detailed=1 to show all attributes, add ?geometry=1 to show geometries, add ?group=<id> to filter by group
|
|
/{id}/ add ?show_redirect=1 to suppress redirects and show them as JSON.
|
|
/search/ only accesses locations that have can_search set to true. Add GET Parameter “s” to search.
|
|
"""
|
|
queryset = LocationSlug.objects.all()
|
|
lookup_field = 'slug'
|
|
|
|
def get_queryset(self, search=False, group=None):
|
|
queryset = super().get_queryset().order_by('id')
|
|
|
|
conditions = []
|
|
for model in get_submodels(Location):
|
|
if group is not None and not hasattr(model, 'groups'):
|
|
continue
|
|
related_name = model._meta.default_related_name
|
|
condition = Q(**{related_name+'__isnull': False})
|
|
if search:
|
|
condition &= Q(**{related_name+'__can_search': True})
|
|
else:
|
|
condition &= Q(**{related_name+'__can_search': True, related_name+'__can_describe': True})
|
|
if group is not None:
|
|
condition &= Q(**{related_name+'__groups': group})
|
|
# noinspection PyUnresolvedReferences
|
|
condition &= model.q_for_request(self.request, prefix=related_name+'__')
|
|
conditions.append(condition)
|
|
queryset = queryset.filter(reduce(operator.or_, conditions))
|
|
|
|
# prefetch locationgroups
|
|
base_qs = LocationGroup.objects.all().select_related('category')
|
|
for model in get_submodels(SpecificLocation):
|
|
queryset = queryset.prefetch_related(Prefetch(model._meta.default_related_name + '__groups',
|
|
queryset=base_qs))
|
|
|
|
return queryset
|
|
|
|
def list(self, request, *args, **kwargs):
|
|
detailed = 'detailed' in request.GET
|
|
geometry = 'geometry' in request.GET
|
|
|
|
group = None
|
|
if 'group' in request.GET:
|
|
if not request.GET['group'].isdigit():
|
|
raise ValidationError(detail={'detail': _('%s is not an integer.') % 'group'})
|
|
try:
|
|
group = LocationGroup.objects.get(pk=request.GET['group'])
|
|
except LocationGroupCategory.DoesNotExist:
|
|
raise NotFound(detail=_('group not found.'))
|
|
|
|
queryset = self.get_queryset(group=group)
|
|
|
|
return Response([obj.get_child().serialize(include_type=True, detailed=detailed, geometry=geometry)
|
|
for obj in queryset])
|
|
|
|
def retrieve(self, request, slug=None, *args, **kwargs):
|
|
result = Location.get_by_slug(slug, self.get_queryset())
|
|
if result is None:
|
|
raise NotFound
|
|
result = result.get_child()
|
|
if isinstance(result, LocationRedirect):
|
|
if 'show_redirects' in request.GET:
|
|
return Response(result.serialize(include_type=True))
|
|
return redirect('../'+result.target.slug) # todo: why does redirect/reverse not work here?
|
|
return Response(result.serialize(include_type=True, detailed='detailed' in request.GET))
|
|
|
|
@list_route(methods=['get'])
|
|
def types(self, request):
|
|
return MapdataViewSet.list_types(get_submodels(Location), geomtype=False)
|
|
|
|
@list_route(methods=['get'])
|
|
def redirects(self, request):
|
|
return Response([obj.serialize(include_type=False) for obj in LocationRedirect.objects.all().order_by('id')])
|
|
|
|
@list_route(methods=['get'])
|
|
def search(self, request):
|
|
detailed = 'detailed' in request.GET
|
|
geometry = 'geometry' in request.GET
|
|
search = request.GET.get('s')
|
|
|
|
queryset = self.get_queryset(search=True)
|
|
|
|
if not search:
|
|
return Response([obj.get_child().serialize(include_type=True, detailed=detailed, geometry=geometry)
|
|
for obj in queryset])
|
|
|
|
words = search.lower().split(' ')[:10]
|
|
results = queryset
|
|
for word in words:
|
|
results = [r for r in results if (word in r.title.lower() or (r.slug and word in r.slug.lower()))]
|
|
# todo: rank results
|
|
return Response([obj.get_child().serialize(include_type=True, detailed=detailed, geometry=geometry)
|
|
for obj in results])
|
|
|
|
|
|
class SourceViewSet(MapdataViewSet):
|
|
queryset = Source.objects.all()
|
|
|
|
@detail_route(methods=['get'])
|
|
def image(self, request, pk=None):
|
|
return self._image(request, pk=pk)
|
|
|
|
def _image(self, request, pk=None):
|
|
source = self.get_object()
|
|
return HttpResponse(open(source.filepath, 'rb'), content_type=mimetypes.guess_type(source.name)[0])
|
|
|
|
|
|
class AccessRestrictionViewSet(MapdataViewSet):
|
|
queryset = AccessRestriction.objects.all()
|