gva/gnuviechadmin/osusers/views.py

"""
This module defines the views for gnuviechadmin operating system user handling.

"""
from __future__ import unicode_literals, absolute_import

from django.core.urlresolvers import reverse
from django.shortcuts import redirect
from django.views.generic import (
    CreateView,
    DeleteView,
    ListView,
    UpdateView,
)
from django.utils.translation import ugettext as _
from django.contrib import messages

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

from .forms import (
    AddSshPublicKeyForm,
    ChangeOsUserPasswordForm,
    EditSshPublicKeyCommentForm,
)
from .models import (
    SshPublicKey,
    User,
)


class SetOsUserPassword(StaffOrSelfLoginRequiredMixin, UpdateView):
    """
    This view is used for setting a new operating system user password.

    """
    model = User
    slug_field = 'username'
    template_name_suffix = '_setpassword'
    context_object_name = 'osuser'
    form_class = ChangeOsUserPasswordForm

    def get_customer_object(self):
        return self.get_object().customer

    def get_context_data(self, *args, **kwargs):
        context = super(SetOsUserPassword, self).get_context_data(
            *args, **kwargs)
        context['customer'] = self.get_customer_object()
        return context

    def form_valid(self, form):
        osuser = form.save()
        messages.success(
            self.request,
            _("New password for {username} has been set successfully.").format(
                username=osuser.username
            ))
        return redirect(osuser.customerhostingpackage)


class AddSshPublicKey(
    HostingPackageAndCustomerMixin, StaffOrSelfLoginRequiredMixin, CreateView
):
    """
    This view is used to add an SSH key for an existing hosting account's
    operating system user.

    """
    model = SshPublicKey
    context_object_name = 'key'
    template_name_suffix = '_create'
    form_class = AddSshPublicKeyForm

    def get_form_kwargs(self):
        kwargs = super(AddSshPublicKey, self).get_form_kwargs()
        kwargs['hostingpackage'] = self.get_hosting_package()
        return kwargs

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

    def form_valid(self, form):
        key = form.save()
        messages.success(
            self.request,
            _('Successfully added new {algorithm} SSH public key').format(
                algorithm=key.algorithm)
        )
        return redirect(self.get_hosting_package())


class ListSshPublicKeys(
    HostingPackageAndCustomerMixin, StaffOrSelfLoginRequiredMixin, ListView
):
    model = SshPublicKey
    context_object_name = 'keys'

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


class DeleteSshPublicKey(
    HostingPackageAndCustomerMixin, StaffOrSelfLoginRequiredMixin, DeleteView
):
    model = SshPublicKey
    context_object_name = 'key'

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

    def get_success_url(self):
        return reverse(
            'list_ssh_keys', kwargs={'package': self.get_hosting_package().id}
        )


class EditSshPublicKeyComment(
    HostingPackageAndCustomerMixin, StaffOrSelfLoginRequiredMixin, UpdateView
):
    model = SshPublicKey
    context_object_name = 'key'
    fields = ['comment']
    template_name_suffix = '_edit_comment'
    form_class = EditSshPublicKeyCommentForm

    def get_form_kwargs(self):
        kwargs = super(EditSshPublicKeyComment, self).get_form_kwargs()
        kwargs['hostingpackage'] = self.get_hosting_package()
        return kwargs

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

    def get_success_url(self):
        return reverse(
            'list_ssh_keys', kwargs={'package': self.get_hosting_package().id}
        )