bornhack-website/news/models.py

49 lines
1.5 KiB
Python

from __future__ import unicode_literals
from django.db import models
from django.utils.text import slugify
from utils.models import CreatedUpdatedModel
from news.managers import NewsItemQuerySet
class NewsItem(CreatedUpdatedModel):
class Meta:
ordering = ['-published_at']
title = models.CharField(max_length=100)
content = models.TextField()
public = models.BooleanField(default=False)
published_at = models.DateTimeField()
slug = models.SlugField(max_length=255, blank=True)
def __str__(self):
return self.title
def save(self, **kwargs):
if (
not self.pk or
not self.slug or
NewsItem.objects.filter(slug=self.slug).count() > 1
):
published_at_string = self.published_at.strftime('%Y-%m-%d')
base_slug = slugify(self.title)
slug = '{}-{}'.format(published_at_string, base_slug)
incrementer = 1
# We have to make sure that the slug won't clash with current slugs
while NewsItem.objects.filter(slug=slug).exists():
if incrementer == 1:
slug = '{}-1'.format(slug)
else:
slug = '{}-{}'.format(
'-'.join(slug.split('-')[:-1]),
incrementer
)
incrementer += 1
self.slug = slug
super(NewsItem, self).save(**kwargs)
objects = NewsItemQuerySet.as_manager()