You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
75 lines
2.1 KiB
75 lines
2.1 KiB
from django import forms
|
|
|
|
#from crispy_forms.helper import FormHelper
|
|
#from crispy_forms.layout import Submit, Layout
|
|
#from django.urls import reverse
|
|
|
|
from .models import Maintainer, Contact, InetNum
|
|
|
|
class WhoisObjectMixin(object):
|
|
def __init__(self, user, *args, **kwargs):
|
|
super(WhoisObjectMixin, self).__init__(*args, **kwargs)
|
|
self._user = user
|
|
|
|
instance = getattr(self, 'instance', None)
|
|
if instance and instance.pk:
|
|
self.fields['handle'].widget.attrs['readonly'] = True
|
|
|
|
def clean_handle(self):
|
|
instance = getattr(self, 'instance', None)
|
|
if instance and instance.pk:
|
|
return instance.handle
|
|
else:
|
|
return self.cleaned_data['handle']
|
|
|
|
def clean(self):
|
|
cleaned_data = super(WhoisObjectMixin, self).clean()
|
|
if cleaned_data.get("handle") == "AUTO":
|
|
cleaned_data['handle'] = self._meta.model.genGenericHandle(cleaned_data.get("name"))
|
|
|
|
class MntForm(forms.ModelForm):
|
|
class Meta:
|
|
model = Maintainer
|
|
fields = ['handle', 'description', 'admin_c']
|
|
|
|
def __init__(self, user, *args, **kwargs):
|
|
super(MntForm, self).__init__(*args, **kwargs)
|
|
self._user = user
|
|
if 'admin_c' in self.fields:
|
|
self.fields['admin_c'].queryset = Contact.objects.filter(mnt_by=user.maintainer_set.all())
|
|
|
|
class MntInitialForm(MntForm):
|
|
class Meta:
|
|
model = Maintainer
|
|
fields = ['handle', 'description']
|
|
|
|
class ContactForm(WhoisObjectMixin, forms.ModelForm):
|
|
class Meta:
|
|
model = Contact
|
|
fields = ['type', 'handle', 'name', 'mnt_by']
|
|
|
|
def __init__(self, *args, **kwargs):
|
|
super(ContactForm, self).__init__(*args, **kwargs)
|
|
|
|
class ContactInitialForm(ContactForm):
|
|
class Meta:
|
|
model = Contact
|
|
fields = ['handle', 'name']
|
|
|
|
class InetNumForm(WhoisObjectMixin, forms.ModelForm):
|
|
class Meta:
|
|
model = InetNum
|
|
fields = ['protocol', 'parent_range', 'address', 'description', 'mnt_lower']
|
|
|
|
def __init__(self, lower=False, *args, **kwargs):
|
|
super(MntForm, self).__init__(*args, **kwargs)
|
|
self._editLower = lower
|
|
if 'admin_c' in self.fields:
|
|
self.fields['admin_c'].queryset = Contact.objects.filter(mnt_by=self.user.maintainer_set.all())
|
|
|
|
if self._editLower:
|
|
|
|
def clean(self):
|
|
# FIXME: Reset certain field sto instance:
|
|
pass
|