qrurls/models.py
author Tero Marttila <terom@paivola.fi>
Mon, 15 Jul 2013 23:46:45 +0200
changeset 19 0742539e94c3
parent 18 dd08de6f1fe3
child 20 2d45215de95a
permissions -rw-r--r--
shorter urls
from django.db import models
from django.core.urlresolvers import reverse
import django.utils.http
from django.contrib.sites.models import get_current_site
from django.utils import timezone

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

class URL(models.Model):
    shorturl = models.SlugField(unique=True)

    class Meta:
        verbose_name = u"URL Feed"
        verbose_name_plural = u"URL Feeds"

    def qrcode_img (self, size=512) :
        return QRCODE_API.format(
                width=size, height=size,
                url=django.utils.http.urlquote(self.qrcode_url()),
        )

    def qrcode_url (self) :
        return 'http://{domain}{url}'.format(
                domain = get_current_site(None).domain,
                url = self.get_short_url(),
        ).upper()

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

    def get_short_url (self) :
        return reverse('minurl', args=[self.id])

    def active_item(self, now=None) :
        """Currently published URLItem."""
        if now is None :
            now = timezone.now()

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

    def upcoming_item(self, now=None) :
        """Following published URLItem."""
        if now is None :
            now = timezone.now()

        try :
            return URLItem.objects.filter(shorturl=self, published__gt=now).order_by('published')[0]
        except IndexError :
            return None

    def __unicode__ (self) :
        return self.shorturl

class URLItem(models.Model):
    shorturl = models.ForeignKey(URL)
    published = models.DateTimeField()
    url = models.URLField()
    
    class Meta:
        verbose_name = u"URL Item"
        verbose_name_plural = u"URL Items"
        ordering = ['published']

    def get_absolute_url (self) :
        return self.url
    
    def published_age (self) :
        now = timezone.now()

        if now > self.published:
            td = now - self.published
        else :
            td = self.published - now

        days, seconds = td.days, td.seconds
        m, s = divmod(seconds, 60)
        h, m = divmod(m, 60)

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

    def published_state (self) :
        date = self.published.strftime("%Y-%m-%d")
        published, days, age = self.published_age()

        if published and days :
            return "[{date}]".format(date=date)
        elif published :
            return "[{age}]".format(age=age)
        elif days :
            return "({when})".format(when=date)
        else :
            return "({age})".format(age=age)

    def __unicode__ (self) :
        return u"{published_state} {url}".format(
                published_state=self.published_state(),
                url=self.get_absolute_url(),
        )