mirror of https://github.com/r4sas/py-i2phosts
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.
55 lines
1.6 KiB
55 lines
1.6 KiB
from django.shortcuts import render_to_response |
|
from django.http import HttpResponseRedirect |
|
from django import forms |
|
from django.forms import ModelForm |
|
from django.template import RequestContext |
|
|
|
from web.postkey.models import i2phost |
|
from web.lib.validation import validate_hostname |
|
from web.lib.validation import validate_b64hash |
|
|
|
class AddForm(ModelForm): |
|
""" |
|
This is our class for host-add form. It's based on django's ModelForm |
|
and uses our model "i2phost" (see postkey/models.py) |
|
""" |
|
class Meta: |
|
model = i2phost |
|
fields = ('name', 'b64hash', 'description') |
|
widgets = { |
|
'name': forms.TextInput(attrs={'size': '67'}), |
|
'b64hash': forms.Textarea(attrs={'rows': '1', 'cols': '100'}), |
|
'description': forms.Textarea(attrs={'rows': '2', 'cols': '72'}) |
|
} |
|
def clean_name(self): |
|
"""Validate hostname""" |
|
data = self.cleaned_data['name'] |
|
data = validate_hostname(data) |
|
return data |
|
def clean_b64hash(self): |
|
"""Validate base64 hash""" |
|
data = self.cleaned_data['b64hash'] |
|
data = validate_b64hash(data) |
|
# Avoid adding non-unique hashes |
|
qs = i2phost.objects.filter(b64hash=data) |
|
if qs.exists(): |
|
raise forms.ValidationError('Base64 hash must be unique') |
|
return data |
|
|
|
def addkey(request): |
|
if request.method == 'POST': |
|
form = AddForm(request.POST) |
|
if form.is_valid(): |
|
form.save() |
|
request.session['hostname'] = form.cleaned_data['name'] |
|
return HttpResponseRedirect('success') |
|
else: |
|
form = AddForm() |
|
return render_to_response('postkey.html', { |
|
'form': form, |
|
}, context_instance=RequestContext(request)) |
|
|
|
def success(request): |
|
return render_to_response('success_submission.html', { |
|
'hostname': request.session['hostname'], |
|
})
|
|
|