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.
72 lines
2.3 KiB
72 lines
2.3 KiB
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)
|
|
|