team-3/src/c3nav/editor/views.py
2017-05-29 10:17:06 +02:00

276 lines
9.6 KiB
Python

from functools import wraps
from django.apps import apps
from django.conf import settings
from django.core.exceptions import FieldDoesNotExist, PermissionDenied
from django.http import HttpResponseRedirect
from django.shortcuts import get_object_or_404, redirect, render
from django.urls import reverse
from django.utils.translation import ugettext_lazy as _
from django.views.decorators.cache import never_cache
from c3nav.mapdata.models import Section, Space
from c3nav.mapdata.models.base import EDITOR_FORM_MODELS
def sidebar_view(func):
@wraps(func)
def with_ajax_check(request, *args, **kwargs):
response = func(request, *args, **kwargs)
if request.is_ajax() or 'ajax' in request.GET:
if isinstance(response, HttpResponseRedirect):
return render(request, 'editor/redirect.html', {'target': response['location']})
return response
return render(request, 'editor/map.html', {'content': response.content})
return never_cache(with_ajax_check)
def child_model(model_name, kwargs=None, parent=None):
model = apps.get_model('mapdata', model_name)
related_name = model._meta.default_related_name
return {
'title': model._meta.verbose_name_plural,
'url': reverse('editor.'+related_name+'.list', kwargs=kwargs),
'count': None if parent is None else getattr(parent, related_name).count(),
}
@sidebar_view
def main_index(request):
return render(request, 'editor/index.html', {
'sections': Section.objects.all(),
'child_models': [
child_model('LocationGroup'),
child_model('Source'),
],
})
@sidebar_view
def section_detail(request, pk):
section = get_object_or_404(Section, pk=pk)
return render(request, 'editor/section.html', {
'sections': Section.objects.all(),
'section': section,
'section_url': 'editor.sections.detail',
'section_as_pk': True,
'child_models': [child_model(model_name, kwargs={'section': pk}, parent=section)
for model_name in ('Building', 'Space', 'Door', 'Hole')],
'geometry_url': '/api/editor/geometries/?section='+pk,
})
@sidebar_view
def space_detail(request, section, pk):
space = get_object_or_404(Space, section__id=section, pk=pk)
return render(request, 'editor/space.html', {
'section': space.section,
'space': space,
'child_models': [child_model(model_name, kwargs={'space': pk}, parent=space)
for model_name in ('Area', 'Stair', 'Obstacle', 'LineObstacle', 'Point')],
'geometry_url': '/api/editor/geometries/?space='+pk,
})
@sidebar_view
def edit(request, pk=None, model=None, section=None, space=None, explicit_edit=False):
model = EDITOR_FORM_MODELS[model]
obj = None
if pk is not None:
# Edit existing map item
kwargs = {'pk': pk}
if section is not None:
kwargs.update({'section__id': section})
elif space is not None:
kwargs.update({'space__id': space})
obj = get_object_or_404(model, **kwargs)
if False: # todo can access
raise PermissionDenied
elif section is not None:
section = get_object_or_404(Section, pk=section)
elif space is not None:
space = get_object_or_404(Space, pk=space)
new = obj is None
# noinspection PyProtectedMember
ctx = {
'path': request.path,
'pk': pk,
'model_name': model.__name__.lower(),
'model_title': model._meta.verbose_name,
'new': new,
'title': obj.title if obj else None,
}
try:
ctx.update({
'geomtype': model._meta.get_field('geometry').geomtype,
})
except FieldDoesNotExist:
pass
if model == Section:
ctx.update({
'section': obj,
'back_url': reverse('editor.index') if new else reverse('editor.sections.detail', kwargs={'pk': pk}),
'geometry_url': '/api/editor/geometries/?section='+pk,
})
elif model == Space and not new:
ctx.update({
'section': obj.section,
'back_url': reverse('editor.spaces.detail', kwargs={'section': obj.section.pk, 'pk': pk}),
'geometry_url': '/api/editor/geometries/?space='+pk,
})
elif model == Space and new:
ctx.update({
'section': section,
'back_url': reverse('editor.spaces.list', kwargs={'section': section.pk}),
'geometry_url': '/api/editor/geometries/?section='+str(section.pk),
})
elif hasattr(model, 'section'):
if obj:
section = obj.section
ctx.update({
'section': section,
'back_url': reverse('editor.sections.detail', kwargs={'pk': section.pk}),
'geometry_url': '/api/editor/geometries/?section='+str(section.pk),
})
elif hasattr(model, 'space'):
if obj:
space = obj.space
ctx.update({
'section': space.section,
'back_url': reverse('editor.spaces.detail', kwargs={'section': obj.space.section.pk, 'pk': space.pk}),
'geometry_url': '/api/editor/geometries/?space='+str(space.pk),
})
else:
kwargs = {}
if section is not None:
kwargs.update({'section': section})
elif space is not None:
kwargs.update({'space': space})
ctx.update({
'back_url': reverse('.'.join(request.resolver_match.url_name.split('.')[:-1]+['list']), kwargs=kwargs),
})
if request.method == 'POST':
if obj 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
obj.delete()
if model == Section:
return redirect(reverse('editor.index'))
elif model == Space:
return redirect(reverse('editor.spaces.list', kwargs={'section': obj.section.pk}))
return redirect(ctx['back_url'])
return render(request, 'editor/delete.html', ctx)
form = model.EditorForm(instance=obj, data=request.POST, request=request)
if form.is_valid():
# Update/create objects
obj = form.save(commit=False)
if form.titles is not None:
obj.titles = {}
for language, title in form.titles.items():
if title:
obj.titles[language] = title
if form.redirect_slugs is not None:
for slug in form.add_redirect_slugs:
obj.redirects.create(slug=slug)
for slug in form.remove_redirect_slugs:
obj.redirects.filter(slug=slug).delete()
if not settings.DIRECT_EDITING:
# todo: suggest changes
raise NotImplementedError
if section is not None:
obj.section = section
if space is not None:
obj.space = space
obj.save()
form.save_m2m()
return redirect(ctx['back_url'])
else:
form = model.EditorForm(instance=obj, request=request)
ctx.update({
'form': form,
})
return render(request, 'editor/edit.html', ctx)
@sidebar_view
def list_objects(request, model=None, section=None, space=None, explicit_edit=False):
model = EDITOR_FORM_MODELS[model]
if not request.resolver_match.url_name.endswith('.list'):
raise ValueError('url_name does not end with .list')
# noinspection PyProtectedMember
ctx = {
'path': request.path,
'model_name': model.__name__.lower(),
'model_title': model._meta.verbose_name,
'model_title_plural': model._meta.verbose_name_plural,
'explicit_edit': explicit_edit,
}
queryset = model.objects.all().order_by('id')
reverse_kwargs = {}
if section is not None:
reverse_kwargs['section'] = section
section = get_object_or_404(Section, pk=section)
queryset = queryset.filter(section=section)
ctx.update({
'back_url': reverse('editor.sections.detail', kwargs={'pk': section.pk}),
'back_title': _('back to section'),
'sections': Section.objects.all(),
'section': section,
'section_url': request.resolver_match.url_name,
'geometry_url': '/api/editor/geometries/?section='+str(section.pk),
})
elif space is not None:
reverse_kwargs['space'] = space
space = get_object_or_404(Space, pk=space)
queryset = queryset.filter(space=space)
ctx.update({
'section': space.section,
'back_url': reverse('editor.spaces.detail', kwargs={'section': space.section.pk, 'pk': space.pk}),
'back_title': _('back to space'),
'geometry_url': '/api/editor/geometries/?space='+str(space.pk),
})
else:
ctx.update({
'back_url': reverse('editor.index'),
'back_title': _('back to overview'),
})
edit_url_name = request.resolver_match.url_name[:-4]+('detail' if explicit_edit else 'edit')
for obj in queryset:
reverse_kwargs['pk'] = obj.pk
obj.edit_url = reverse(edit_url_name, kwargs=reverse_kwargs)
reverse_kwargs.pop('pk', None)
ctx.update({
'create_url': reverse(request.resolver_match.url_name[:-4] + 'create', kwargs=reverse_kwargs),
'objects': queryset,
})
return render(request, 'editor/list.html', ctx)