91 lines
2.7 KiB
Python
91 lines
2.7 KiB
Python
from django import forms
|
|
from django.contrib.gis.geos import Point
|
|
from django.contrib.auth import get_user_model
|
|
from django.utils.translation import gettext_lazy as _
|
|
from django.template.defaultfilters import slugify
|
|
from django.conf import settings
|
|
from django.forms.utils import ErrorList
|
|
|
|
from .models import Map, DataLayer
|
|
|
|
DEFAULT_LATITUDE = settings.LEAFLET_LATITUDE if hasattr(settings, "LEAFLET_LATITUDE") else 51
|
|
DEFAULT_LONGITUDE = settings.LEAFLET_LONGITUDE if hasattr(settings, "LEAFLET_LONGITUDE") else 2
|
|
DEFAULT_CENTER = Point(DEFAULT_LONGITUDE, DEFAULT_LATITUDE)
|
|
|
|
User = get_user_model()
|
|
|
|
|
|
class FlatErrorList(ErrorList):
|
|
def __unicode__(self):
|
|
return self.flat()
|
|
|
|
def flat(self):
|
|
if not self:
|
|
return u''
|
|
return u' — '.join([e for e in self])
|
|
|
|
|
|
class SendLinkForm(forms.Form):
|
|
email = forms.EmailField()
|
|
|
|
|
|
class UpdateMapPermissionsForm(forms.ModelForm):
|
|
|
|
class Meta:
|
|
model = Map
|
|
fields = ('edit_status', 'editors', 'share_status', 'owner')
|
|
|
|
|
|
class AnonymousMapPermissionsForm(forms.ModelForm):
|
|
|
|
def __init__(self, *args, **kwargs):
|
|
super(AnonymousMapPermissionsForm, self).__init__(*args, **kwargs)
|
|
help_text = _('Secret edit link is %s') % self.instance.get_anonymous_edit_url()
|
|
self.fields['edit_status'].help_text = _(help_text)
|
|
|
|
STATUS = (
|
|
(Map.ANONYMOUS, _('Everyone can edit')),
|
|
(Map.OWNER, _('Only editable with secret edit link'))
|
|
)
|
|
|
|
edit_status = forms.ChoiceField(choices=STATUS)
|
|
|
|
class Meta:
|
|
model = Map
|
|
fields = ('edit_status', )
|
|
|
|
|
|
class DataLayerForm(forms.ModelForm):
|
|
|
|
class Meta:
|
|
model = DataLayer
|
|
fields = ('geojson', 'name', 'display_on_load', 'rank')
|
|
|
|
|
|
class MapSettingsForm(forms.ModelForm):
|
|
|
|
def __init__(self, *args, **kwargs):
|
|
super(MapSettingsForm, self).__init__(*args, **kwargs)
|
|
self.fields['slug'].required = False
|
|
self.fields['center'].widget.map_srid = 4326
|
|
|
|
def clean_slug(self):
|
|
slug = self.cleaned_data.get('slug', None)
|
|
name = self.cleaned_data.get('name', None)
|
|
if not slug and name:
|
|
# If name is empty, don't do nothing, validation will raise
|
|
# later on the process because name is required
|
|
self.cleaned_data['slug'] = slugify(name) or "map"
|
|
return self.cleaned_data['slug'][:50]
|
|
else:
|
|
return ""
|
|
|
|
def clean_center(self):
|
|
if not self.cleaned_data['center']:
|
|
point = DEFAULT_CENTER
|
|
self.cleaned_data['center'] = point
|
|
return self.cleaned_data['center']
|
|
|
|
class Meta:
|
|
fields = ('settings', 'name', 'center', 'slug')
|
|
model = Map
|