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

67 lines
1.9 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
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
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
2020-08-14 14:53:30 +02:00
class User(AbstractUser):
pass
2020-08-14 15:53:42 +02:00
class Article(models.Model):
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)
author = models.ForeignKey(User, on_delete=models.PROTECT)
views_count = models.IntegerField(default=0)
2020-08-14 15:53:42 +02:00
class Meta:
ordering = ["-published_at"]
2020-08-16 18:15:19 +02:00
def __str__(self):
return self.title
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):
return reverse("article-detail", kwargs={"pk": self.pk})
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):
md = markdown.Markdown(extensions=["extra"])
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()