Mercurial > public > sg101

annotate gpp/potd/models.py @ 408:7e0997b08b50

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
Fixes for #200; fix bug in deciding which forums have unread posts.
author Brian Neal <bgneal@gmail.com>
date Sat, 02 Apr 2011 00:12:40 +0000
parents 3c951521e0ec
children 7dbdbb08e68c
rev   line source
gremmie@1 1 """
gremmie@1 2 Models for the Photo Of The Day (POTD) application.
gremmie@1 3 """
gremmie@1 4 import os
gremmie@1 5 from PIL import ImageFile
gremmie@1 6 from PIL import Image
gremmie@1 7 try:
gremmie@1 8 from cStringIO import StringIO
gremmie@1 9 except:
gremmie@1 10 from StringIO import StringIO
gremmie@1 11
gremmie@1 12 from django.db import models
gremmie@1 13 from django.contrib.auth.models import User
gremmie@1 14 from django.core.files.base import ContentFile
gremmie@1 15
gremmie@1 16 POTD_THUMB_WIDTH = 120
gremmie@1 17
gremmie@1 18 def scale_image(image):
gremmie@1 19 (w, h) = image.size
gremmie@1 20 if w <= POTD_THUMB_WIDTH:
gremmie@1 21 return image
gremmie@1 22 scale_factor = float(POTD_THUMB_WIDTH) / w
gremmie@1 23 new_height = int(scale_factor * h)
gremmie@1 24 return image.resize((POTD_THUMB_WIDTH, new_height), Image.ANTIALIAS)
gremmie@1 25
gremmie@1 26
gremmie@1 27 class Photo(models.Model):
gremmie@1 28 """Model to represent a POTD"""
gremmie@1 29 photo = models.ImageField(upload_to='potd/%Y/%m/%d')
gremmie@1 30 thumb = models.ImageField(upload_to='potd/%Y/%m/%d/thumbs', blank=True, null=True)
gremmie@1 31 caption = models.CharField(max_length=128)
gremmie@1 32 description = models.TextField()
gremmie@1 33 user = models.ForeignKey(User)
gremmie@1 34 date_added = models.DateField(auto_now_add=True)
gremmie@1 35 potd_count = models.IntegerField(default=0)
gremmie@1 36
bgneal@14 37 class Meta:
bgneal@14 38 ordering = ('-date_added', '-caption')
bgneal@14 39
gremmie@1 40 def __unicode__(self):
gremmie@1 41 return u'%s (%s)' % (self.caption, self.pk)
gremmie@1 42
bgneal@14 43 @models.permalink
bgneal@14 44 def get_absolute_url(self):
bgneal@14 45 return ('potd-archive', [str(self.id)])
gremmie@1 46
bgneal@182 47 def save(self, *args, **kwargs):
bgneal@330 48 if not self.pk:
bgneal@330 49 self.generate_thumb()
gremmie@1 50
bgneal@330 51 super(Photo, self).save(*args, **kwargs)
bgneal@330 52 Sequence.objects.insert_photo(self.pk)
bgneal@330 53
bgneal@330 54 def delete(self):
bgneal@330 55 Sequence.objects.remove_photo(self.pk)
bgneal@330 56 super(Photo, self).delete()
bgneal@330 57
bgneal@330 58 def can_comment_on(self):
bgneal@330 59 return Current.objects.get_current_id() == self.id
bgneal@330 60
bgneal@330 61 def generate_thumb(self):
gremmie@1 62 if self.thumb:
gremmie@1 63 self.thumb.delete(save=False)
gremmie@1 64
gremmie@1 65 parser = ImageFile.Parser()
gremmie@1 66 for chunk in self.photo.chunks():
gremmie@1 67 parser.feed(chunk)
gremmie@1 68 image = parser.close()
gremmie@1 69 format = image.format
gremmie@1 70 image = scale_image(image)
gremmie@1 71 s = StringIO()
gremmie@1 72 image.save(s, format)
gremmie@1 73 thumb_name = os.path.basename(self.photo.path)
gremmie@1 74 self.thumb.save(thumb_name, ContentFile(s.getvalue()), save=False)
gremmie@1 75
gremmie@1 76
gremmie@1 77 class CurrentManager(models.Manager):
gremmie@1 78 def get_current_photo(self):
gremmie@1 79 try:
gremmie@1 80 c = self.get(pk=1)
gremmie@1 81 return c.potd
gremmie@1 82 except Current.DoesNotExist:
gremmie@1 83 return None
gremmie@1 84
gremmie@1 85 def get_current_id(self):
gremmie@1 86 potd = self.get_current_photo()
gremmie@1 87 if potd is not None:
gremmie@1 88 return potd.pk
gremmie@1 89 return None
gremmie@1 90
gremmie@1 91
gremmie@1 92 class Current(models.Model):
gremmie@1 93 """This model simply stores the current POTD."""
gremmie@1 94 potd = models.ForeignKey(Photo)
gremmie@1 95
gremmie@1 96 objects = CurrentManager()
gremmie@1 97
gremmie@1 98 def __unicode__(self):
gremmie@1 99 return self.potd.__unicode__()
gremmie@1 100
gremmie@1 101 class Meta:
gremmie@1 102 verbose_name_plural = 'Current'
gremmie@1 103
gremmie@1 104
gremmie@1 105 class SequenceManager(models.Manager):
gremmie@1 106 def insert_photo(self, photo_id):
gremmie@1 107 current = Current.objects.get_current_id()
gremmie@1 108 if current is not None:
bgneal@330 109 s = self.get(pk=1)
bgneal@330 110 seq = [int(x) for x in s.seq.split(',')]
bgneal@330 111 if photo_id not in seq:
bgneal@330 112 i = seq.index(current)
bgneal@330 113 seq.insert(i + 1, photo_id)
bgneal@330 114 s.seq = ','.join([str(x) for x in seq])
bgneal@330 115 s.save()
gremmie@1 116
gremmie@1 117 def remove_photo(self, photo_id):
gremmie@1 118 try:
gremmie@1 119 s = self.get(pk=1)
bgneal@330 120 except Sequence.DoesNotExist:
bgneal@330 121 pass
bgneal@330 122 else:
gremmie@1 123 seq = [int(x) for x in s.seq.split(',')]
gremmie@1 124 if photo_id in seq:
gremmie@1 125 seq.remove(photo_id)
gremmie@1 126 s.seq = ','.join([str(x) for x in seq])
gremmie@1 127 s.save()
gremmie@1 128
gremmie@1 129
gremmie@1 130 class Sequence(models.Model):
gremmie@1 131 """This model stores the sequence of photos for the POTD."""
gremmie@1 132 seq = models.CommaSeparatedIntegerField(max_length=4096)
gremmie@1 133
gremmie@1 134 objects = SequenceManager()
gremmie@1 135
gremmie@1 136 def __unicode__(self):
gremmie@1 137 return self.seq
gremmie@1 138
gremmie@1 139 class Meta:
gremmie@1 140 verbose_name_plural = 'Sequence'
gremmie@1 141