gva/gnuviechadmin/hostingpackages/views.py

"""
This module defines views related to hosting packages.

"""
from __future__ import absolute_import

from django.conf import settings
from django.contrib import messages
from django.contrib.auth import get_user_model
from django.contrib.auth.mixins import PermissionRequiredMixin, UserPassesTestMixin
from django.http import Http404
from django.shortcuts import get_object_or_404, redirect
from django.utils.translation import gettext as _
from django.views.generic import DetailView, ListView
from django.views.generic.edit import CreateView, FormView
from gvacommon.viewmixins import StaffOrSelfLoginRequiredMixin

from .forms import (
    AddDiskspaceOptionForm,
    AddMailboxOptionForm,
    AddUserDatabaseOptionForm,
    CreateCustomerHostingPackageForm,
    CreateHostingPackageForm,
)
from .models import (
    CustomerHostingPackage,
    DiskSpaceOption,
    MailboxOption,
    UserDatabaseOption,
)


class CreateHostingPackage(PermissionRequiredMixin, CreateView):
    """
    Create a hosting package.

    """

    model = CustomerHostingPackage
    raise_exception = True
    permission_required = "domains.add_customerhostingpackage"
    template_name_suffix = "_create"
    form_class = CreateHostingPackageForm

    def form_valid(self, form):
        hosting_package = form.save()
        messages.success(
            self.request,
            _("Started setup of new hosting package {name}.").format(
                name=hosting_package.name
            ),
        )
        return redirect(hosting_package)


class CreateCustomerHostingPackage(CreateHostingPackage):
    """
    Create a hosting package for a selected customer.

    """

    form_class = CreateCustomerHostingPackageForm

    def get_form_kwargs(self):
        kwargs = super(CreateCustomerHostingPackage, self).get_form_kwargs()
        kwargs.update(self.kwargs)
        return kwargs

    def get_customer_object(self):
        return get_object_or_404(get_user_model(), username=self.kwargs["user"])

    def get_context_data(self, **kwargs):
        context = super(CreateCustomerHostingPackage, self).get_context_data(**kwargs)
        context["customer"] = self.get_customer_object()
        return context

    def form_valid(self, form):
        hosting_package = form.save(commit=False)
        hosting_package.customer = self.get_customer_object()
        hosting_package.save()
        messages.success(
            self.request,
            _("Started setup of new hosting package {name}.").format(
                name=hosting_package.name
            ),
        )
        return redirect(hosting_package)


class CustomerHostingPackageDetails(StaffOrSelfLoginRequiredMixin, DetailView):
    """
    This view is for showing details of a customer hosting package.

    """

    model = CustomerHostingPackage
    context_object_name = "hostingpackage"
    customer = None

    def get_customer_object(self):
        if self.customer is None:
            self.customer = get_object_or_404(
                get_user_model(), username=self.kwargs["user"]
            )
        return self.customer

    def get_context_data(self, **kwargs):
        context = super(CustomerHostingPackageDetails, self).get_context_data(**kwargs)
        context.update(
            {
                "customer": self.get_customer_object(),
                "uploadserver": settings.OSUSER_UPLOAD_SERVER,
                "databases": context["hostingpackage"].databases,
                "osuser": context["hostingpackage"].osuser,
                "hostingoptions": context["hostingpackage"].get_hostingoptions(),
                "domains": context["hostingpackage"].domains.all(),
                "mailboxes": context["hostingpackage"].mailboxes,
            }
        )
        context["sshkeys"] = context["osuser"].sshpublickey_set.all()
        return context


class StaffUserRequiredMixin(UserPassesTestMixin):
    """
    Mixin to make views available to staff members only.

    """

    def test_func(self):
        return self.request.user.is_staff


class AllCustomerHostingPackageList(StaffUserRequiredMixin, ListView):
    """
    This view is used for showing a list of all hosting packages.

    """

    model = CustomerHostingPackage
    template_name_suffix = "_admin_list"

    def get_queryset(self):
        return (
            super()
            .get_queryset()
            .select_related("osuser", "customer")
            .only("name", "pk", "created", "customer__username", "osuser__username")
        )


class HostingOptionChoices(StaffUserRequiredMixin, DetailView):
    """
    This view displays choices of hosting options for a customer hosting
    package.

    """

    model = CustomerHostingPackage
    context_object_name = "hostingpackage"
    template_name_suffix = "_option_choices"

    def get_context_data(self, **kwargs):
        context = super(HostingOptionChoices, self).get_context_data(**kwargs)
        context.update(
            {
                "customer": self.get_object().customer,
                "hosting_options": (
                    (
                        _("Disk space"),
                        [
                            (option, "diskspace")
                            for option in DiskSpaceOption.objects.all()
                        ],
                    ),
                    (
                        _("Mailboxes"),
                        [
                            (option, "mailboxes")
                            for option in MailboxOption.objects.all()
                        ],
                    ),
                    (
                        _("Databases"),
                        [
                            (option, "databases")
                            for option in UserDatabaseOption.objects.all()
                        ],
                    ),
                ),
            }
        )
        return context


class AddHostingOption(StaffUserRequiredMixin, FormView):
    template_name = "hostingpackages/add_hosting_option.html"

    def get_form_class(self):
        optiontype = self.kwargs["type"]
        if optiontype == "diskspace":
            return AddDiskspaceOptionForm
        elif optiontype == "mailboxes":
            return AddMailboxOptionForm
        elif optiontype == "databases":
            return AddUserDatabaseOptionForm
        raise Http404()

    def get_hosting_package(self):
        return get_object_or_404(CustomerHostingPackage, pk=int(self.kwargs["package"]))

    def get_option_template(self):
        optiontype = self.kwargs["type"]
        optionid = int(self.kwargs["optionid"])
        if optiontype == "diskspace":
            return get_object_or_404(DiskSpaceOption, pk=optionid)
        elif optiontype == "mailboxes":
            return get_object_or_404(MailboxOption, pk=optionid)
        elif optiontype == "databases":
            return get_object_or_404(UserDatabaseOption, pk=optionid)
        raise Http404()

    def get_form_kwargs(self):
        kwargs = super(AddHostingOption, self).get_form_kwargs()
        kwargs["hostingpackage"] = self.get_hosting_package()
        kwargs["option_template"] = self.get_option_template()
        return kwargs

    def get_initial(self):
        initial = super(AddHostingOption, self).get_initial()
        template = self.get_option_template()
        if type(template) == DiskSpaceOption:
            initial.update(
                {
                    "diskspace": template.diskspace,
                    "diskspace_unit": template.diskspace_unit,
                }
            )
        elif type(template) == MailboxOption:
            initial["number"] = template.number
        elif type(template) == UserDatabaseOption:
            initial["number"] = template.number
        else:
            raise Http404()
        return initial

    def get_context_data(self, **kwargs):
        context = super(AddHostingOption, self).get_context_data(**kwargs)
        context["option_template"] = self.get_option_template()
        return context

    def form_valid(self, form):
        option = form.save()
        hosting_package = self.get_hosting_package()
        messages.success(
            self.request,
            _(
                "Successfully added option {option} to hosting package " "{package}."
            ).format(option=option, package=hosting_package.name),
        )
        return redirect(hosting_package)