webvirtcloud/accounts/forms.py

from appsettings.settings import app_settings
from django.contrib.auth import get_user_model
from django.forms import ModelForm, ValidationError
from django.utils.translation import gettext_lazy as _

from .models import UserInstance, UserSSHKey
from .utils import validate_ssh_key


class UserInstanceForm(ModelForm):
    def __init__(self, *args, **kwargs):
        super(UserInstanceForm, self).__init__(*args, **kwargs)

        # Make user and instance fields not editable after creation
        instance = getattr(self, 'instance', None)
        if instance and instance.id is not None:
            self.fields['user'].disabled = True
            self.fields['instance'].disabled = True

    def clean_instance(self):
        instance = self.cleaned_data['instance']
        if app_settings.ALLOW_INSTANCE_MULTIPLE_OWNER == 'False':
            exists = UserInstance.objects.filter(instance=instance)
            if exists:
                raise ValidationError(_('Instance owned by another user'))

        return instance

    class Meta:
        model = UserInstance
        fields = '__all__'


class ProfileForm(ModelForm):
    class Meta:
        model = get_user_model()
        fields = ('first_name', 'last_name', 'email')


class UserSSHKeyForm(ModelForm):
    def __init__(self, *args, **kwargs):
        self.user = kwargs.pop('user', None)
        self.publickeys = UserSSHKey.objects.filter(user=self.user)
        super().__init__(*args, **kwargs)

    def clean_keyname(self):
        for key in self.publickeys:
            if self.cleaned_data['keyname'] == key.keyname:
                raise ValidationError(_("Key name already exist"))

        return self.cleaned_data['keyname']

    def clean_keypublic(self):
        for key in self.publickeys:
            if self.cleaned_data['keypublic'] == key.keypublic:
                raise ValidationError(_("Public key already exist"))

        if not validate_ssh_key(self.cleaned_data['keypublic']):
            raise ValidationError(_('Invalid key'))
        return self.cleaned_data['keypublic']

    def save(self, commit=True):
        ssh_key = super().save(commit=False)
        ssh_key.user = self.user
        if commit:
            ssh_key.save()
        return ssh_key

    class Meta:
        model = UserSSHKey
        fields = ('keyname', 'keypublic')