Mercurial > public > sg101

annotate gpp/news/models.py @ 334:6805d15cda13

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
Adding a script I had to write on the fly to filter out posts from the posts csv file that had no parent topics. MyISAM let me get away with that, but InnoDB won't.
author Brian Neal <bgneal@gmail.com>
date Sat, 26 Feb 2011 01:28:22 +0000
parents d424b8bae71d
children 53fdaf0da539
rev   line source
gremmie@1 1 """
gremmie@1 2 Models for the news application.
gremmie@1 3 """
gremmie@1 4
gremmie@1 5 import datetime
gremmie@1 6 from django.db import models
bgneal@49 7 from django.contrib.auth.models import User
gremmie@1 8 from tagging.fields import TagField
gremmie@1 9
gremmie@1 10
gremmie@1 11 class Category(models.Model):
bgneal@54 12 """News stories belong to categories"""
bgneal@240 13 title = models.CharField(max_length=64)
bgneal@240 14 slug = models.SlugField(max_length=64)
bgneal@54 15 icon = models.ImageField(upload_to='news/categories/', blank=True)
gremmie@1 16
bgneal@54 17 def __unicode__(self):
bgneal@54 18 return self.title
gremmie@1 19
bgneal@54 20 def num_stories(self):
bgneal@54 21 return News.objects.filter(category = self.pk).count()
gremmie@1 22
bgneal@54 23 class Meta:
bgneal@54 24 verbose_name_plural = 'Categories'
bgneal@54 25 ordering = ('title', )
gremmie@1 26
gremmie@1 27
bgneal@204 28 class StoryBase(models.Model):
bgneal@204 29 """Abstract model to collect common fields."""
bgneal@54 30 title = models.CharField(max_length=255)
bgneal@54 31 submitter = models.ForeignKey(User)
bgneal@54 32 category = models.ForeignKey(Category)
bgneal@54 33 short_text = models.TextField()
bgneal@54 34 long_text = models.TextField(blank=True)
bgneal@204 35 date_submitted = models.DateTimeField(db_index=True)
bgneal@54 36 allow_comments = models.BooleanField(default=True)
bgneal@54 37 tags = TagField()
bgneal@218 38 front_page_expiration = models.DateField(null=True, blank=True)
bgneal@277 39 update_date = models.DateTimeField(db_index=True, blank=True)
gremmie@1 40
bgneal@204 41 class Meta:
bgneal@204 42 abstract = True
bgneal@204 43
bgneal@204 44
bgneal@204 45 class PendingStory(StoryBase):
bgneal@204 46 """Stories submitted by users are held pending admin approval"""
bgneal@204 47
bgneal@182 48 def save(self, *args, **kwargs):
bgneal@204 49 if not self.pk:
bgneal@277 50 if not self.date_submitted:
bgneal@277 51 self.date_submitted = datetime.datetime.now()
bgneal@277 52 self.update_date = self.date_submitted
bgneal@277 53 else:
bgneal@277 54 self.update_date = datetime.datetime.now()
bgneal@204 55
bgneal@204 56 super(PendingStory, self).save(*args, **kwargs)
gremmie@1 57
bgneal@54 58 def __unicode__(self):
bgneal@54 59 return self.title
gremmie@1 60
bgneal@54 61 class Meta:
bgneal@54 62 ordering = ('-date_submitted', )
bgneal@54 63 verbose_name_plural = 'Pending Stories'
gremmie@1 64
gremmie@1 65
bgneal@204 66 class Story(StoryBase):
bgneal@54 67 """Model for news stories"""
gremmie@1 68
bgneal@54 69 @models.permalink
bgneal@54 70 def get_absolute_url(self):
bgneal@54 71 return ('news.views.story', [str(self.id)])
gremmie@1 72
bgneal@54 73 def __unicode__(self):
bgneal@54 74 return self.title
gremmie@1 75
bgneal@54 76 class Meta:
bgneal@204 77 ordering = ('-date_submitted', )
bgneal@226 78 verbose_name = 'news story'
bgneal@226 79 verbose_name_plural = 'news stories'
gremmie@1 80
bgneal@277 81 def save(self, *args, **kwargs):
bgneal@277 82 if not self.pk:
bgneal@277 83 self.date_submitted = datetime.datetime.now()
bgneal@277 84 self.update_date = self.date_submitted
bgneal@277 85 else:
bgneal@277 86 self.update_date = datetime.datetime.now()
bgneal@277 87
bgneal@277 88 super(Story, self).save(*args, **kwargs)
bgneal@277 89
bgneal@54 90 def can_comment_on(self):
bgneal@54 91 now = datetime.datetime.now()
bgneal@204 92 delta = now - self.date_submitted
bgneal@204 93 return self.allow_comments and delta.days < 30
bgneal@220 94
bgneal@220 95 def search_title(self):
bgneal@220 96 return self.title
bgneal@220 97
bgneal@220 98 def search_summary(self):
bgneal@220 99 return u"\n".join((self.short_text, self.long_text))