playlistShorten/urlweb/shortener/views.py

115 lines
3.8 KiB
Python
Raw Normal View History

2011-07-20 15:25:26 +00:00
import logging
from django.contrib.auth.decorators import login_required
from django.contrib.auth import authenticate
from django.views.generic import list_detail
from django.shortcuts import get_object_or_404, get_list_or_404, render_to_response
from django.http import HttpResponse, Http404, HttpResponseRedirect, HttpResponsePermanentRedirect
from django.utils import simplejson
from django.template import RequestContext
from django.views.decorators.http import require_POST
from django.db import transaction
from django.conf import settings
from urlweb.shortener.baseconv import base62
from urlweb.shortener.models import Link, LinkSubmitForm
def follow(request, base62_id):
"""
View which gets the link for the given base62_id value
and redirects to it.
"""
key = base62.to_decimal(base62_id)
link = get_object_or_404(Link, pk = key)
link.usage_count += 1
link.save()
2011-07-28 19:02:17 +00:00
if link.url.startswith('data:text/html;charset=utf-8;base64,'):
html = link.url[len('data:text/html;charset=utf-8;base64,'):]
html = html.decode('base64')
return HttpResponse(html);
else:
return HttpResponse("<script>window.location = '%s';</script>" % (link.url,));
2011-07-20 15:25:26 +00:00
def default_values(request, link_form=None):
"""
Return a new object with the default values that are typically
returned in a request.
"""
if not link_form:
link_form = LinkSubmitForm()
allowed_to_submit = is_allowed_to_submit(request)
return { 'show_bookmarklet': allowed_to_submit,
'show_url_form': allowed_to_submit,
'site_name': settings.SITE_NAME,
'site_base_url': settings.SITE_BASE_URL,
'link_form': link_form,
}
def info(request, base62_id):
"""
View which shows information on a particular link
"""
key = base62.to_decimal(base62_id)
link = get_object_or_404(Link, pk = key)
values = default_values(request)
values['link'] = link
return render_to_response(
'shortener/link_info.html',
values,
context_instance=RequestContext(request))
def submit(request):
"""
View for submitting a URL
"""
if settings.REQUIRE_LOGIN and not request.user.is_authenticated():
# TODO redirect to an error page
foo = 'foo'
# raise Http404
url = None
link_form = None
if request.GET:
link_form = LinkSubmitForm(request.GET)
elif request.POST:
link_form = LinkSubmitForm(request.POST)
if link_form and link_form.is_valid():
url = link_form.cleaned_data['u']
link = None
try:
link = Link.objects.get(url = url)
except Link.DoesNotExist:
pass
if link == None:
new_link = Link(url = url)
new_link.save()
link = new_link
values = default_values(request)
values['link'] = link
return render_to_response(
'shortener/submit_success.html',
values,
context_instance=RequestContext(request))
values = default_values(request, link_form=link_form)
return render_to_response(
'shortener/submit_failed.html',
values,
context_instance=RequestContext(request))
def index(request):
"""
View for main page (lists recent and popular links)
"""
values = default_values(request)
values['recent_links'] = Link.objects.all().order_by('-date_submitted')[0:10]
values['most_popular_links'] = Link.objects.all().order_by('-usage_count')[0:10]
return render_to_response(
'shortener/index.html',
values,
context_instance=RequestContext(request))
def is_allowed_to_submit(request):
"""
Return true if user is allowed to submit URLs
"""
return not settings.REQUIRE_LOGIN or request.user.is_authenticated()