|
|
|
@ -6,7 +6,7 @@
|
|
|
|
|
from django.shortcuts import render
|
|
|
|
|
from django.http import HttpResponseRedirect
|
|
|
|
|
|
|
|
|
|
from .settings import TEMPLATES
|
|
|
|
|
from .settings import TEMPLATES, RANDOM_SLUG_LEN, UID_LEN
|
|
|
|
|
from .models import Slug
|
|
|
|
|
from .forms import SlugAddForm
|
|
|
|
|
|
|
|
|
@ -25,19 +25,23 @@ def AddSlug(request):
|
|
|
|
|
"""
|
|
|
|
|
if request.method == 'POST':
|
|
|
|
|
form = SlugAddForm(request.POST)
|
|
|
|
|
|
|
|
|
|
if form.is_valid():
|
|
|
|
|
slug = form.save(commit=False)
|
|
|
|
|
if slug.slug == '':
|
|
|
|
|
slug.slug = ''.join(random.choice(ensemble) for _ in range(5))
|
|
|
|
|
|
|
|
|
|
slug.uid = ''.join(random.choice(ensemble) for _ in range(20))
|
|
|
|
|
|
|
|
|
|
slug.slug = ''.join(random.choice(ensemble) for _ in range(RANDOM_SLUG_LEN))
|
|
|
|
|
slug.uid = ''.join(random.choice(ensemble) for _ in range(UID_LEN))
|
|
|
|
|
slug.save()
|
|
|
|
|
|
|
|
|
|
return HttpResponseRedirect('')
|
|
|
|
|
return HttpResponseRedirect('display/{}'.format(slug.slug))
|
|
|
|
|
else:
|
|
|
|
|
form = SlugAddForm()
|
|
|
|
|
|
|
|
|
|
return render(request,'add_slug.html',{'form':form})
|
|
|
|
|
|
|
|
|
|
def display(request, slug):
|
|
|
|
|
slug = Slug.objects.filter(slug = slug).all()
|
|
|
|
|
if slug.count()!=1:
|
|
|
|
|
# TODO Redirect to 'unknown and display a message
|
|
|
|
|
return HttpResponsRedirect('display')
|
|
|
|
|
else:
|
|
|
|
|
context = {'slug':slug.get()}
|
|
|
|
|
return render(request,'display.html',context)
|
|
|
|
|
|
|
|
|
|