models.py

import datetime
import re

import misaka
import pytz
try:  # django <= 1.6
    from django.core.urlresolvers import reverse
except ImportError:  # from django 1.7 to django 2.0 (and more)
    from django.urls import reverse
from django.db import models
from django.utils.text import slugify
from django.utils.translation import ugettext_lazy as _
from django_markdown.models import MarkdownField

from knowledge_share.conf import KNOWLEDGE_APP_NAME

URL_REGEX_AND_SPACES = '\s+http[s]?://(?:[a-zA-Z]|[0-9]|[$-_@.&+]|[!*\(\),]|(?:%[0-9a-fA-F][0-9a-fA-F]))+\s'


class MicroBlogCategoryBase(models.Model):
    name = models.CharField(max_length=200)

    class Meta:
        verbose_name_plural = _("categories")
        abstract = True

    @property
    def hashtag(self):
        joined_name = ''.join(self.name.title().split())
        return '#{}'.format(joined_name)

    def __str__(self):
        return self.name


class MicroBlogPostBase(models.Model):
    title = models.CharField(
        blank=True,
        max_length=100,
        verbose_name=_('Title')
    )
    slug = models.SlugField(
        _('slug'),
        max_length=255,
        blank=True,
        db_index=True,
        unique=True
    )
    content = MarkdownField()
    pub_date = models.DateTimeField(verbose_name=_('date published'))
    # This should be 'categories' but keeping it for backward compatibility
    category = models.ManyToManyField(KNOWLEDGE_APP_NAME + '.MicroBlogCategory')
    posted_on_twitter = models.BooleanField(default=False)

    class Meta:
        abstract = True
        verbose_name_plural = _("posts")

    def _remove_url_from_content(self):
        return re.sub(URL_REGEX_AND_SPACES, ' ', self.content)

    def _content_to_slug(self):
        content = self._remove_url_from_content()
        new_slug = re.sub('<[^<]+?>', '', misaka.html(content))
        new_slug = new_slug.split()
        new_slug = '-'.join(new_slug[:6])
        return new_slug

    def get_absolute_url(self):
        return reverse(KNOWLEDGE_APP_NAME + ':microblog-post', kwargs={'slug': self.slug})

    def save(self, *args, **kwargs):
        if 'timezone' in kwargs:
            tz = pytz.timezone(kwargs['timezone'])
            pub_date = tz.localize(pub_date)
        else:
            pub_date = pytz.utc.localize(datetime.datetime.now())

        if not self.id:
            self.slug = slugify(self._content_to_slug())
            self.pub_date = pub_date
        super(MicroBlogPostBase, self).save(*args, **kwargs)