import collections from django.http import HttpResponseRedirect from django.shortcuts import get_object_or_404, render from django.utils.translation import gettext_lazy as _ from mptt.exceptions import InvalidMove from mptt.forms import MoveNodeForm from mptt.templatetags.mptt_tags import cache_tree_children from rest_framework.decorators import api_view, permission_classes from rest_framework.exceptions import APIException from rest_framework.permissions import IsAuthenticated from rest_framework.response import Response from poi_manager.models import PoiCategory, Poi from poi_manager.serializers import PoiSerializer, PoiCategorySerializer @api_view(['GET', ]) def get_poi_by_id(request, poi_id, format=None): try: poi_qs = Poi.objects.filter(id=poi_id).filter(enabled=True) serializer = PoiSerializer(poi_qs, many=True) return Response(serializer.data) except Exception as e: return Response(f"no data found error {e}", status=404) @api_view(['GET', ]) def poi_category_json(request, format=None): def recursive_node_to_dict(node): result = collections.OrderedDict() result['id'] = node.pk result['name'] = node.cat_name result['icon'] = node.icon children = [recursive_node_to_dict(c) for c in node.get_children()] if children: result['children'] = children return result root_nodes = cache_tree_children(PoiCategory.objects.filter(enabled=True)) dicts = [] for n in root_nodes: dicts.append(recursive_node_to_dict(n)) return Response(dicts) @api_view(['GET', ]) def poi_root_nodes(request, format=None): def recursive_node_to_dict(node): result = collections.OrderedDict() result['id'] = node.pk result['name'] = _(node.cat_name) result['icon'] = node.icon children = [recursive_node_to_dict(c) for c in node.get_children()] if children: result['nodes'] = children return result root_nodes = cache_tree_children(PoiCategory.objects.filter(enabled=True).filter(parent__isnull=True)) dicts = [] for n in root_nodes: dicts.append(recursive_node_to_dict(n)) return Response(dicts) @api_view(['GET', ]) @permission_classes((IsAuthenticated,)) def poi_json_tree(request, format=None): def recursive_node_to_dict(node): result = collections.OrderedDict() result['id'] = node.pk result['name'] = node.cat_name_en result['name_en'] = node.cat_name_en result['name_de'] = node.cat_name_de result['icon'] = node.icon if node.pk in (1, 2, 3, 4, 5): result['selectable'] = False # result['state'] = {"checked": False, "disabled": True, "expanded": False, "selected": False} children = [recursive_node_to_dict(c) for c in node.get_children()] if children: result['children'] = children return result root_nodes = cache_tree_children(PoiCategory.objects.filter(enabled=True)) dicts = [] for n in root_nodes: dicts.append(recursive_node_to_dict(n)) return Response(dicts) @api_view(['GET', ]) def get_poi_by_category(request, category_name, format=None): if request.method == 'GET': try: cats = PoiCategory.objects.filter(enabled=True).get(cat_name__contains=category_name) if cats: poi_qs = Poi.objects.filter(category=cats.id).filter(enabled=True) if poi_qs: serializer = PoiSerializer(poi_qs, many=True) return Response(serializer.data) except Exception as e: raise APIException(detail=e) @api_view(['GET', ]) def get_poi_by_cat_id(request, cat_id, format=None): """ Get all poi features for children and grandchildren of incomping parent cateogry id :param request: :param campus_id: id of campus as integer :param cat_id: :param format: :return: """ if request.method == 'GET': floor = request.GET.get('floor') if floor is not None: try: floor = float(floor) except ValueError: return Response({'error': 'Invalid floor value'}, status=400) poicat_qs = PoiCategory.objects.filter(enabled=True).get(pk=cat_id).get_descendants() poi_ids = [x.id for x in poicat_qs] qs_objs = Poi.objects.filter(category_id__in=poi_ids, enabled=True) if qs_objs: serializer = PoiSerializer(qs_objs, many=True) return Response(serializer.data) elif len(poi_ids) == 0: filter_params = {'category_id': cat_id, 'enabled': True} if floor is not None: filter_params['floor_num'] = floor qs = Poi.objects.filter(**filter_params) serializer = PoiSerializer(qs, many=True) return Response(serializer.data) else: return Response({'error': 'something booomed with category id : ' + cat_id}) @api_view(['GET', ]) def get_poi_by_cat_name(request, category_name, format=None): cats = PoiCategory.objects.filter(cat_name__exact=category_name).filter(enabled=True) # list = cats.get_descendants() # from itertools import chain # result_list = list(chain(page_list, article_list, post_list)) if cats: if len(cats) > 1: cat_ids = [] for cat in cats: cat_ids.append(cat.id) poi_qs = Poi.objects.filter(category__in=cat_ids).filter(enabled=True) else: poi_qs = Poi.objects.filter(category=cats[0].id).filter(enabled=True) else: # return Response({'error': 'No Poi found with the given category name: ' + category_name} ) return Response({'error': 'no category found with the given category name: ' + category_name}) if poi_qs: serializer = PoiSerializer(poi_qs, many=True) return Response(serializer.data) else: return Response({'error': 'sorry no poi entries found assigned to the category name : ' + category_name}) @api_view(['GET', ]) def get_poicat_by_id(request, cat_id, format=None): cats = PoiCategory.objects.filter(enabled=True).get(pk=cat_id) # todo poi_ids = [int(x) for x in request.GET.get('poi-cat-id', '').split(',')] if len(poi_ids) > 1: cats = PoiCategory.objects.filter(enabled=True).filter(pk__in=poi_ids) serializer = PoiCategorySerializer(cats) return Response(serializer.data) @api_view(['GET', ]) def poi_category_by_name(request, category_name, format=None): def recursive_node_to_dict(node): result = collections.OrderedDict() result['id'] = node.pk result['name'] = node.cat_name result['icon'] = node.icon_css_name children = [recursive_node_to_dict(c) for c in node.get_children()] if children: result['children'] = children return result root_nodes = cache_tree_children(PoiCategory.objects.filter(cat_name__icontains=category_name).filter(enabled=True)) dicts = [] if root_nodes: for n in root_nodes: dicts.append(recursive_node_to_dict(n)) return Response(dicts) else: return Response({'error': 'no category found with the name : ' + category_name}) @api_view(['GET', ]) def search_poi_by_name(request, poi_name, format=None, **kwargs): poi_qs = Poi.objects.filter(name__icontains=poi_name).filter(enabled=True) floor = request.GET.get('floor') if floor: poi_qs = Poi.objects.filter(name__icontains=poi_name).filter(floor_num=floor).filter(enabled=True) if poi_qs: att = poi_qs.values() poi_entries = [] for value in att: poi_entries.append(value) return Response(poi_entries) else: return Response({'error': 'something went wrong no POI with that name found'}) def move_category(request, category_pk): category = get_object_or_404(PoiCategory, pk=category_pk) if request.method == 'POST': form = MoveNodeForm(category, request.POST) if form.is_valid(): try: category = form.save() return HttpResponseRedirect(category.get_absolute_url()) except InvalidMove: pass else: form = MoveNodeForm(category) return render('poi/poi-form.html', { 'form': form, 'category': category, 'category_tree': PoiCategory.objects.all(), })