mirror of
https://github.com/jwinterm/modoboa.git
synced 2026-10-19 05:58:34 +02:00
383 lines
13 KiB
Python
383 lines
13 KiB
Python
"""Forms related to domains management."""
|
|
|
|
from django import forms
|
|
from django.core.urlresolvers import reverse
|
|
from django.http import QueryDict
|
|
from django.utils.translation import ugettext as _, ugettext_lazy
|
|
|
|
from modoboa.core import signals as core_signals
|
|
from modoboa.core.models import User
|
|
from modoboa.lib.exceptions import Conflict
|
|
from modoboa.lib.fields import DomainNameField
|
|
from modoboa.lib.form_utils import (
|
|
YesNoField, WizardForm, WizardStep, DynamicForm, TabForms
|
|
)
|
|
from modoboa.lib.web_utils import render_to_json_response
|
|
from modoboa.parameters import tools as param_tools
|
|
|
|
from .. import signals
|
|
from ..lib import check_if_domain_exists
|
|
from ..models import (
|
|
Domain, DomainAlias, Mailbox, Alias
|
|
)
|
|
|
|
|
|
DOMAIN_TYPES = [
|
|
("domain", _("Domain")),
|
|
]
|
|
|
|
|
|
class DomainFormGeneral(forms.ModelForm, DynamicForm):
|
|
|
|
"""A form to create/edit a domain."""
|
|
|
|
type = forms.ChoiceField(
|
|
label=ugettext_lazy("Type"),
|
|
)
|
|
quota = forms.IntegerField(
|
|
label=ugettext_lazy("Quota"),
|
|
required=False,
|
|
help_text=ugettext_lazy(
|
|
"Default quota in MB applied to mailboxes. Leave empty to use the "
|
|
"default value."
|
|
)
|
|
)
|
|
aliases = DomainNameField(
|
|
label=ugettext_lazy("Alias(es)"),
|
|
required=False,
|
|
help_text=ugettext_lazy(
|
|
"Alias(es) of this domain. Indicate only one name per input, "
|
|
"press ENTER to add a new input."
|
|
)
|
|
)
|
|
|
|
class Meta:
|
|
model = Domain
|
|
fields = ("name", "type", "quota", "aliases", "enabled")
|
|
|
|
def __init__(self, *args, **kwargs):
|
|
self.oldname = None
|
|
if "instance" in kwargs:
|
|
self.oldname = kwargs["instance"].name
|
|
super(DomainFormGeneral, self).__init__(*args, **kwargs)
|
|
|
|
extra_domain_types = reduce(
|
|
lambda a, b: a + b,
|
|
[result[1] for result in signals.extra_domain_types.send(
|
|
sender=self.__class__)]
|
|
)
|
|
self.fields["type"].choices = DOMAIN_TYPES + extra_domain_types
|
|
self.field_widths = {
|
|
"quota": 3
|
|
}
|
|
|
|
if len(args) and isinstance(args[0], QueryDict):
|
|
self._load_from_qdict(args[0], "aliases", DomainNameField)
|
|
elif "instance" in kwargs:
|
|
d = kwargs["instance"]
|
|
for pos, dalias in enumerate(d.domainalias_set.all()):
|
|
name = "aliases_%d" % (pos + 1)
|
|
self._create_field(forms.CharField, name, dalias.name, 3)
|
|
|
|
def clean_quota(self):
|
|
"""Check quota value."""
|
|
if self.cleaned_data["quota"] is None:
|
|
return param_tools.get_global_parameter("default_domain_quota")
|
|
if self.cleaned_data["quota"] < 0:
|
|
raise forms.ValidationError(_("Must be a positive integer"))
|
|
return self.cleaned_data["quota"]
|
|
|
|
def clean(self):
|
|
"""Custom fields validation.
|
|
|
|
We want to prevent duplicate names between domains and domain
|
|
aliases.
|
|
|
|
The validation way is not very smart...
|
|
"""
|
|
cleaned_data = super(DomainFormGeneral, self).clean()
|
|
if self._errors:
|
|
return cleaned_data
|
|
name = cleaned_data["name"]
|
|
label = check_if_domain_exists(
|
|
name, [(DomainAlias, _('domain alias'))])
|
|
if label is not None:
|
|
self.add_error(
|
|
"name", _("A %s with this name already exists")
|
|
% unicode(label)
|
|
)
|
|
self.aliases = []
|
|
for k in cleaned_data.keys():
|
|
if not k.startswith("aliases"):
|
|
continue
|
|
if cleaned_data[k] == "":
|
|
del cleaned_data[k]
|
|
continue
|
|
if cleaned_data[k] == name:
|
|
self._errors[k] = self.error_class(
|
|
[_("A %s with this name already exists") % _("domain")]
|
|
)
|
|
del cleaned_data[k]
|
|
continue
|
|
label = check_if_domain_exists(
|
|
cleaned_data[k], [(Domain, _("domain"))])
|
|
if label is not None:
|
|
self.add_error(
|
|
k, _("A %s with this name already exists")
|
|
% unicode(label)
|
|
)
|
|
else:
|
|
self.aliases.append(cleaned_data[k])
|
|
return cleaned_data
|
|
|
|
def save(self, user, commit=True, domalias_post_create=False):
|
|
"""Custom save method.
|
|
|
|
Updating a domain may have consequences on other objects
|
|
(domain alias, mailbox, quota). The most tricky part concerns
|
|
quotas update.
|
|
|
|
"""
|
|
d = super(DomainFormGeneral, self).save(commit=False)
|
|
if not commit:
|
|
return d
|
|
d.save()
|
|
for dalias in d.domainalias_set.all():
|
|
if dalias.name not in self.aliases:
|
|
dalias.delete()
|
|
else:
|
|
self.aliases.remove(dalias.name)
|
|
if self.aliases:
|
|
core_signals.can_create_object.send(
|
|
self.__class__, context=user, object_type="domain_aliases",
|
|
count=len(self.aliases))
|
|
core_signals.can_create_object.send(
|
|
self.__class__, context=d, object_type="domain_aliases",
|
|
count=len(self.aliases))
|
|
for alias in self.aliases:
|
|
if d.domainalias_set.filter(name=alias).exists():
|
|
continue
|
|
options = {"creator": user} if domalias_post_create else {}
|
|
DomainAlias(name=alias, target=d, enabled=d.enabled).save(
|
|
**options)
|
|
return d
|
|
|
|
|
|
class DomainFormOptions(forms.Form):
|
|
|
|
"""A form containing options for domain creation."""
|
|
|
|
create_dom_admin = YesNoField(
|
|
label=ugettext_lazy("Create a domain administrator"),
|
|
initial=False,
|
|
help_text=ugettext_lazy(
|
|
"Automatically create an administrator for this domain"
|
|
)
|
|
)
|
|
|
|
dom_admin_username = forms.CharField(
|
|
label=ugettext_lazy("Name"),
|
|
initial="admin",
|
|
help_text=ugettext_lazy(
|
|
"The administrator's name. Don't include the domain's name here, "
|
|
"it will be automatically appended."
|
|
),
|
|
required=False
|
|
)
|
|
|
|
create_aliases = YesNoField(
|
|
label=ugettext_lazy("Create aliases"),
|
|
initial=True,
|
|
help_text=ugettext_lazy(
|
|
"Automatically create standard aliases for this domain"
|
|
),
|
|
required=False
|
|
)
|
|
|
|
def __init__(self, user, *args, **kwargs):
|
|
super(DomainFormOptions, self).__init__(*args, **kwargs)
|
|
results = core_signals.user_can_set_role.send(
|
|
sender=self.__class__, user=user, role="DomainAdmins")
|
|
if False in [result[1] for result in results]:
|
|
self.fields = {}
|
|
return
|
|
|
|
def clean_dom_admin_username(self):
|
|
"""Ensure admin username is an email address."""
|
|
if "@" in self.cleaned_data["dom_admin_username"]:
|
|
raise forms.ValidationError(_("Invalid format"))
|
|
return self.cleaned_data["dom_admin_username"]
|
|
|
|
def clean(self):
|
|
"""Check required values."""
|
|
cleaned_data = super(DomainFormOptions, self).clean()
|
|
if cleaned_data.get("create_dom_admin"):
|
|
if not cleaned_data.get("dom_admin_username"):
|
|
self.add_error(
|
|
"dom_admin_username", _("This field is required."))
|
|
if "create_aliases" not in cleaned_data:
|
|
self.add_error(
|
|
"create_aliases", _("This field is required."))
|
|
return cleaned_data
|
|
|
|
def save(self, *args, **kwargs):
|
|
if not self.fields:
|
|
return
|
|
if not self.cleaned_data["create_dom_admin"]:
|
|
return
|
|
user = kwargs.pop("user")
|
|
domain = kwargs.pop("domain")
|
|
username = "%s@%s" % (
|
|
self.cleaned_data["dom_admin_username"], domain.name)
|
|
try:
|
|
da = User.objects.get(username=username)
|
|
except User.DoesNotExist:
|
|
pass
|
|
else:
|
|
raise Conflict(_("User '%s' already exists") % username)
|
|
core_signals.can_create_object.send(
|
|
self.__class__, context=user, object_type="mailboxes")
|
|
da = User(username=username, email=username, is_active=True)
|
|
da.set_password(
|
|
param_tools.get_global_parameter("default_password", app="core"))
|
|
da.save()
|
|
da.role = "DomainAdmins"
|
|
da.post_create(user)
|
|
|
|
dom_admin_username = self.cleaned_data["dom_admin_username"]
|
|
mb = Mailbox(
|
|
address=dom_admin_username, domain=domain,
|
|
user=da, use_domain_quota=True
|
|
)
|
|
mb.set_quota(
|
|
override_rules=user.has_perm("admin.change_domain"))
|
|
mb.save(creator=user)
|
|
|
|
condition = (
|
|
domain.type == "domain" and
|
|
self.cleaned_data["create_aliases"] and
|
|
dom_admin_username != "postmaster"
|
|
)
|
|
if condition:
|
|
core_signals.can_create_object.send(
|
|
self.__class__, context=user, object_type="mailbox_aliases")
|
|
address = u"postmaster@{}".format(domain.name)
|
|
alias = Alias.objects.create(
|
|
address=address, domain=domain, enabled=True)
|
|
alias.set_recipients([mb.full_address])
|
|
alias.post_create(user)
|
|
|
|
domain.add_admin(da)
|
|
|
|
|
|
class DomainForm(TabForms):
|
|
|
|
"""Domain edition form."""
|
|
|
|
def __init__(self, request, *args, **kwargs):
|
|
self.user = request.user
|
|
self.forms = []
|
|
self.domain = None
|
|
if self.user.has_perm("admin.change_domain"):
|
|
self.forms.append({
|
|
"id": "general",
|
|
"title": _("General"),
|
|
"formtpl": "admin/domain_general_form.html",
|
|
"cls": DomainFormGeneral,
|
|
"mandatory": True
|
|
})
|
|
|
|
cbargs = {"user": self.user}
|
|
if "instances" in kwargs:
|
|
self.domain = kwargs["instances"]["general"]
|
|
cbargs["domain"] = self.domain
|
|
results = signals.extra_domain_forms.send(
|
|
sender=self.__class__, **cbargs)
|
|
self.forms += reduce(
|
|
lambda a, b: a + b, [result[1] for result in results])
|
|
if not self.forms:
|
|
self.active_id = "admins"
|
|
super(DomainForm, self).__init__(request, *args, **kwargs)
|
|
|
|
def extra_context(self, context):
|
|
"""Add information to template context."""
|
|
context.update({
|
|
"title": self.domain.name,
|
|
"action": reverse(
|
|
"admin:domain_change", args=[self.domain.pk]),
|
|
"formid": "domform",
|
|
"domain": self.domain
|
|
})
|
|
|
|
def is_valid(self):
|
|
"""Custom validation.
|
|
|
|
We just save the current name before it is potentially
|
|
modified.
|
|
|
|
"""
|
|
if "general" in self.instances:
|
|
self.instances["general"].oldname = self.instances["general"].name
|
|
return super(DomainForm, self).is_valid()
|
|
|
|
def save(self):
|
|
"""Custom save method.
|
|
|
|
As forms interact with each other, it is easier to make custom
|
|
code to save them.
|
|
"""
|
|
if not self.forms:
|
|
return
|
|
first_form = self.forms[0]["instance"]
|
|
options = {}
|
|
if isinstance(first_form, DomainFormGeneral):
|
|
options["domalias_post_create"] = True
|
|
first_form.save(self.request.user, **options)
|
|
for f in self.forms[1:]:
|
|
f["instance"].save(self.request.user)
|
|
|
|
def done(self):
|
|
return render_to_json_response(_("Domain modified"))
|
|
|
|
|
|
class DomainWizard(WizardForm):
|
|
|
|
"""Domain creation wizard."""
|
|
|
|
def __init__(self, request):
|
|
super(DomainWizard, self).__init__(request)
|
|
self.add_step(
|
|
WizardStep(
|
|
"general", DomainFormGeneral, _("General"),
|
|
"admin/domain_general_form.html"
|
|
)
|
|
)
|
|
results = signals.extra_domain_wizard_steps.send(sender=self.__class__)
|
|
for result in results:
|
|
for step in result[1]:
|
|
self.add_step(step)
|
|
self.add_step(
|
|
WizardStep(
|
|
"options", DomainFormOptions, _("Options"),
|
|
"admin/domain_options_form.html",
|
|
[self.request.user]
|
|
)
|
|
)
|
|
|
|
def extra_context(self, context):
|
|
context.update({
|
|
"title": _("New domain"),
|
|
"action": reverse("admin:domain_add"),
|
|
"formid": "domform"
|
|
})
|
|
|
|
def done(self):
|
|
genform = self.first_step.form
|
|
domain = genform.save(self.request.user)
|
|
domain.post_create(self.request.user)
|
|
for step in self.steps[1:]:
|
|
if not step.check_access(self):
|
|
continue
|
|
step.form.save(user=self.request.user, domain=domain)
|
|
return render_to_json_response(_("Domain created"))
|