Jan Dittberner
30600ce107
This commit adds a test suite for the views defined in managemails.views. One issue discovered when writing tests is fixed by checking the request method in CreateMailbox.dispatch.
227 lines
7 KiB
Python
227 lines
7 KiB
Python
"""
|
|
This module defines views for mailbox and mail address handling.
|
|
|
|
"""
|
|
from __future__ import absolute_import, unicode_literals
|
|
|
|
from django.http import HttpResponseForbidden
|
|
from django.shortcuts import get_object_or_404, redirect
|
|
from django.utils.translation import ugettext as _
|
|
from django.views.generic.edit import (
|
|
CreateView,
|
|
DeleteView,
|
|
UpdateView,
|
|
)
|
|
from django.contrib import messages
|
|
|
|
from gvacommon.viewmixins import StaffOrSelfLoginRequiredMixin
|
|
from gvawebcore.views import HostingPackageAndCustomerMixin
|
|
|
|
from domains.models import MailDomain
|
|
from .forms import (
|
|
AddMailAddressForm,
|
|
ChangeMailboxPasswordForm,
|
|
CreateMailboxForm,
|
|
EditMailAddressForm,
|
|
MAILBOX_OR_FORWARDS,
|
|
)
|
|
from .models import (
|
|
MailAddress,
|
|
MailAddressMailbox,
|
|
Mailbox,
|
|
)
|
|
|
|
|
|
class CreateMailbox(
|
|
HostingPackageAndCustomerMixin, StaffOrSelfLoginRequiredMixin, CreateView
|
|
):
|
|
"""
|
|
This view is used to setup new mailboxes for a customer hosting package.
|
|
|
|
"""
|
|
model = Mailbox
|
|
context_object_name = 'mailbox'
|
|
template_name_suffix = '_create'
|
|
form_class = CreateMailboxForm
|
|
|
|
def dispatch(self, request, *args, **kwargs):
|
|
resp = super(CreateMailbox, self).dispatch(request, *args, **kwargs)
|
|
if request.method != 'POST':
|
|
if not self.get_hosting_package().may_add_mailbox():
|
|
resp = HttpResponseForbidden(
|
|
_('You are not allowed to add more mailboxes to this'
|
|
' hosting package'))
|
|
return resp
|
|
|
|
def get_context_data(self, **kwargs):
|
|
context = super(CreateMailbox, self).get_context_data(**kwargs)
|
|
context['hostingpackage'] = self.get_hosting_package()
|
|
context['customer'] = self.get_customer_object()
|
|
return context
|
|
|
|
def get_form_kwargs(self):
|
|
kwargs = super(CreateMailbox, self).get_form_kwargs()
|
|
kwargs['hostingpackage'] = self.get_hosting_package()
|
|
return kwargs
|
|
|
|
def form_valid(self, form):
|
|
mailbox = form.save()
|
|
messages.success(
|
|
self.request,
|
|
_('Mailbox {mailbox} created successfully.').format(
|
|
mailbox=mailbox.username
|
|
)
|
|
)
|
|
return redirect(self.get_hosting_package())
|
|
|
|
|
|
class ChangeMailboxPassword(
|
|
HostingPackageAndCustomerMixin, StaffOrSelfLoginRequiredMixin, UpdateView
|
|
):
|
|
"""
|
|
This view is used to set a new password for an existing mailbox.
|
|
|
|
"""
|
|
context_object_name = 'mailbox'
|
|
form_class = ChangeMailboxPasswordForm
|
|
model = Mailbox
|
|
slug_field = 'username'
|
|
template_name_suffix = '_setpassword'
|
|
|
|
def get_context_data(self, **kwargs):
|
|
context = super(ChangeMailboxPassword, self).get_context_data(**kwargs)
|
|
context['hostingpackage'] = self.get_hosting_package()
|
|
context['customer'] = self.get_customer_object()
|
|
return context
|
|
|
|
def get_form_kwargs(self):
|
|
kwargs = super(ChangeMailboxPassword, self).get_form_kwargs()
|
|
kwargs['hostingpackage'] = self.get_hosting_package()
|
|
return kwargs
|
|
|
|
def form_valid(self, form):
|
|
mailbox = form.save()
|
|
messages.success(
|
|
self.request,
|
|
_('Successfully set new password for mailbox {mailbox}.').format(
|
|
mailbox=mailbox.username
|
|
)
|
|
)
|
|
return redirect(self.get_hosting_package())
|
|
|
|
|
|
class AddMailAddress(
|
|
HostingPackageAndCustomerMixin, StaffOrSelfLoginRequiredMixin, CreateView
|
|
):
|
|
"""
|
|
This view is used to add a new mail address to a domain.
|
|
|
|
"""
|
|
context_object_name = 'mailaddress'
|
|
form_class = AddMailAddressForm
|
|
model = MailAddress
|
|
template_name_suffix = '_create'
|
|
|
|
def get_context_data(self, **kwargs):
|
|
context = super(AddMailAddress, self).get_context_data(**kwargs)
|
|
context['customer'] = self.get_customer_object()
|
|
return context
|
|
|
|
def get_maildomain(self):
|
|
return get_object_or_404(MailDomain, domain=self.kwargs['domain'])
|
|
|
|
def get_form_kwargs(self):
|
|
kwargs = super(AddMailAddress, self).get_form_kwargs()
|
|
kwargs.update({
|
|
'hostingpackage': self.get_hosting_package(),
|
|
'maildomain': self.get_maildomain(),
|
|
})
|
|
return kwargs
|
|
|
|
def form_valid(self, form):
|
|
address = form.save()
|
|
messages.success(
|
|
self.request,
|
|
_('Successfully added mail address {mailaddress}').format(
|
|
mailaddress=address)
|
|
)
|
|
return redirect(self.get_hosting_package())
|
|
|
|
|
|
class DeleteMailAddress(
|
|
HostingPackageAndCustomerMixin, StaffOrSelfLoginRequiredMixin, DeleteView
|
|
):
|
|
"""
|
|
This view is used to delete a mail address.
|
|
|
|
"""
|
|
context_object_name = 'mailaddress'
|
|
model = MailAddress
|
|
|
|
def get_maildomain(self):
|
|
return get_object_or_404(MailDomain, domain=self.kwargs['domain'])
|
|
|
|
def get_context_data(self, **kwargs):
|
|
context = super(DeleteMailAddress, self).get_context_data(**kwargs)
|
|
context.update({
|
|
'customer': self.get_customer_object(),
|
|
'hostingpackage': self.get_hosting_package(),
|
|
'maildomain': self.get_maildomain(),
|
|
})
|
|
return context
|
|
|
|
def get_success_url(self):
|
|
return self.get_hosting_package().get_absolute_url()
|
|
|
|
|
|
class EditMailAddress(
|
|
HostingPackageAndCustomerMixin, StaffOrSelfLoginRequiredMixin, UpdateView
|
|
):
|
|
"""
|
|
This view is used to edit a mail address' target mailbox or forwarding
|
|
addresses.
|
|
|
|
"""
|
|
context_object_name = 'mailaddress'
|
|
form_class = EditMailAddressForm
|
|
model = MailAddress
|
|
template_name_suffix = '_edit'
|
|
|
|
def get_maildomain(self):
|
|
return get_object_or_404(MailDomain, domain=self.kwargs['domain'])
|
|
|
|
def get_context_data(self, **kwargs):
|
|
context = super(EditMailAddress, self).get_context_data(**kwargs)
|
|
context['customer'] = self.get_customer_object()
|
|
return context
|
|
|
|
def get_form_kwargs(self):
|
|
kwargs = super(EditMailAddress, self).get_form_kwargs()
|
|
kwargs.update({
|
|
'hostingpackage': self.get_hosting_package(),
|
|
'maildomain': self.get_maildomain(),
|
|
})
|
|
return kwargs
|
|
|
|
def get_initial(self):
|
|
initial = super(EditMailAddress, self).get_initial()
|
|
mailaddress = self.get_object()
|
|
if MailAddressMailbox.objects.filter(mailaddress=mailaddress).exists():
|
|
initial['mailbox'] = mailaddress.mailaddressmailbox.mailbox
|
|
initial['mailbox_or_forwards'] = MAILBOX_OR_FORWARDS.mailbox
|
|
elif mailaddress.mailaddressforward_set.exists():
|
|
initial['forwards'] = ", ".join(
|
|
fwd.target for fwd in mailaddress.mailaddressforward_set.all()
|
|
)
|
|
initial['mailbox_or_forwards'] = MAILBOX_OR_FORWARDS.forwards
|
|
return initial
|
|
|
|
def form_valid(self, form):
|
|
mailaddress = form.save()
|
|
messages.success(
|
|
self.request,
|
|
_('Successfully updated mail address {mailaddress} '
|
|
'targets.').format(mailaddress=mailaddress)
|
|
)
|
|
return redirect(self.get_hosting_package())
|