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}&chld=H'

     8

     9

class URL(models.Model):

    10

    shorturl = models.SlugField(unique=True)

    11

    12

    class Meta:

    13

        verbose_name = u"URL Feed"

    14

        verbose_name_plural = u"URL Feeds"

    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) :

    25

                url = self.get_maxi_url(),

    27

    28

    def get_absolute_url (self) :

    29

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

    30

    31

    def get_short_url (self) :

    32

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

    33

    34

    def get_maxi_url (self) :

    35

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

    36

    37

    def active_item(self, now=None) :

    38

        """Currently published URLItem."""

    39

        if now is None :

    40

            now = timezone.now()

    41

    42

        try :

    43

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

    44

        except IndexError :

    45

            return None

    46

    47

    def upcoming_item(self, now=None) :

    48

        """Following published URLItem."""

    49

        if now is None :

    50

            now = timezone.now()

    51

    52

        try :

    53

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

    54

        except IndexError :

    55

            return None

    56

    57

    def __unicode__ (self) :

    58

        return self.shorturl

    59

    60

class URLItem(models.Model):

    61

    shorturl = models.ForeignKey(URL)

    62

    published = models.DateTimeField()

    63

    url = models.URLField()

    64

    65

    class Meta:

    66

        verbose_name = u"URL Item"

    67

        verbose_name_plural = u"URL Items"

    68

        ordering = ['published']

    69

    70

    def get_absolute_url (self) :

    71

        return self.url

    72

    73

    def published_age (self) :

    74

        now = timezone.now()

    75

    76

        if now > self.published:

    77

            td = now - self.published

    78

        else :

    79

            td = self.published - now

    80

    81

        days, seconds = td.days, td.seconds

    82

        m, s = divmod(seconds, 60)

    83

        h, m = divmod(m, 60)

    84

    85

        return (self.published < now), days, "{h}h {m}m {s}s".format(h=h, m=m, s=s)

    86

    87

    def published_state (self) :

    88

        date = self.published.strftime("%Y-%m-%d")

    89

        published, days, age = self.published_age()

    90

    91

        if published and days :

    92

            return "[{date}]".format(date=date)

    93

        elif published :

    94

            return "[{age}]".format(age=age)

    95

        elif days :

    96

            return "({when})".format(when=date)

    97

        else :

    98

            return "({age})".format(age=age)

    99

   100

    def __unicode__ (self) :

   101

        return u"{published_state} {url}".format(

   102

                published_state=self.published_state(),

   103

                url=self.get_absolute_url(),

   104

        )