py-i2phosts/pyi2phosts/postkey/views.py

import re
import datetime

from django import forms
from django.shortcuts import render_to_response
from django.shortcuts import redirect
from django.template import RequestContext

import settings
from pyi2phosts.postkey.models import i2phost
from pyi2phosts.lib.utils import get_logger
from pyi2phosts.lib.validation import validate_hostname
from pyi2phosts.lib.validation import validate_b64hash

class AddForm(forms.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']
		log.debug(u'hostname: %s', self.data['name'])
		data = validate_hostname(data)
		# Another set of reserved hostnames (suggested by zzz)
		if re.search(r'(^|\.)(i2p|i2p2|geti2p|mail|project|i2project|i2pproject|i2p-project).i2p$', data):
			raise forms.ValidationError('Trying to use hostname from additional reserved set')
		if data.count('.') > 1:
			raise forms.ValidationError('Currently only 2-level domains are allowed')
		return data
	def clean_b64hash(self):
		"""Validate base64 hash"""
		data = self.cleaned_data['b64hash']
		log.debug(u'hash: %s', self.data['b64hash'])
		data = validate_b64hash(data)
		return data

def addkey(request):
	if request.method == 'POST':
		form = AddForm(request.POST)
		if form.is_valid():
			log.debug('submit is valid, saving')
			newhost = form.save(commit=False)
			newhost.date_added = datetime.datetime.utcnow()
			newhost.save()
			request.session['hostname'] = form.cleaned_data['name']
			return redirect('pyi2phosts.postkey.views.success')
	else:
		form = AddForm()
	return render_to_response('postkey.html', {
		'title': settings.SITE_NAME,
		'form': form,
		}, context_instance=RequestContext(request))

def success(request):
	if 'hostname' in request.session:
		return render_to_response('success_submission.html', {
			'title': settings.SITE_NAME,
			'hostname': request.session['hostname'],
			})
	else:
		return redirect('/')

log = get_logger(filename=settings.LOG_FILE, log_level=settings.LOG_LEVEL)