Synnefo » snf-cloudcms

from datetime import datetime

from django.db import models

from django.conf import settings

from django.contrib.auth.models import User

from django.utils.translation import ugettext_lazy as _, ugettext, ungettext

from feincms import translations

from feincms.models import Base

from feincms.content.richtext.models import RichTextContent

from feincms.content.section.models import SectionContent

from feincms.module.medialibrary.fields import MediaFileForeignKey

from feincms.module.medialibrary.models import MediaFile

from feincms.module.page.extensions.navigation import NavigationExtension

from feincms.module.page.extensions.navigation import PagePretender

from cloudcms.models import Application

class Category(models.Model, translations.TranslatedObjectMixin):

    """

    Blog entry category. Each blog post may belong to multiple categories.

    """

    ordering = models.SmallIntegerField(_('ordering'), default=0)

    display_on_menu = models.BooleanField(default=False)

    class Meta:

        verbose_name = _('category')

        verbose_name_plural = _('categories')

        ordering = ['-ordering',]

    objects = translations.TranslatedObjectManager()

    def __unicode__(self):

        trans = translations.TranslatedObjectMixin.__unicode__(self)

        return trans or _('Unnamed category')

class CategoryTranslation(translations.Translation(Category)):

    """

    Category translation

    """

    title = models.CharField(_('category title'), max_length=100)

    slug = models.SlugField(_('slug'), unique=True)

    description = models.CharField(_('description'), max_length=250, blank=True)

    class Meta:

        verbose_name = _('category translation')

        verbose_name_plural = _('category translations')

        ordering = ['title']

    def __unicode__(self):

        return self.title

    @models.permalink

    def get_absolute_url(self):

        return ('cloudcmsblog_entries_archive', (), {

            'category': self.slug,

            })

    def save(self, *args, **kwargs):

        if not self.slug:

            self.slug = slugify(self.title)

        super(CategoryTranslation, self).save(*args, **kwargs)

class EntryManager(models.Manager):

    def active(self):

        return self.filter(is_active=True)

class Entry(Base):

    """

    Blog post entry

    """

    is_active = models.BooleanField(_('is active'), default=True)

    is_featured = models.BooleanField(_('is featured'), default=False)

    title = models.CharField(_('title'), max_length=100)

    slug = models.SlugField(_('slug'), max_length=100, unique_for_date='published_on')

    author = models.ForeignKey(User, related_name='blogentries', verbose_name=_('author'))

    language = models.CharField(max_length=255, choices=settings.LANGUAGES)

    intro_text = models.TextField(max_length=255,

            help_text="Displayed in list views", blank=True)

    image = MediaFileForeignKey(MediaFile, null=True, blank=True)

    application = models.ManyToManyField(Application,

            related_name="blogentries",

            verbose_name=_('application'))

    published_on = models.DateTimeField(_('published on'), blank=True, null=True, default=datetime.now,

        help_text=_('Will be filled in automatically when entry gets published.'))

    last_changed = models.DateTimeField(_('last change'), auto_now=True, editable=False)

    categories = models.ManyToManyField(Category, verbose_name=_('categories'),

        related_name='blogentries', null=True, blank=True)

    objects = EntryManager()

    class Meta:

        get_latest_by = 'published_on'

        ordering = ['-published_on']

        verbose_name = _('entry')

        verbose_name_plural = _('entries')

    def __unicode__(self):

        return self.title

    @models.permalink

    def get_absolute_url(self):

        return ('cloudcmsblog_entry_detail', (), {

            'year': self.published_on.strftime('%Y'),

            'month': self.published_on.strftime('%m'),

            'day': self.published_on.strftime('%d'),

            'slug': self.slug,

            })

# Feincms navigation extension

class BlogCategoriesNavigationExtension(NavigationExtension):

    """

    Navigation extension for FeinCMS which lists all categories that user

    wants to include in global site navigation.

    """

    name = _('blog categories')

    def children(self, page, **kwargs):

        for category in Category.objects.filter(display_on_menu=True):

            url='%scategory/%s/' % (page.get_absolute_url(), category.translation.slug)

            yield PagePretender(

                title=category.translation.title,

                tree_id=page.tree_id,

                url=url,

                lft=0,

                rght=0,

                slug=category.translation.slug,

            )