mirror of
https://github.com/r4sas/py-i2phosts
synced 2025-01-10 06:47:58 +00:00
52 lines
1.5 KiB
Python
52 lines
1.5 KiB
Python
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)
|
|
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'],
|
|
})
|