42 lines
1.4 KiB
Python
42 lines
1.4 KiB
Python
from django.db import models
|
|
from django.urls import reverse
|
|
from django.utils.text import slugify
|
|
from utils.models import CreatedUpdatedModel
|
|
from utils.slugs import unique_slugify
|
|
|
|
|
|
class NewsItem(CreatedUpdatedModel):
|
|
class Meta:
|
|
ordering = ["-published_at"]
|
|
|
|
title = models.CharField(max_length=100)
|
|
content = models.TextField()
|
|
published_at = models.DateTimeField(null=True, blank=True)
|
|
slug = models.SlugField(max_length=255, blank=True)
|
|
archived = models.BooleanField(default=False)
|
|
|
|
def __str__(self):
|
|
return self.title
|
|
|
|
def save(self, **kwargs):
|
|
if self.published_at:
|
|
# if this is a new newsitem, or it doesn't have a slug, or the slug is in use on another item, create a new slug
|
|
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)
|
|
self.slug = unique_slugify(
|
|
slug,
|
|
slugs_in_use=self.__class__.objects.all().values_list(
|
|
"slug", flat=True
|
|
),
|
|
)
|
|
super().save(**kwargs)
|
|
|
|
def get_absolute_url(self):
|
|
return reverse("news:detail", kwargs={"slug": self.slug})
|