view smiley/models.py @ 821:71db8076dc3d

Bandmap WIP: geocoding integrated with add form. Add form works. Before submitting the form, client side JS makes a geocode request to Google and populates hidden lat/lon fields with the result. Successfully created a model instance on the server side. Still need to update admin dashboard, admin approval, and give out badges for adding bands to the map. Once that is done, then work on displaying the map with filtering.
author Brian Neal <bgneal@gmail.com>
date Tue, 23 Sep 2014 20:40:31 -0500
parents 7429c98c8ece
children e14f54f16dbc
line wrap: on
line source
"""
Models for the smiley application.
"""
import re

from django.db import models
from django.core.cache import cache
from django.contrib.sites.models import Site

CACHE_TIMEOUT = 60 * 60      # seconds


class SmileyManager(models.Manager):

    def get_smiley_map(self):
        """
        Returns a dictionary, the keys are smiley codes.
        The values are the HTML representations of the keys.
        The dictionary is cached.
        """
        map = cache.get('smiley_map')
        if map:
            return map

        map = dict((s.code, s.html()) for s in self.all())
        cache.set('smiley_map', map, CACHE_TIMEOUT)
        return map

    def get_smilies(self, extra=False):
        """
        Returns smiley model instances filtered by the extra flag.
        """
        key = 'smileys' if not extra else 'smileys_extra'
        smilies = cache.get(key)
        if smilies:
            return smilies

        smilies = self.filter(is_extra=extra)
        cache.set(key, smilies, CACHE_TIMEOUT)
        return smilies

    def get_smiley_regexes(self, relative_urls=True):
        """
        Returns a list of 2-tuples of the form: (regex, repl)
        where regex is a regular expression for a smiley and
        repl is the replacement image in Markdown format.

        If relative_urls is true, the smiley images will use relative URLs. If
        False, absolute URLs will be used.

        """
        key = 'smiley_regexes_rel' if relative_urls else 'smiley_regexes_abs'
        regexes = cache.get(key)
        if regexes:
            return regexes

        regexes = [(re.compile(r"(^|\s|(?<=\s))%s(\s|$)" % re.escape(s.code)),
            r"\1%s\2" % s.markdown(relative_urls=relative_urls)) for s in self.all()]
        cache.set(key, regexes, CACHE_TIMEOUT)
        return regexes


class Smiley(models.Model):
    image = models.ImageField(upload_to='smiley/images/')
    title = models.CharField(max_length=32)
    code = models.CharField(max_length=32)
    is_extra = models.BooleanField()

    objects = SmileyManager()

    class Meta:
        verbose_name_plural = 'Smilies'
        ordering = ('title', )

    def __unicode__(self):
        return self.title

    def get_absolute_url(self):
        return self.image.url

    def html(self):
        """Returns a HTML img tag representation of the smiley."""
        if self.image:
            site = Site.objects.get_current()
            return (u'<img src="http://%s%s" alt="%s" title="%s" />' %
                    (site.domain, self.get_absolute_url(), self.title, self.title))
        return u''
    html.allow_tags = True

    def markdown(self, relative_urls=True):
        """Returns a markdown representation of the smiley.

        If relative_urls is True, relative URLs will be generated. If False,
        absolute URLs will be used.

        """
        if self.image:
            if relative_urls:
                return u'![%s](%s "%s")' % (self.title, self.image.url,
                        self.title)
            else:
                site = Site.objects.get_current()
                return (u'![%s](http://%s%s "%s")' %
                    (self.title, site.domain, self.image.url, self.title))
        return u''