1

from django.http import HttpResponse, HttpResponseRedirect, Http404

     2

from django.shortcuts import render

     3

from django.utils import timezone

     4

     5

from qrurls.models import URL, URLItem

     6

    11

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

    12

    if shorturl :

    13

        shorturl = shorturl.strip(' /')

    14

        try :

    15

            return URL.objects.get(shorturl=shorturl)

    16

        except URL.DoesNotExist :

    17

            raise Http404()

    18

    elif shorturl_id :

    19

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

    20

        try :

    21

            return URL.objects.get(id=shorturl_id)

    22

        except URL.DoesNotExist :

    23

            raise Http404()

    24

    else :

    25

        raise Http404()

    26

    27

def index (request) :

    28

    urls = URL.objects.all()

    29

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

    30

        urls    = urls,

    31

    ))

    32

    33

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

    34

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

    35

    36

    # most recent

    37

    url_item = url_feed.active_item()

    38

    39

    if not url_item :

    40

        raise Http404()

    41

    42

    # redirect, either directly, or to image()

    43

    url = url_item.get_absolute_url()

    44

    45

    return HttpResponseRedirect(url)

    46

    47

def item (request, shorturl, item_id) :

    48

    url_feed = _get_url_feed(shorturl=shorturl)

    49

    50

    now = timezone.now()

    51

    try :

    52

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

    53

    except URLItem.DoesNotExist :

    54

        raise Http404()

    55

    56

    if url_item.url :

    57

        return HttpResponseRedirect(url_item.url)

    58

    elif url_item.image :

    59

        title = url_item.image.title or url_item.shorturl.title

    60

    61

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

    62

            title   = title,

    63

            image   = url_item.image,

    64

        ))

    65

    else :

    66

        raise Http404()