team-3/src/c3nav/editor/views.py

from functools import wraps

from django.conf import settings
from django.core.exceptions import PermissionDenied
from django.http.response import Http404
from django.shortcuts import get_object_or_404, redirect, render
from django.urls import reverse

from c3nav.mapdata.models import Section
from c3nav.mapdata.models.base import EDITOR_FORM_MODELS


def sidebar_view(func):
    @wraps(func)
    def with_ajax_check(request, *args, **kwargs):
        if not request.is_ajax():
            return render(request, 'editor/map.html', {})
        return func(request, *args, **kwargs)
    return with_ajax_check


@sidebar_view
def main_index(request, section=None):
    if section is None:
        first_section = Section.objects.first()
        if first_section:
            return render(request, 'editor/redirect.html', {
                'target': reverse('editor.index', kwargs={'section': first_section.id})
            })
        else:
            return render(request, 'editor/redirect.html', {
                'target': reverse('editor.sections')
            })
    else:
        section = get_object_or_404(Section, pk=section)

    return render(request, 'editor/index.html', {
        'sections': Section.objects.all(),
        'section': section,
        'section_url': 'editor.index',
    })


@sidebar_view
def sections_list(request):
    return render(request, 'editor/sections.html', {})


def list_mapitemtypes(request, section):
    section = get_object_or_404(Section, pk=section)

    def get_item_count(mapitemtype):
        if hasattr(mapitemtype, 'section'):
            return filter_queryset_by_access(request, mapitemtype.objects.filter(section=section)).count()
        return 0

    return render(request, 'editor/mapitemtypes.html', {
        'section': section,
        'mapitemtypes': [
            {
                'name': name,
                'title': mapitemtype._meta.verbose_name_plural,
                'has_section': hasattr(mapitemtype, 'section'),
                'count': get_item_count(mapitemtype),
            } for name, mapitemtype in EDITOR_FORM_MODELS.items()
        ],
    })


def list_mapitems(request, mapitem_type, section=None):
    mapitemtype = EDITOR_FORM_MODELS.get(mapitem_type)
    if mapitemtype is None:
        raise Http404('Unknown mapitemtype.')

    has_section = hasattr(mapitemtype, 'section')
    if has_section and section is None:
        raise Http404('Missing section.')
    elif not has_section and section is not None:
        return redirect('editor.mapitems', mapitem_type=mapitem_type)

    queryset = mapitemtype.objects.all().order_by('name')

    if section is not None:
        section = get_object_or_404(Section, section)
        if hasattr(mapitemtype, 'section'):
            queryset = queryset.filter(section=section)

    queryset = filter_queryset_by_access(request, queryset)

    if issubclass(mapitemtype, AreaLocation):
        queryset = sorted(queryset, key=AreaLocation.get_sort_key)

    return render(request, 'editor/mapitems.html', {
        'mapitem_type': mapitem_type,
        'title': mapitemtype._meta.verbose_name_plural,
        'has_section': section is not None,
        'has_altitude': hasattr(mapitemtype, 'altitude'),
        'section': section.id,
        'items': queryset,
    })


def edit_mapitem(request, mapitem_type, name=None):
    mapitemtype = EDITOR_FORM_MODELS.get(mapitem_type)
    if mapitemtype is None:
        raise Http404()

    mapitem = None
    if name is not None:
        # Edit existing map item
        mapitem = get_object_or_404(mapitemtype, name=name)
        if not can_access(request, mapitem):
            raise PermissionDenied

    new = mapitem is None
    orig_name = mapitem.name if mapitem is not None else None

    if request.method == 'POST':
        if mapitem is not None and request.POST.get('delete') == '1':
            # Delete this mapitem!
            if request.POST.get('delete_confirm') == '1':
                if not settings.DIRECT_EDITING:
                    # todo: suggest changes
                    raise NotImplementedError

                mapitem.delete()
                return render(request, 'editor/mapitem_success.html', {
                    'mapitem_type': mapitem_type
                })

            return render(request, 'editor/mapitem_delete.html', {
                'name': mapitem.name,
                'mapitem_type': mapitem_type,
                'path': request.path
            })

        form = mapitemtype.EditorForm(instance=mapitem, data=request.POST, request=request)
        if form.is_valid():
            # Update/create mapitem
            mapitem = form.save(commit=False)

            if form.titles is not None:
                mapitem.titles = {}
                for language, title in form.titles.items():
                    if title:
                        mapitem.titles[language] = title

            if not settings.DIRECT_EDITING:
                # todo: suggest changes
                raise NotImplementedError

            mapitem.save()
            form.save_m2m()

            return render(request, 'editor/mapitem_success.html', {
                'mapitem_type': mapitem_type
            })
    else:
        form = mapitemtype.EditorForm(instance=mapitem, request=request)

    return render(request, 'editor/mapitem.html', {
        'form': form,
        'mapitem_type': mapitem_type,
        'title': mapitemtype._meta.verbose_name,
        'has_geometry': hasattr(mapitemtype, 'geometry'),
        'name': orig_name,
        'geomtype': getattr(mapitemtype, 'geomtype', None),
        'path': request.path,
        'new': new
    })