1

from django.http import HttpResponse, HttpResponseRedirect, Http404

     2

from django.shortcuts import render

     4

     5

from qrurls.models import URL, URLItem

     6

     7

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

     8

    if shorturl :

     9

        shorturl = shorturl.strip(' /')

    10

        try :

    11

            return URL.objects.get(shorturl=shorturl)

    12

        except URL.DoesNotExist :

    13

            raise Http404()

    14

    elif shorturl_id :

    15

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

    16

        try :

    17

            return URL.objects.get(id=shorturl_id)

    18

        except URL.DoesNotExist :

    19

            raise Http404()

    20

    else :

    21

        raise Http404()

    22

    23

def index (request) :

    24

    urls = URL.objects.all()

    25

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

    26

        urls    = urls,

    27

    ))

    28

    29

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

    30

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

    31

    32

    # most recent

    33

    url_item = url_feed.active_item()

    34

    35

    if not url_item :

    36

        raise Http404()

    37

    38

    # redirect, either directly, or to image()

    39

    url = url_item.get_absolute_url()

    40

    41

    return HttpResponseRedirect(url)

    42

    43

def image (request, shorturl, item_id) :

    44

    url_feed = _get_url_feed(shorturl=shorturl)

    47

    try :

    49

    except URLItem.DoesNotExist :

    50

        raise Http404()

    51

    52

    if url_item.image :

    53

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

    54

            title   = url_item.image.title,

    55

            image   = url_item.image,

    56

        ))

    57

    elif url_item.url :

    58

        return HttpResponseRedirect(url_item.url)

    59

    else :

    60

        raise Http404()