88 lines
3.1 KiB
Python
88 lines
3.1 KiB
Python
from django.db import models
|
|
from django.urls import reverse
|
|
from django.utils import timezone
|
|
from django.utils.text import slugify
|
|
|
|
|
|
class Topic(models.Model):
|
|
name = models.CharField(max_length=80, unique=True)
|
|
slug = models.SlugField(max_length=90, unique=True)
|
|
description = models.TextField(blank=True)
|
|
accent_color = models.CharField(max_length=7, default="#00c2a8")
|
|
|
|
class Meta:
|
|
ordering = ["name"]
|
|
|
|
def __str__(self):
|
|
return self.name
|
|
|
|
def save(self, *args, **kwargs):
|
|
if not self.slug:
|
|
self.slug = slugify(self.name)
|
|
super().save(*args, **kwargs)
|
|
|
|
|
|
class NewsSource(models.Model):
|
|
name = models.CharField(max_length=120, unique=True)
|
|
slug = models.SlugField(max_length=140, unique=True)
|
|
site_url = models.URLField(blank=True)
|
|
feed_url = models.URLField(unique=True)
|
|
description = models.TextField(blank=True)
|
|
is_active = models.BooleanField(default=True)
|
|
last_synced_at = models.DateTimeField(blank=True, null=True)
|
|
created_at = models.DateTimeField(auto_now_add=True)
|
|
|
|
class Meta:
|
|
ordering = ["name"]
|
|
|
|
def __str__(self):
|
|
return self.name
|
|
|
|
def save(self, *args, **kwargs):
|
|
if not self.slug:
|
|
self.slug = slugify(self.name)
|
|
super().save(*args, **kwargs)
|
|
|
|
|
|
class Article(models.Model):
|
|
class ArticleKind(models.TextChoices):
|
|
RSS = "rss", "RSS import"
|
|
ORIGINAL = "original", "Original story"
|
|
|
|
title = models.CharField(max_length=220)
|
|
slug = models.SlugField(max_length=240, unique=True)
|
|
excerpt = models.TextField(blank=True)
|
|
content = models.TextField(blank=True)
|
|
article_kind = models.CharField(max_length=20, choices=ArticleKind.choices, default=ArticleKind.RSS)
|
|
topic = models.ForeignKey(Topic, on_delete=models.SET_NULL, null=True, blank=True, related_name='articles')
|
|
source = models.ForeignKey(NewsSource, on_delete=models.SET_NULL, null=True, blank=True, related_name='articles')
|
|
external_url = models.URLField(blank=True)
|
|
image_url = models.URLField(blank=True)
|
|
author_name = models.CharField(max_length=120, blank=True)
|
|
published_at = models.DateTimeField(default=timezone.now)
|
|
dedupe_key = models.CharField(max_length=64, unique=True, blank=True)
|
|
is_featured = models.BooleanField(default=False)
|
|
is_published = models.BooleanField(default=True)
|
|
created_at = models.DateTimeField(auto_now_add=True)
|
|
updated_at = models.DateTimeField(auto_now=True)
|
|
|
|
class Meta:
|
|
ordering = ['-published_at', '-created_at']
|
|
|
|
def __str__(self):
|
|
return self.title
|
|
|
|
def save(self, *args, **kwargs):
|
|
if not self.slug:
|
|
base_slug = slugify(self.title)[:220] or 'story'
|
|
slug = base_slug
|
|
suffix = 2
|
|
while Article.objects.exclude(pk=self.pk).filter(slug=slug).exists():
|
|
slug = f"{base_slug[:210]}-{suffix}"
|
|
suffix += 1
|
|
self.slug = slug
|
|
super().save(*args, **kwargs)
|
|
|
|
def get_absolute_url(self):
|
|
return reverse('article_detail', args=[self.slug])
|