1

from django.core.cache import cache

     2

from django.http import HttpResponse, HttpResponseRedirect, Http404

     3

from django.shortcuts import render

     4

from django.utils import timezone, http

     5

     6

from qrurls.models import URL, URLItem

     7

     8

import calendar # timegm

     9

    10

"""

    11

    Public frontend UI.

    12

"""

    13

    14

class URLNotFound (Http404):

    15

    pass

    16

    17

class URLItemNotFound (Http404):

    18

    pass

    19

    20

def http_datetime (dt) :

    21

    return http.http_date(calendar.timegm(dt.utctimetuple()))

    22

    23

def _get_url_item (shorturl=None, shorturl_id=None, item_id=None, related=()) :

    24

    # JOIN against shorturl, urlimage

    25

    url_item = URLItem.objects.select_related(*related)

    26

    27

    if shorturl:

    28

        url_item = url_item.filter(shorturl__shorturl=shorturl)

    29

    elif shorturl_id:

    30

        shorturl_id = int(shorturl_id)

    31

        url_item = url_item.filter(shorturl__id=shorturl_id)

    32

    else:

    33

        raise URLNotFound()

    34

    35

    # match for published items

    36

    now = timezone.now()

    37

    url_item = url_item.filter(published__lt=now).order_by('-published')

    38

    39

    if item_id :

    40

        return url_item.get(id=int(item_id))

    41

    else :

    42

        # most recent

    43

        try:

    44

            return url_item[0]

    45

        except IndexError:

    46

            raise URLItemNotFound()

    47

    48

def index (request) :

    49

    urls = URL.objects.all()

    50

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

    51

        urls    = urls,

    52

    ))

    53

    55

    """

    56

        Primary frontend for redirecting based on current time.

    57

    """

    58

    60

    data = cache.get(key)

    61

    62

    if data :

    63

        url, modified = data

    64

    else:

    67

        modified = url_item.last_modified()

    68

        url = url_item.get_absolute_url()

    69

    70

        cache.set(key, (url, modified)) # XXX: expiry

    71

    72

    # redirect, either directly, or to image()

    73

    response = HttpResponseRedirect(url)

    74

    response['Vary'] = ''

    75

    response['Last-Modified'] = http_datetime(modified)

    76

    return response

    77

    79

    """

    80

        Frontend for a specific item.

    81

    """

    87

    88

    if url_item.url :

    89

        response = HttpResponseRedirect(url_item.url)

    90

    elif url_item.image :

    91

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

    92

    93

        response = render(request, 'qrurls/image.html', dict(

    94

            title   = title,

    95

            image   = url_item.image,

    96

        ))

    97

    else :

    98

        raise Http404()

    99

   100

    # set cache

   101

    response['Vary'] = ''

   102

    response['Last-Modified'] = http_datetime(url_item.last_modified())

   103

    return response

   104