1

from django.http import HttpResponse, HttpResponseRedirect, Http404

     2

from django.shortcuts import render

     3

     4

from qrurls.models import URL, URLItem

     5

     6

def _get_url_feed (shorturl=None, shorturl_id=None) :

     7

    if shorturl :

     8

        shorturl = shorturl.strip(' /')

     9

        try :

    10

            return URL.objects.get(shorturl=shorturl)

    11

        except URL.DoesNotExist :

    12

            raise Http404()

    13

    elif shorturl_id :

    14

        shorturl_id = int(shorturl_id.strip(' /'))

    15

        try :

    16

            return URL.objects.get(id=shorturl_id)

    17

        except URL.DoesNotExist :

    18

            raise Http404()

    19

    else :

    20

        raise Http404()

    21

    22

def index (request) :

    23

    urls = URL.objects.all()

    24

    return render(request, 'qrurls/index.html', dict(

    25

        urls    = urls,

    26

    ))

    27

    28

def shorturl (request, shorturl=None, shorturl_id=None) :

    29

    url_feed = _get_url_feed(shorturl=shorturl, shorturl_id=shorturl_id)

    30

    31

    # most recent

    32

    url_item = url_feed.active_item()

    33

    34

    if not url_item :

    35

        raise Http404()

    36

    38

    url = url_item.get_absolute_url()

    39

    40

    return HttpResponseRedirect(url)

    41

    42

def image (request, shorturl, item_id) :

    43

    url_feed = _get_url_feed(shorturl=shorturl)

    44

    try :

    45

        url_item = URLItem.objects.get(shorturl=url_feed, id=int(item_id))

    46

    except URLItem.DoesNotExist :

    47

        raise Http404()

    57

        raise Http404()