This repository has been archived on 2023-05-31. You can view files and clone it, but cannot push or open issues or pull requests.
python-blog/articles/models.py

165 lines
5 KiB
Python
Raw Normal View History

2020-08-14 16:15:40 +02:00
import re
2020-08-14 15:53:42 +02:00
import markdown
from django.conf import settings
2020-08-14 14:53:30 +02:00
from django.contrib.auth.models import AbstractUser
2020-08-16 19:23:38 +02:00
from django.contrib.contenttypes.models import ContentType
2020-08-14 14:53:30 +02:00
from django.db import models
from django.http import HttpRequest
2020-08-16 19:45:38 +02:00
from django.template.defaultfilters import slugify
2020-08-16 19:23:38 +02:00
from django.urls import reverse
2020-08-16 18:15:19 +02:00
from django.utils import timezone
from markdown.extensions.codehilite import CodeHiliteExtension
2020-08-14 14:53:30 +02:00
2020-08-25 23:06:12 +02:00
from articles.markdown import LazyLoadingImageExtension
2020-08-14 14:53:30 +02:00
class User(AbstractUser):
pass
2020-08-14 15:53:42 +02:00
2020-08-17 09:57:24 +02:00
class ArticleManager(models.Manager):
def get_queryset(self):
return super().get_queryset().filter(page__isnull=True)
2020-08-18 19:51:54 +02:00
class AdminUrlMixin:
def get_admin_url(self):
content_type = ContentType.objects.get_for_model(self.__class__)
return reverse(
"admin:%s_%s_change" % (content_type.app_label, content_type.model),
args=(self.id,),
)
class Article(AdminUrlMixin, models.Model):
2020-08-14 15:53:42 +02:00
DRAFT = "draft"
PUBLISHED = "published"
STATUS_CHOICES = [
(DRAFT, "Draft"),
(PUBLISHED, "Published"),
]
title = models.CharField(max_length=255)
content = models.TextField()
status = models.CharField(max_length=15, choices=STATUS_CHOICES, default=DRAFT)
published_at = models.DateTimeField(null=True, blank=True)
created_at = models.DateTimeField(auto_now_add=True)
updated_at = models.DateTimeField(auto_now=True)
2020-08-17 17:49:19 +02:00
author = models.ForeignKey(User, on_delete=models.PROTECT, default=1)
views_count = models.IntegerField(default=0)
2020-08-21 12:33:03 +02:00
slug = models.SlugField(unique=True, max_length=255)
comments_allowed = models.BooleanField(default=True)
2020-08-14 15:53:42 +02:00
objects = models.Manager()
without_pages = ArticleManager()
2020-08-17 09:57:24 +02:00
2020-08-14 15:53:42 +02:00
class Meta:
ordering = ["-published_at"]
2020-08-16 18:15:19 +02:00
def __str__(self):
type_ = "Article"
if hasattr(self, "page"):
type_ = "Page"
return f"{self.title} ({type_})"
2020-08-16 18:15:19 +02:00
2020-08-16 19:23:38 +02:00
def get_admin_url(self):
content_type = ContentType.objects.get_for_model(self.__class__)
return reverse(
"admin:%s_%s_change" % (content_type.app_label, content_type.model),
args=(self.id,),
)
def get_absolute_url(self):
2020-08-16 19:45:38 +02:00
return reverse("article-detail", kwargs={"slug": self.slug})
2020-08-16 19:23:38 +02:00
def get_full_absolute_url(self, request: HttpRequest = None):
url = self.get_absolute_url()
if request:
return request.build_absolute_uri(url)
else:
return (settings.BLOG["base_url"] + url).replace("//", "/")
2020-08-14 15:53:42 +02:00
def get_abstract(self):
2020-08-14 16:15:40 +02:00
html = self.get_formatted_content()
2020-08-14 15:53:42 +02:00
return html.split("<!--more-->")[0]
def get_formatted_content(self):
2020-08-18 10:30:58 +02:00
md = markdown.Markdown(
2020-08-25 23:06:12 +02:00
extensions=[
"extra",
CodeHiliteExtension(linenums=False),
LazyLoadingImageExtension(),
]
2020-08-18 10:30:58 +02:00
)
2020-08-14 16:15:40 +02:00
content = self.content
content = re.sub(r"(\s)#(\w+)", r"\1\#\2", content)
return md.convert(content)
2020-08-16 18:15:19 +02:00
def publish(self):
2020-08-16 18:15:19 +02:00
if not self.published_at:
self.published_at = timezone.now()
self.status = self.PUBLISHED
self.save()
2020-08-16 18:15:19 +02:00
def unpublish(self):
2020-08-16 18:15:19 +02:00
self.published_at = None
self.status = self.DRAFT
self.save()
2020-08-16 19:45:38 +02:00
2020-08-17 09:57:24 +02:00
def save(self, *args, **kwargs):
2020-08-16 19:45:38 +02:00
if not self.slug:
self.slug = slugify(self.title)
return super().save(*args, **kwargs)
2020-08-17 09:57:24 +02:00
class Page(Article):
objects = models.Manager()
position = models.IntegerField(default=0)
class Meta:
ordering = ["position", "-published_at"]
2020-08-18 08:24:40 +02:00
2020-08-18 19:51:54 +02:00
class Comment(AdminUrlMixin, models.Model):
2020-08-18 21:40:54 +02:00
PENDING = "pending"
APPROVED = "approved"
REJECTED = "rejected"
STATUS_CHOICES = (
(PENDING, "Pending"),
(APPROVED, "Approved"),
(REJECTED, "Rejected"),
)
2020-08-18 18:49:41 +02:00
username = models.CharField(
max_length=255, help_text="Will be displayed with your comment."
)
email = models.EmailField(
blank=True,
null=True,
help_text=(
"Not mandatory, fill only if you want me to be able to contact you. "
2020-08-18 19:53:02 +02:00
"It will never be displayed here nor shared with any third party."
2020-08-18 18:49:41 +02:00
),
)
content = models.TextField(
2020-08-18 19:12:47 +02:00
max_length=500,
help_text="Your comment, limited to 500 characters. No formatting.",
2020-08-18 18:49:41 +02:00
)
article = models.ForeignKey(
Article, on_delete=models.CASCADE, related_name="comments"
)
created_at = models.DateTimeField(auto_now_add=True)
2020-08-18 21:40:54 +02:00
status = models.CharField(max_length=10, choices=STATUS_CHOICES, default=PENDING)
user_notified = models.BooleanField(default=False)
2020-08-18 18:49:41 +02:00
class Meta:
ordering = ["-created_at"]
2020-08-18 19:51:54 +02:00
2020-08-20 10:22:18 +02:00
def __str__(self):
return f"{self.username} - {self.content[:50]}"
2020-08-18 19:51:54 +02:00
def get_absolute_url(self):
return self.article.get_absolute_url() + "#" + str(self.id)
def get_full_absolute_url(self, request: HttpRequest = None):
return self.article.get_full_absolute_url(request) + "#" + str(self.id)