aircox/aircox_cms/models.py

from django.db import models
from django.contrib.auth.models import User
from django.contrib.contenttypes.fields import GenericForeignKey
from django.contrib.contenttypes.models import ContentType
from django.utils import timezone
from django.utils.text import slugify
from django.utils.translation import ugettext as _, ugettext_lazy
from django.core.urlresolvers import reverse

from django.db.models.signals import post_init, post_save, post_delete
from django.dispatch import receiver

from taggit.managers import TaggableManager

class Thread (models.Model):
    """
    Object assigned to any Post and children that can be used to have parent and
    children relationship between posts of different kind.

    We use this system instead of having directly a GenericForeignKey into the
    Post because it avoids having to define the relationship with two models for
    routing (one for the parent and one for the children).
    """
    post_type = models.ForeignKey(ContentType)
    post_id = models.PositiveIntegerField()
    post = GenericForeignKey('post_type', 'post_id')

    __initial_post = None

    @classmethod
    def __get_query_set (cl, function, model, post, kwargs):
        if post:
            model = type(post)
            kwargs['post_id'] = post.id

        kwargs['post_type'] = ContentType.objects.get_for_model(model)
        return getattr(cl.objects, function)(**kwargs)

    @classmethod
    def get (cl, model = None, post = None, **kwargs):
        return cl.__get_query_set('get', model, post, kwargs)

    @classmethod
    def filter (cl, model = None, post = None, **kwargs):
        return self.__get_query_set('filter', model, post, kwargs)

    @classmethod
    def exclude (cl, model = None, post = None, **kwargs):
        return self.__get_query_set('exclude', model, post, kwargs)

    def save (self, *args, **kwargs):
        self.post = self.__initial_post or self.post
        super().save(*args, **kwargs)

    def __str__ (self):
        return self.post_type.name + ': ' + str(self.post)


class Post (models.Model):
    thread = models.ForeignKey(
        Thread,
        on_delete=models.SET_NULL,
        blank = True, null = True,
        help_text = _('the publication is posted on this thread'),
    )
    author = models.ForeignKey(
        User,
        verbose_name = _('author'),
        blank = True, null = True,
    )
    date = models.DateTimeField(
        _('date'),
        default = timezone.datetime.now
    )
    published = models.BooleanField(
        verbose_name = _('public'),
        default = True
    )

    title = models.CharField (
        _('title'),
        max_length = 128,
    )
    content = models.TextField (
        _('description'),
        blank = True, null = True
    )
    image = models.ImageField(
        blank = True, null = True
    )
    tags = TaggableManager(
        _('tags'),
        blank = True,
    )

    def detail_url (self):
        return reverse(self._meta.verbose_name_plural.lower() + '_detail',
                       kwargs = { 'pk': self.pk,
                                  'slug': slugify(self.title) })

    class Meta:
        abstract = True


class Article (Post):
    static_page = models.BooleanField(
        _('static page'),
        default = False,
    )
    focus = models.BooleanField(
        _('article is focus'),
        default = False,
    )

    class Meta:
        verbose_name = _('Article')
        verbose_name_plural = _('Articles')


class RelatedPostBase (models.base.ModelBase):
    """
    Metaclass for RelatedPost children.
    """
    def __new__ (cls, name, bases, attrs):
        rel = attrs.get('Relation')
        rel = (rel and rel.__dict__) or {}

        related_model = rel.get('related_model')
        if related_model:
            attrs['related'] = models.ForeignKey(related_model)

        if not '__str__' in attrs:
            attrs['__str__'] = lambda self: str(self.related)

        if name is not 'RelatedPost':
            _relation = RelatedPost.Relation()
            _relation.__dict__.update(rel)
            attrs['_relation'] = _relation

        return super().__new__(cls, name, bases, attrs)


class RelatedPost (Post, metaclass = RelatedPostBase):
    related = None

    class Meta:
        abstract = True

    class Relation:
        related_model = None
        mapping = None          # dict of related mapping values
        bind_mapping = False    # update fields of related data on save


    def get_attribute (self, attr):
        attr = self._relation.mappings.get(attr)
        return self.related.__dict__[attr] if attr else None

    def save (self, *args, **kwargs):
        if not self.title and self.related:
            self.title = self.get_attribute('title')

        if self._relation.bind_mapping:
            self.related.__dict__.update({
                rel_attr: self.__dict__[attr]
                for attr, rel_attr in self.Relation.mapping
            })

            self.related.save()

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


@receiver(post_init)
def on_thread_init (sender, instance, **kwargs):
    if not issubclass(Thread, sender):
        return
    instance.__initial_post = instance.post

@receiver(post_save)
def on_post_save (sender, instance, created, *args, **kwargs):
    if not issubclass(sender, Post) or not created:
        return

    thread = Thread(post = instance)
    thread.save()

@receiver(post_delete)
def on_post_delete (sender, instance, using, *args, **kwargs):
    try:
        Thread.get(sender, post = instance).delete()
    except:
        pass