100 lines
3.3 KiB
Python
100 lines
3.3 KiB
Python
# Binder Forms
|
|
|
|
# 3rd Party
|
|
from django import forms
|
|
from django.conf import settings
|
|
from django.forms import ValidationError
|
|
|
|
# App Imports
|
|
from models import Key
|
|
|
|
|
|
class CustomUnicodeListField(forms.CharField):
|
|
|
|
"""Convert unicode item list to list of strings."""
|
|
|
|
def clean(self, value):
|
|
try:
|
|
string_list = [str(cur_rr) for cur_rr in eval(value)]
|
|
except:
|
|
raise ValidationError("Error in converting Unicode list to list of Strings: %r" % value)
|
|
return string_list
|
|
|
|
|
|
class CustomStringPeriodSuffix(forms.CharField):
|
|
|
|
"""Convert unicode to string and make sure period is last character.
|
|
|
|
This seems very unclean. Need a better to way to complete the fqdn
|
|
depending on if it ends in a period.
|
|
TODO(jforman): Add Regex check in here for valid rr data
|
|
http://www.zytrax.com/books/dns/apa/names.html
|
|
"""
|
|
|
|
def clean(self, value):
|
|
try:
|
|
new_string = str(value)
|
|
if new_string[-1] != ".":
|
|
new_string += "."
|
|
except:
|
|
raise ValidationError("Unable to stick a period on the end of your input: %r" % value)
|
|
|
|
return new_string
|
|
|
|
|
|
class FormAddForwardRecord(forms.Form):
|
|
|
|
"""Form used to add a Forward DNS record."""
|
|
|
|
dns_server = forms.CharField(max_length=100)
|
|
record_name = forms.RegexField(max_length=100,
|
|
regex="^[a-zA-Z0-9-_]+$",
|
|
required=False)
|
|
record_type = forms.ChoiceField(choices=settings.RECORD_TYPE_CHOICES)
|
|
zone_name = forms.CharField(max_length=100)
|
|
record_data = forms.GenericIPAddressField()
|
|
ttl = forms.ChoiceField(choices=settings.TTL_CHOICES)
|
|
create_reverse = forms.BooleanField(required=False)
|
|
key_name = forms.ModelChoiceField(queryset=Key.objects.all(),
|
|
required=False)
|
|
|
|
|
|
class FormAddReverseRecord(forms.Form):
|
|
|
|
"""Form used to add a Reverse (PTR) DNS record."""
|
|
|
|
dns_server = forms.CharField(max_length=100)
|
|
record_name = forms.IntegerField(min_value=0, max_value=255)
|
|
record_type = forms.RegexField(regex=r"^PTR$",
|
|
error_messages={"invalid": "The only valid choice here is PTR."})
|
|
zone_name = forms.CharField(max_length=100)
|
|
record_data = CustomStringPeriodSuffix(required=True)
|
|
ttl = forms.ChoiceField(choices=settings.TTL_CHOICES)
|
|
key_name = forms.ModelChoiceField(queryset=Key.objects.all(),
|
|
required=False)
|
|
create_reverse = forms.BooleanField(required=False)
|
|
|
|
|
|
class FormAddCnameRecord(forms.Form):
|
|
|
|
"""Form used to add a CNAME record."""
|
|
|
|
dns_server = forms.CharField(max_length=100)
|
|
originating_record = forms.CharField(max_length=100)
|
|
cname = forms.RegexField(max_length=100, regex="^[a-zA-Z0-9-_]+$")
|
|
zone_name = forms.CharField(max_length=256)
|
|
ttl = forms.ChoiceField(choices=settings.TTL_CHOICES)
|
|
key_name = forms.ModelChoiceField(queryset=Key.objects.all(),
|
|
required=False)
|
|
|
|
|
|
class FormDeleteRecord(forms.Form):
|
|
|
|
"""Final form to delete DNS record(s)."""
|
|
|
|
dns_server = forms.CharField(max_length=100)
|
|
zone_name = forms.CharField(max_length=256)
|
|
rr_list = CustomUnicodeListField()
|
|
key_name = forms.ModelChoiceField(queryset=Key.objects.all(),
|
|
required=False)
|