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

     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

    37

    def __unicode__ (self) :

    38

        return self.shorturl

    39

    40

class URLItem(models.Model):

    41

    shorturl = models.ForeignKey(URL)

    42

    published = models.DateTimeField()

    43

    url = models.URLField()

    44

    45

    class Meta:

    46

        verbose_name = u"URL Item"

    47

        verbose_name_plural = u"URL Items"

    48

        ordering = ['published']

    49

    50

    def get_absolute_url (self) :

    51

        return self.url

    52

    79

    def __unicode__ (self) :

    80

        return self.url