89 lines
3.5 KiB
Python
89 lines
3.5 KiB
Python
import json
|
|
import uuid
|
|
from collections import OrderedDict
|
|
|
|
from django.conf import settings
|
|
from django.forms import CharField, ModelForm
|
|
from django.forms.models import ModelChoiceField
|
|
from django.forms.widgets import HiddenInput
|
|
from shapely.geometry.geo import mapping
|
|
|
|
from c3nav.mapdata.models import Package
|
|
from c3nav.mapdata.models.geometry import Area, Building, Door, Obstacle
|
|
from c3nav.mapdata.permissions import get_unlocked_packages
|
|
|
|
|
|
class MapitemFormMixin(ModelForm):
|
|
def __init__(self, *args, request=None, **kwargs):
|
|
self.request = request
|
|
super().__init__(*args, **kwargs)
|
|
creating = not self.instance.pk
|
|
|
|
# disable name on non-direct editing
|
|
if not creating and not settings.DIRECT_EDITING:
|
|
self.fields['name'].disabled = True
|
|
|
|
if creating and self._meta.model in (Door, Obstacle, ):
|
|
self.fields['name'].initial = uuid.uuid4()
|
|
|
|
# restrict package choices and field_name
|
|
if not creating:
|
|
if not settings.DIRECT_EDITING:
|
|
self.fields['package'].widget = HiddenInput()
|
|
self.fields['package'].disabled = True
|
|
self.initial['package'] = self.instance.package.name
|
|
elif not settings.DIRECT_EDITING:
|
|
unlocked_packages = get_unlocked_packages(request)
|
|
if len(unlocked_packages) == 1:
|
|
self.fields['package'].widget = HiddenInput()
|
|
self.fields['package'].initial = next(iter(unlocked_packages))
|
|
self.fields['package'].disabled = True
|
|
else:
|
|
self.fields['package'] = ModelChoiceField(
|
|
queryset=Package.objects.filter(name__in=unlocked_packages),
|
|
)
|
|
self.fields['package'].to_field_name = 'name'
|
|
|
|
# hide level widget and set field_name
|
|
self.fields['level'].widget = HiddenInput()
|
|
self.fields['level'].to_field_name = 'name'
|
|
if not creating:
|
|
self.initial['level'] = self.instance.level.name
|
|
|
|
# hide geometry widget
|
|
self.fields['geometry'].widget = HiddenInput()
|
|
if not creating:
|
|
self.initial['geometry'] = json.dumps(mapping(self.instance.geometry), separators=(',', ':'))
|
|
|
|
# parse titles
|
|
self.titles = None
|
|
if hasattr(self.instance, 'titles'):
|
|
titles = OrderedDict((lang_code, '') for lang_code, language in settings.LANGUAGES)
|
|
if self.instance is not None and self.instance.pk:
|
|
titles.update(self.instance.titles)
|
|
|
|
language_titles = dict(settings.LANGUAGES)
|
|
for language in titles.keys():
|
|
new_title = self.data.get('title_' + language)
|
|
if new_title is not None:
|
|
titles[language] = new_title
|
|
self.fields['title_' + language] = CharField(label=language_titles.get(language, language),
|
|
required=False,
|
|
initial=titles[language].strip(), max_length=50)
|
|
self.titles = titles
|
|
|
|
|
|
def create_editor_form(mapdata_model, add_fields=None):
|
|
class EditorForm(MapitemFormMixin, ModelForm):
|
|
class Meta:
|
|
model = mapdata_model
|
|
fields = ['name', 'package', 'level', 'geometry'] + (add_fields if add_fields is not None else [])
|
|
|
|
mapdata_model.EditorForm = EditorForm
|
|
|
|
|
|
def create_editor_forms():
|
|
create_editor_form(Building)
|
|
create_editor_form(Area)
|
|
create_editor_form(Obstacle, ['height'])
|
|
create_editor_form(Door)
|