gva/gnuviechadmin/userdbs/forms.py

"""
This module defines form classes for user database editing.

"""
from __future__ import absolute_import, unicode_literals

from django import forms
from django.core.urlresolvers import reverse
from django.utils.translation import ugettext_lazy as _

from crispy_forms.helper import FormHelper
from crispy_forms.layout import (
    Submit,
)

from .models import (
    DB_TYPES,
    DatabaseUser,
    UserDatabase,
)
from gvawebcore.forms import PasswordModelFormMixin


class AddUserDatabaseForm(forms.ModelForm, PasswordModelFormMixin):
    """
    This form is used to create new user database instances.

    """
    db_type = forms.TypedChoiceField(
        label=_('Database type'),
        choices=DB_TYPES,
        widget=forms.RadioSelect,
        coerce=int,
    )

    class Meta:
        model = UserDatabase
        fields = []

    def __init__(self, *args, **kwargs):
        self.hosting_package = kwargs.pop('hostingpackage')
        self.available_dbtypes = kwargs.pop('dbtypes')
        super(AddUserDatabaseForm, self).__init__(*args, **kwargs)
        self.fields['db_type'].choices = self.available_dbtypes
        if len(self.available_dbtypes) == 1:
            self.fields['db_type'].initial = self.available_dbtypes[0][0]
            self.fields['db_type'].widget = forms.HiddenInput()
        self.helper = FormHelper()
        self.helper.form_action = reverse(
            'add_userdatabase', kwargs={'package': self.hosting_package.id})
        self.helper.add_input(Submit('submit', _('Create database')))

    def save(self, commit=True):
        """
        Setup a new database with a new database user with the specified
        password.

        :param boolean commit: whether to save the created database
        :return: database instance
        :rtype: :py:class:`userdbs.models.UserDatabase`

        """
        data = self.cleaned_data
        self.instance = UserDatabase.objects.create_userdatabase_with_user(
            data['db_type'], self.hosting_package.osuser,
            password=data['password1'], commit=commit)
        return super(AddUserDatabaseForm, self).save(commit)


class ChangeDatabaseUserPasswordForm(forms.ModelForm, PasswordModelFormMixin):
    """
    This form is used to change the password of a database user.

    """
    class Meta:
        model = DatabaseUser
        fields = []

    def __init__(self, *args, **kwargs):
        self.hosting_package = kwargs.pop('hostingpackage')
        super(ChangeDatabaseUserPasswordForm, self).__init__(*args, **kwargs)
        self.helper = FormHelper()
        self.helper.form_action = reverse(
            'change_dbuser_password', kwargs={
                'slug': self.instance.name,
                'package': self.hosting_package.id,
            })
        self.helper.add_input(Submit('submit', _('Set password')))

    def save(self, commit=True):
        self.instance.set_password(self.cleaned_data['password1'])
        return super(ChangeDatabaseUserPasswordForm, self).save()