1

from django.db import models

     2

from django.core.urlresolvers import reverse

     3

import django.utils.http

     4

from django.contrib.sites.models import get_current_site

     5

from django.utils import timezone

     6

     7

QRCODE_API = 'https://chart.googleapis.com/chart?cht=qr&chs={width}x{height}&chl={url}'

     8

     9

class URL(models.Model):

    10

    shorturl = models.SlugField(unique=True)

    11

    12

    class Meta:

    13

        verbose_name = u"URL"

    14

        verbose_name_plural = u"URLs"

    15

    16

    def qrcode_img (self, size=512) :

    17

        return QRCODE_API.format(

    18

                width=size, height=size,

    19

                url=django.utils.http.urlquote(self.qrcode_url()),

    20

        )

    21

    22

    def qrcode_url (self) :

    23

        return 'http://{domain}{url}'.format(

    24

                domain = get_current_site(None).domain,

    25

                url = self.get_absolute_url(),

    26

        )

    27

    28

    def get_absolute_url (self) :

    29

        return reverse('shorturl', args=[self.shorturl])

    30

    31

    def active_item(self, now=None) :

    32

        """Currently published URLItem."""

    33

        if now is None :

    34

            now = timezone.now()

    35

    36

        return URLItem.objects.filter(shorturl=self, published__lt=now).order_by('-published')[0]

    37

    38

    def upcoming_item(self, now=None) :

    39

        """Following published URLItem."""

    40

        if now is None :

    41

            now = timezone.now()

    42

    43

        return URLItem.objects.filter(shorturl=self, published__lt=now).order_by('published')[0]

    44

    45

    def __unicode__ (self) :

    46

        return self.shorturl

    47

    48

class URLItem(models.Model):

    49

    shorturl = models.ForeignKey(URL)

    50

    published = models.DateTimeField()

    51

    url = models.URLField()

    52

    53

    class Meta:

    54

        verbose_name = u"URL Item"

    55

        verbose_name_plural = u"URL Items"

    56

        ordering = ['published']

    57

    58

    def get_absolute_url (self) :

    59

        return self.url

    60

    61

    def published_age (self) :

    62

        now = timezone.now()

    63

    64

        if now > self.published:

    65

            td = now - self.published

    66

        else :

    67

            td = self.published - now

    68

    69

        days, seconds = td.days, td.seconds

    70

        m, s = divmod(seconds, 60)

    71

        h, m = divmod(m, 60)

    72

    74

    75

    def published_state (self) :

    78

    83

        elif days :

    85

        else :

    87

    88

    def __unicode__ (self) :

    91

                url=self.get_absolute_url(),

    92

        )