pbp_com/apps/historical/models.py

from datetime import datetime
from django.db import models
from django.utils.translation import ugettext_lazy as _

from django.contrib.sites.models import Site
from django.contrib.auth.models import User
from django_extensions.db.models import TimeStampedModel
from historical.managers import PublishedManager
from newsroom.models import Story, WebEdition

try:
    from adzone.models import Advertiser, BannerAd
    ADVERTISER_FIELD=models.ForeignKey(Advertiser, blank=True, null=True)
    ADVERT_FIELD=models.ForeignKey(BannerAd, blank=True, null=True)
except:
    ADVERTISER_FIELD=models.CharField(_('Advertiser'), blank=True, null=True, max_length=140)
    ADVERT_FIELD=models.ImageField(_('Advertisment'), blank=True, null=True, upload_to='historical/ads/')

class PublishModel(TimeStampedModel):
    published=models.BooleanField(_('Published'), default=False)
    published_on=models.DateTimeField(_('Published on'), default=datetime.now())

    objects = PublishedManager()

    class Meta:
        abstract=True

class Group(PublishModel):
    '''Group model.
    Holds historical objects for organizational purposes.'''
    title=models.CharField(_('Title'), max_length=140)
    slug=models.SlugField(_('Slug'), unique=True)
    graphic=models.ImageField(_('Graphic'), upload_to='historical/group/', blank=True, null=True)
    description=models.TextField(_('Description'), blank=True, null=True)
    sites=models.ManyToManyField(Site)
    template_name = models.CharField(_('template name'), max_length=200, blank=True, help_text=_("Example: 'historical/group/special_history_section.html'. If this isn't provided, the system will use 'historical/group/default.html'."))
    created_by=models.ForeignKey(User, editable=False)

    class Meta:
        ordering=('title',)
        verbose_name = _('Group')
        verbose_name_plural = _('Groups')
        
    def __unicode__(self):
        return u'%s' % (self.title)    
        
    @models.permalink
    def get_absolute_url(self):
        return ('hi-group-detail', None, {'slug': self.slug})

class HistoricArticle(PublishModel):
    '''Historic Article model.
       An historic article, belongs to a group and holds a story and an advertiser.'''
    story=models.ForeignKey(Story)
    group=models.ForeignKey(Group)
    edition=models.ForeignKey(WebEdition, blank=True, null=True)
    description=models.TextField(_('Description'), blank=True, null=True)
    #advertisment=models.ImageField(_('Ad'), blank=True, null=True, upload_to='historical/ads/')
    advert=ADVERT_FIELD
    advertiser=ADVERTISER_FIELD
    view_count = models.PositiveIntegerField(default=0, editable=False)
    created_by=models.ForeignKey(User, editable=False)
        
    class Meta:
        ordering=('published_on',)
        verbose_name = _('Historic article')
        verbose_name_plural = _('Historic articles')
            
    def __unicode__(self):
        return u'%s' % self.story.web_hed
        
    @models.permalink 
    def get_absolute_url(self, site=''):
        return ('hi-article-detail', (), {
                'group_slug': self.group.slug,
                'year': self.story.published_on.year,
                'month': self.story.published_on.strftime('%b').lower(),
                'day': self.story.published_on.day,
                'slug': self.story.slug})