gva/gnuviechadmin/userdbs/forms.py

95 lines
2.9 KiB
Python
Raw Normal View History

"""
This module defines form classes for user database editing.
"""
from __future__ import absolute_import
from crispy_forms.helper import FormHelper
from crispy_forms.layout import Submit
from django import forms
from django.urls import reverse
from django.utils.translation import gettext_lazy as _
from gvawebcore.forms import PasswordModelFormMixin
from .models import DB_TYPES, DatabaseUser, UserDatabase
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()