document.open();document.writeln('
');document.writeln('
from django.shortcuts import render_to_response, get_object_or_404');document.writeln('from django.template import RequestContext');document.writeln('from django.http import Http404, HttpResponseRedirect');document.writeln('from urlfier.models import *');document.writeln('import urllib2');document.writeln('from shorturl.settings import URL_RESTRICTS');document.writeln('import re');document.writeln('');document.writeln('def shorturl_form(request):');document.writeln('');document.writeln('    return render_to_response('urlfier/shorturl.html', {})');document.writeln('');document.writeln('def do_redirect(request, tinykey=None):');document.writeln('');document.writeln('    try:');document.writeln('        u = ShortUrl.objects.get(tinykey=tinykey)');document.writeln('        print u.url');document.writeln('    except ShortUrl.DoesNotExist:');document.writeln('        raise Http404');document.writeln('');document.writeln('    for pattern in URL_RESTRICTS:');document.writeln('        if re.match(pattern, u.url, re.I):');document.writeln('            raise Http404');document.writeln('    try:');document.writeln('        opener = urllib2.build_opener(SmartRedirectHandler())');document.writeln('        f = opener.open(u.url)');document.writeln('    except (urllib2.URLError,urllib2.HTTPError):');document.writeln('        u.contor += 1');document.writeln('        u.save()');document.writeln('        return HttpResponseRedirect(u.url)');document.writeln('    except Exception:');document.writeln('        raise Http404');document.writeln('');document.writeln('    u.contor += 1');document.writeln('    u.save()');document.writeln('    return HttpResponseRedirect(u.url)');document.writeln('');document.writeln('');document.writeln('class SmartRedirectHandler(urllib2.HTTPRedirectHandler):');document.writeln('    recursion = 0');document.writeln('');document.writeln('    def http_error_301(self, req, fp, code, msg, headers):');document.writeln('        if self.recursion>2:');document.writeln('            raise Exception("I won't follow redirects")');document.writeln('        self.recursion += 1');document.writeln('        result = urllib2.HTTPRedirectHandler.http_error_301(self, req, fp, code, msg, headers)');document.writeln('        result.status = code');document.writeln('        return result');document.writeln('');document.writeln('    def http_error_302(self, req, fp, code, msg, headers):');document.writeln('        if self.recursion>2:');document.writeln('            raise Exception("I won't follow redirects")');document.writeln('        self.recursion += 1');document.writeln('        result = urllib2.HTTPRedirectHandler.http_error_302(self, req, fp, code, msg, headers)');document.writeln('        result.status = code');document.writeln('        return result');document.writeln('
');document.writeln('');document.writeln('');document.writeln('
');document.close();