django-orchestra/orchestra/contrib/mailboxes/forms.py

80 lines
3.3 KiB
Python
Raw Normal View History

2014-10-06 14:57:02 +00:00
from django import forms
2014-10-21 15:29:36 +00:00
from django.contrib.admin import widgets
2015-10-07 11:44:30 +00:00
from django.core.exceptions import ValidationError
2014-10-21 15:29:36 +00:00
from django.utils.safestring import mark_safe
from django.utils.translation import gettext_lazy as _
2014-10-06 14:57:02 +00:00
2014-10-09 17:04:12 +00:00
from orchestra.forms import UserCreationForm, UserChangeForm
2014-10-21 15:29:36 +00:00
from orchestra.utils.python import AttrDict
2014-10-06 14:57:02 +00:00
2015-10-07 11:44:30 +00:00
from . import settings
from .models import Address, Mailbox
2014-10-06 14:57:02 +00:00
2014-10-21 15:29:36 +00:00
class MailboxForm(forms.ModelForm):
""" hacky form for adding reverse M2M form field for Mailbox.addresses """
# TODO keep track of this ticket for future reimplementation
# https://code.djangoproject.com/ticket/897
addresses = forms.ModelMultipleChoiceField(required=False,
queryset=Address.objects.select_related('domain'),
2015-04-05 10:46:24 +00:00
widget=widgets.FilteredSelectMultiple(verbose_name=_('addresses'), is_stacked=False))
2014-10-21 15:29:36 +00:00
def __init__(self, *args, **kwargs):
super(MailboxForm, self).__init__(*args, **kwargs)
# Hack the widget in order to display add button
2016-05-04 13:00:03 +00:00
remote_field_mock = AttrDict(**{
'model': Address,
2014-10-21 15:29:36 +00:00
'get_related_field': lambda: AttrDict(name='id'),
2016-05-04 13:00:03 +00:00
2014-10-21 15:29:36 +00:00
})
widget = self.fields['addresses'].widget
2016-05-04 13:00:03 +00:00
self.fields['addresses'].widget = widgets.RelatedFieldWidgetWrapper(
widget, remote_field_mock, self.modeladmin.admin_site, can_add_related=True)
account = self.modeladmin.account
# Filter related addresses by account
2014-10-21 15:29:36 +00:00
old_render = self.fields['addresses'].widget.render
def render(*args, **kwargs):
output = old_render(*args, **kwargs)
2016-05-04 13:00:03 +00:00
args = 'account=%i&mailboxes=%s' % (account.pk, self.instance.pk)
2014-10-21 15:29:36 +00:00
output = output.replace('/add/?', '/add/?%s&' % args)
return mark_safe(output)
self.fields['addresses'].widget.render = render
queryset = self.fields['addresses'].queryset
realted_addresses = queryset.filter(account_id=account.pk).order_by('name')
2015-06-17 10:34:14 +00:00
self.fields['addresses'].queryset = realted_addresses
2014-10-21 15:29:36 +00:00
if self.instance and self.instance.pk:
self.fields['addresses'].initial = self.instance.addresses.all()
def clean_name(self):
name = self.cleaned_data['name']
max_length = settings.MAILBOXES_NAME_MAX_LENGTH
if len(name) > max_length:
raise ValidationError("Name length should be less than %i." % max_length)
return name
2014-10-09 17:04:12 +00:00
2014-10-21 15:29:36 +00:00
class MailboxChangeForm(UserChangeForm, MailboxForm):
2014-10-09 17:04:12 +00:00
pass
2014-10-21 15:29:36 +00:00
class MailboxCreationForm(UserCreationForm, MailboxForm):
2014-10-06 14:57:02 +00:00
def clean_name(self):
# Since model.clean() will check this, this is redundant,
# but it sets a nicer error message than the ORM and avoids conflicts with contrib.auth
name = super().clean_name()
2014-10-06 14:57:02 +00:00
try:
self._meta.model._default_manager.get(name=name)
except self._meta.model.DoesNotExist:
return name
2014-10-21 15:29:36 +00:00
raise forms.ValidationError(self.error_messages['duplicate_username'])
2014-10-06 14:57:02 +00:00
class AddressForm(forms.ModelForm):
def clean(self):
cleaned_data = super(AddressForm, self).clean()
2015-10-07 11:44:30 +00:00
forward = cleaned_data.get('forward', '')
if not cleaned_data.get('mailboxes', True) and not forward:
raise ValidationError(_("Mailboxes or forward address should be provided."))