gva/gnuviechadmin/websites/views.py

"""
This module defines views for website handling.

"""
from __future__ import absolute_import, unicode_literals

from django.shortcuts import get_object_or_404, redirect
from django.utils.translation import ugettext as _
from django.views.generic.edit import (
    CreateView,
    DeleteView,
)
from django.contrib import messages

from gvacommon.viewmixins import StaffOrSelfLoginRequiredMixin
from gvawebcore.views import HostingPackageAndCustomerMixin

from domains.models import HostingDomain
from .forms import AddWebsiteForm
from .models import Website


class AddWebsite(
    HostingPackageAndCustomerMixin, StaffOrSelfLoginRequiredMixin, CreateView
):
    """
    This view is used to setup new websites for a customer hosting package.

    """
    model = Website
    context_object_name = 'website'
    template_name_suffix = '_create'
    form_class = AddWebsiteForm

    def get_form_kwargs(self):
        kwargs = super(AddWebsite, self).get_form_kwargs()
        kwargs.update({
            'hostingpackage': self.get_hosting_package(),
            'domain': get_object_or_404(
                HostingDomain, domain=self.kwargs['domain']),
        })
        return kwargs

    def get_context_data(self, **kwargs):
        context = super(AddWebsite, self).get_context_data(**kwargs)
        context.update({
            'customer': self.get_customer_object(),
            'domain': get_object_or_404(
                HostingDomain, domain=self.kwargs['domain'])
        })
        return context

    def form_valid(self, form):
        website = form.save()
        messages.success(
            self.request,
            _('Successfully added website {subdomain}.{domain}').format(
                subdomain=website.subdomain, domain=website.domain.domain
            )
        )
        return redirect(self.get_hosting_package())


class DeleteWebsite(
    HostingPackageAndCustomerMixin, StaffOrSelfLoginRequiredMixin, DeleteView
):
    """
    This view is used to delete websites in a customer hosting package.

    """
    context_object_name = 'website'
    model = Website

    def get_context_data(self, **kwargs):
        context = super(DeleteWebsite, self).get_context_data(**kwargs)
        context.update({
            'customer': self.get_customer_object(),
            'hostingpackage': self.get_hosting_package(),
        })
        return context

    def get_success_url(self):
        return self.get_hosting_package().get_absolute_url()