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

import json
import time
from collections import OrderedDict

from django.conf import settings
from django.forms import CharField, ModelForm, ValidationError
from django.forms.models import ModelChoiceField
from django.forms.widgets import HiddenInput
from django.utils.translation import ugettext_lazy as _
from shapely.geometry.geo import mapping

from c3nav.mapdata.models import Package
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:
            self.fields['name'].initial = hex(int(time.time()*1000000))[2:]

        # 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'

        if 'level' in self.fields:
            # 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

        if 'crop_to_level' in self.fields:
            # set field_name
            self.fields['crop_to_level'].to_field_name = 'name'
            if not creating and self.instance.crop_to_level is not None:
                self.initial['crop_to_level'] = self.instance.crop_to_level.name
            self.fields['crop_to_level'].queryset = self.fields['crop_to_level'].queryset.order_by('altitude')

        if 'levels' in self.fields:
            # set field_name
            self.fields['levels'].to_field_name = 'name'
            self.fields['levels'].queryset = self.fields['levels'].queryset.order_by('altitude')

        if 'geometry' in self.fields:
            # 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 clean_levels(self):
        levels = self.cleaned_data.get('levels')
        if len(levels) < 2:
            raise ValidationError(_('Please select at least two levels.'))
        return levels

    def clean(self):
        if 'geometry' in self.fields:
            if not self.cleaned_data.get('geometry'):
                raise ValidationError('Missing geometry.')


def create_editor_form(mapitemtype):
    possible_fields = ['name', 'package', 'altitude', 'level', 'intermediate', 'levels', 'geometry',
                       'elevator', 'button', 'crop_to_level']
    existing_fields = [field for field in possible_fields if hasattr(mapitemtype, field)]

    class EditorForm(MapitemFormMixin, ModelForm):
        class Meta:
            model = mapitemtype
            fields = existing_fields

    mapitemtype.EditorForm = EditorForm


def create_editor_forms():
    from c3nav.mapdata.models.base import MAPITEM_TYPES
    for mapitemtype in MAPITEM_TYPES.values():
        create_editor_form(mapitemtype)